mockGCP log

gemmahou · Nov 22, 2024 · 78cc28c · 78cc28c
1 parent 78a132c
commit 78cc28c
Show file tree

Hide file tree

Showing 9 changed files with 184 additions and 1,654 deletions.
diff --git a/mockgcp/mockcompute/globaltargettcpproxyv1.go b/mockgcp/mockcompute/globaltargettcpproxyv1.go
@@ -16,6 +16,7 @@ package mockcompute
 
 import (
 	"context"
+	"fmt"
 	"strings"
 
 	"github.com/GoogleCloudPlatform/k8s-config-connector/mockgcp/common/projects"
@@ -63,6 +64,15 @@ func (s *GlobalTargetTcpProxyV1) Insert(ctx context.Context, req *pb.InsertTarge
 	obj.CreationTimestamp = PtrTo(s.nowString())
 	obj.Id = &id
 	obj.Kind = PtrTo("compute#targetTcpProxy")
+
+	if obj.Service != nil {
+		service, err := s.parseGlobalBackendServiceName(obj.GetService())
+		if err != nil {
+			return nil, status.Errorf(codes.InvalidArgument, "backend service %q is not valid", obj.GetService())
+		}
+		obj.Service = PtrTo(fmt.Sprintf("https://www.googleapis.com/compute/v1/projects/%s/global/backendServices/%s", service.Project.ID, service.Name))
+	}
+
 	if obj.ProxyHeader == nil {
 		obj.ProxyHeader = PtrTo("NONE")
 	}
@@ -122,6 +132,14 @@ func (s *GlobalTargetTcpProxyV1) SetBackendService(ctx context.Context, req *pb.
 	}
 
 	obj.Service = req.GetTargetTcpProxiesSetBackendServiceRequestResource().Service
+	if obj.Service != nil {
+		service, err := s.parseGlobalBackendServiceName(obj.GetService())
+		if err != nil {
+			return nil, status.Errorf(codes.InvalidArgument, "backend service %q is not valid", obj.GetService())
+		}
+		obj.Service = PtrTo(fmt.Sprintf("https://www.googleapis.com/compute/v1/projects/%s/global/backendServices/%s", service.Project.ID, service.Name))
+	}
+
 	if err := s.storage.Update(ctx, fqn, obj); err != nil {
 		return nil, err
 	}

diff --git a/mockgcp/mockcompute/regionaltargettcpproxyv1.go b/mockgcp/mockcompute/regionaltargettcpproxyv1.go
@@ -16,6 +16,7 @@ package mockcompute
 
 import (
 	"context"
+	"fmt"
 	"strings"
 
 	"github.com/GoogleCloudPlatform/k8s-config-connector/mockgcp/common/projects"
@@ -65,6 +66,15 @@ func (s *RegionalTargetTcpProxyV1) Insert(ctx context.Context, req *pb.InsertReg
 	obj.Kind = PtrTo("compute#targetTcpProxy")
 	obj.Region = PtrTo("https://www.googleapis.com/compute/v1/projects/${projectId}/regions/" + req.GetRegion())
 
+	obj.Service = req.GetTargetTcpProxyResource().Service
+	if obj.Service != nil {
+		service, err := s.parseRegionalBackendServiceName(obj.GetService())
+		if err != nil {
+			return nil, status.Errorf(codes.InvalidArgument, "backend service %q is not valid", obj.GetService())
+		}
+		obj.Service = PtrTo(fmt.Sprintf("https://www.googleapis.com/compute/v1/projects/%s/regions/%s/backendServices/%s", service.Project.ID, service.Region, service.Name))
+	}
+
 	if err := s.storage.Create(ctx, fqn, obj); err != nil {
 		return nil, err
 	}