[beam] branch master updated: Add missing registrations to integration tests. (#27738)

[lo...@apache.org]

This is an automated email from the ASF dual-hosted git repository.

lostluck pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/beam.git


The following commit(s) were added to refs/heads/master by this push:
     new 7bfead01656 Add missing registrations to integration tests. (#27738)
7bfead01656 is described below

commit 7bfead0165627433b629118b95f4e1ba6b15db7f
Author: Robert Burke <lo...@users.noreply.github.com>
AuthorDate: Fri Jul 28 10:05:54 2023 -0700

    Add missing registrations to integration tests. (#27738)
    
    Co-authored-by: lostluck <13...@users.noreply.github.com>
---
 sdks/go/test/integration/primitives/cogbk.go | 14 ++++++++++++++
 sdks/go/test/integration/primitives/pardo.go |  1 +
 2 files changed, 15 insertions(+)

diff --git a/sdks/go/test/integration/primitives/cogbk.go b/sdks/go/test/integration/primitives/cogbk.go
index a624efc0b81..4a3a39b819e 100644
--- a/sdks/go/test/integration/primitives/cogbk.go
+++ b/sdks/go/test/integration/primitives/cogbk.go
@@ -20,9 +20,23 @@ import (
 	"fmt"
 
 	"github.com/apache/beam/sdks/v2/go/pkg/beam"
+	"github.com/apache/beam/sdks/v2/go/pkg/beam/register"
 	"github.com/apache/beam/sdks/v2/go/pkg/beam/testing/passert"
 )
 
+func init() {
+	register.Function2x0(genA)
+	register.Function2x0(genB)
+	register.Function2x0(genC)
+	register.Function2x0(genD)
+	register.Function3x0(shortFn)
+	register.Function5x0(joinFn)
+	register.Function6x0(splitFn)
+	register.Emitter2[string, int]()
+	register.Emitter2[string, string]()
+	register.Iter1[int]()
+}
+
 func genA(_ []byte, emit func(string, int)) {
 	emit("a", 1)
 	emit("a", 2)
diff --git a/sdks/go/test/integration/primitives/pardo.go b/sdks/go/test/integration/primitives/pardo.go
index c444dedfe9d..2c2383ea90b 100644
--- a/sdks/go/test/integration/primitives/pardo.go
+++ b/sdks/go/test/integration/primitives/pardo.go
@@ -31,6 +31,7 @@ func init() {
 	register.Function1x2(splitStringPair)
 	register.Function3x2(asymJoinFn)
 	register.Function5x0(splitByName)
+	register.Function2x0(emitPipelineOptions)
 
 	register.Iter1[int]()
 	register.Iter2[int, int]()