Add missing registrations to integration tests. (#27738)

Co-authored-by: lostluck <[email protected]>
apache · Jul 28, 2023 · 7bfead0 · 7bfead0
1 parent cc410ca
commit 7bfead0
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 0 deletions.
diff --git 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
@@ -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]()