From 8020363954be0942143257cc0dbec76c5a891bf1 Mon Sep 17 00:00:00 2001
From: lostluck <13907733+lostluck@users.noreply.github.com>
Date: Fri, 28 Jul 2023 08:34:34 -0700
Subject: [PATCH] Add missing registrations to integration tests.

---
 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 a624efc0b81b..4a3a39b819e8 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 c444dedfe9d3..2c2383ea90ba 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]()