fix: lint & test

googleforgames · Dec 10, 2024 · 4e338b1 · 4e338b1
1 parent f91758f
commit 4e338b1
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/pkg/sdkserver/localsdk_test.go b/pkg/sdkserver/localsdk_test.go
@@ -327,7 +327,7 @@ func TestLocalSDKServerSetAnnotations(t *testing.T) {
 			assertInitialWatchUpdate(t, stream)
 
 			// make sure length of l.updateObservers is at least 1
-			err = wait.PollUntilContextTimeout(context.Background(), time.Second, 10*time.Second, true, func(ctx context.Context) (bool, error) {
+			err = wait.PollUntilContextTimeout(context.Background(), time.Second, 10*time.Second, true, func(context.Context) (bool, error) {
 				ret := false
 				l.updateObservers.Range(func(_, _ interface{}) bool {
 					ret = true

diff --git a/sdks/go/sdk_test.go b/sdks/go/sdk_test.go
@@ -153,6 +153,13 @@ func (m *sdkMock) SetAnnotation(_ context.Context, in *sdk.KeyValue, _ ...grpc.C
 	return &sdk.Empty{}, nil
 }
 
+func (m *sdkMock) SetAnnotations(_ context.Context, in *sdk.KeyValues, _ ...grpc.CallOption) (*sdk.Empty, error) {
+	for _, kv := range in.KeyValues {
+		m.annotations["agones.dev/sdk-"+kv.Key] = kv.Value
+	}
+	return &sdk.Empty{}, nil
+}
+
 func (m *sdkMock) WatchGameServer(_ context.Context, _ *sdk.Empty, _ ...grpc.CallOption) (sdk.SDK_WatchGameServerClient, error) {
 	return m.wm, nil
 }