Fix updates in mockalloydb

GoogleCloudPlatform · Dec 17, 2024 · 431125c · 431125c
1 parent e8292a7
commit 431125c
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/mockgcp/mockalloydb/instance.go b/mockgcp/mockalloydb/instance.go
@@ -17,6 +17,7 @@ package mockalloydb
 import (
 	"context"
 	"fmt"
+	"strings"
 
 	"google.golang.org/genproto/googleapis/longrunning"
 	"google.golang.org/grpc/codes"
@@ -188,7 +189,8 @@ func (s *AlloyDBAdminV1) UpdateInstance(ctx context.Context, req *pb.UpdateInsta
 
 	// TODO: Some sort of helper for fieldmask?
 	for _, path := range paths {
-		switch path {
+		topLevelField := strings.Split(path, ".")[0]
+		switch topLevelField {
 		case "labels":
 			obj.Labels = req.Instance.GetLabels()
 		case "annotations":