conductor: "Adding mock service and resource for deploy-deliverypipeline"

Author: acpana

mockgcp/mockdeploy/deliverypipeline.go

@@ -0,0 +1,299 @@
+google.cloud.deploy.v1.CloudDeploy
+in.proto.service.definition: service CloudDeploy {
+  option (google.api.default_host) = "clouddeploy.googleapis.com";
+  option (google.api.oauth_scopes) =
+      "https://www.googleapis.com/auth/cloud-platform";
+
+  // Lists DeliveryPipelines in a given project and location.
+  rpc ListDeliveryPipelines(ListDeliveryPipelinesRequest)
+      returns (ListDeliveryPipelinesResponse) {
+    option (google.api.http) = {
+      get: "/v1/{parent=projects/*/locations/*}/deliveryPipelines"
+    };
+    option (google.api.method_signature) = "parent";
+  }
+
+  // Gets details of a single DeliveryPipeline.
+  rpc GetDeliveryPipeline(GetDeliveryPipelineRequest)
+      returns (DeliveryPipeline) {
+    option (google.api.http) = {
+      get: "/v1/{name=projects/*/locations/*/deliveryPipelines/*}"
+    };
+    option (google.api.method_signature) = "name";
+  }
+
+  // Creates a new DeliveryPipeline in a given project and location.
+  rpc CreateDeliveryPipeline(CreateDeliveryPipelineRequest)
+      returns (google.longrunning.Operation) {
+    option (google.api.http) = {
+      post: "/v1/{parent=projects/*/locations/*}/deliveryPipelines"
+      body: "delivery_pipeline"
+    };
+    option (google.api.method_signature) =
+        "parent,delivery_pipeline,delivery_pipeline_id";
+    option (google.longrunning.operation_info) = {
+      response_type: "DeliveryPipeline"
+      metadata_type: "OperationMetadata"
+    };
+  }
+
+  // Updates the parameters of a single DeliveryPipeline.
+  rpc UpdateDeliveryPipeline(UpdateDeliveryPipelineRequest)
+      returns (google.longrunning.Operation) {
+    option (google.api.http) = {
+      patch: "/v1/{delivery_pipeline.name=projects/*/locations/*/deliveryPipelines/*}"
+      body: "delivery_pipeline"
+    };
+    option (google.api.method_signature) = "delivery_pipeline,update_mask";
+    option (google.longrunning.operation_info) = {
+      response_type: "DeliveryPipeline"
+      metadata_type: "OperationMetadata"
+    };
+  }
+
+  // Deletes a single DeliveryPipeline.
+  rpc DeleteDeliveryPipeline(DeleteDeliveryPipelineRequest)
+      returns (google.longrunning.Operation) {
+    option (google.api.http) = {
+      delete: "/v1/{name=projects/*/locations/*/deliveryPipelines/*}"
+    };
+    option (google.api.method_signature) = "name";
+    option (google.longrunning.operation_info) = {
+      response_type: "google.protobuf.Empty"
+      metadata_type: "OperationMetadata"
+    };
+  }
+}
+out: // Copyright 2024 Google LLC
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//      http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+// +tool:mockgcp-support
+// proto.service: google.cloud.deploy.v1.CloudDeploy
+// proto.message: google.cloud.deploy.v1.DeliveryPipeline
+
+package mockclouddeploy
+
+import (
+	"context"
+	"fmt"
+	"regexp"
+	"strings"
+	"time"
+
+	"google.golang.org/grpc/codes"
+	"google.golang.org/grpc/status"
+	"google.golang.org/protobuf/proto"
+	"google.golang.org/protobuf/types/known/emptypb"
+	"google.golang.org/protobuf/types/known/timestamppb"
+
+	"github.com/GoogleCloudPlatform/k8s-config-connector/mockgcp/common/projects"
+	pb "github.com/GoogleCloudPlatform/k8s-config-connector/mockgcp/generated/mockgcp/cloud/deploy/v1"
+	longrunningpb "google.golang.org/genproto/googleapis/longrunning"
+)
+
+type cloudDeploy struct {
+	*MockService
+	pb.UnimplementedCloudDeployServer
+}
+
+func (s *cloudDeploy) GetDeliveryPipeline(ctx context.Context, req *pb.GetDeliveryPipelineRequest) (*pb.DeliveryPipeline, error) {
+	name, err := s.parseDeliveryPipelineName(req.Name)
+	if err != nil {
+		return nil, err
+	}
+
+	fqn := name.String()
+
+	obj := &pb.DeliveryPipeline{}
+	if err := s.storage.Get(ctx, fqn, obj); err != nil {
+		if status.Code(err) == codes.NotFound {
+			return nil, status.Errorf(codes.NotFound, "DeliveryPipeline %q not found", fqn)
+		}
+		return nil, err
+	}
+
+	return obj, nil
+}
+
+func (s *cloudDeploy) ListDeliveryPipelines(ctx context.Context, req *pb.ListDeliveryPipelinesRequest) (*pb.ListDeliveryPipelinesResponse, error) {
+	response := &pb.ListDeliveryPipelinesResponse{}
+
+	parent, err := s.parseParentName(req.GetParent())
+	if err != nil {
+		return nil, err
+	}
+
+	findPrefix := parent.String() + "/deliveryPipelines/"
+
+	deliveryPipelineKind := (&pb.DeliveryPipeline{}).ProtoReflect().Descriptor()
+	if err := s.storage.List(ctx, deliveryPipelineKind, storage.ListOptions{}, func(obj proto.Message) error {
+		deliveryPipeline := obj.(*pb.DeliveryPipeline)
+		if strings.HasPrefix(deliveryPipeline.Name, findPrefix) {
+			response.DeliveryPipelines = append(response.DeliveryPipelines, deliveryPipeline)
+		}
+		return nil
+	}); err != nil {
+		return nil, fmt.Errorf("error listing %v: %w", deliveryPipelineKind, err)
+	}
+
+	return response, nil
+}
+
+func (s *cloudDeploy) CreateDeliveryPipeline(ctx context.Context, req *pb.CreateDeliveryPipelineRequest) (*longrunningpb.Operation, error) {
+	reqName := req.Parent + "/deliveryPipelines/" + req.DeliveryPipelineId
+	name, err := s.parseDeliveryPipelineName(reqName)
+	if err != nil {
+		return nil, err
+	}
+
+	fqn := name.String()
+
+	now := time.Now()
+
+	obj := proto.Clone(req.DeliveryPipeline).(*pb.DeliveryPipeline)
+	obj.Name = fqn
+	obj.Uid = "mock-uid-" + name.DeliveryPipelineID + "-" + name.Project.ID
+	obj.CreateTime = timestamppb.New(now)
+	obj.UpdateTime = timestamppb.New(now)
+	obj.Condition = &pb.PipelineCondition{
+		PipelineReadyCondition: &pb.PipelineReadyCondition{
+			Status:             true,
+			LastUpdateTime:     timestamppb.New(now),
+			StatusMessage:      "",
+			StatusErrorCode:    "",
+			StatusErrorDetails: "",
+		},
+	}
+	s.populateDefaultsForDeliveryPipeline(obj)
+
+	if err := s.storage.Create(ctx, fqn, obj); err != nil {
+		return nil, err
+	}
+
+	lroPrefix := fmt.Sprintf("projects/%s/locations/%s", name.Project.ID, name.Location)
+	lroMetadata := &pb.OperationMetadata{
+		ApiVersion: "v1",
+		CreateTime: timestamppb.New(now),
+		Target:     name.String(),
+		Verb:       "create",
+	}
+	return s.operations.StartLRO(ctx, lroPrefix, lroMetadata, func() (proto.Message, error) {
+		lroMetadata.EndTime = timestamppb.Now()
+
+		return obj, nil
+	})
+}
+
+func (s *cloudDeploy) populateDefaultsForDeliveryPipeline(obj *pb.DeliveryPipeline) {
+	if obj.Etag == "" {
+		obj.Etag = computeEtag(obj)
+	}
+}
+
+func (s *cloudDeploy) DeleteDeliveryPipeline(ctx context.Context, req *pb.DeleteDeliveryPipelineRequest) (*longrunningpb.Operation, error) {
+	name, err := s.parseDeliveryPipelineName(req.Name)
+	if err != nil {
+		return nil, err
+	}
+
+	fqn := name.String()
+
+	deleted := &pb.DeliveryPipeline{}
+	if err := s.storage.Delete(ctx, fqn, deleted); err != nil {
+		return nil, err
+	}
+
+	lroPrefix := fmt.Sprintf("projects/%s/locations/%s", name.Project.ID, name.Location)
+	lroMetadata := &pb.OperationMetadata{
+		ApiVersion: "v1",
+		CreateTime: timestamppb.New(time.Now()),
+		Target:     name.String(),
+		Verb:       "delete",
+	}
+
+	return s.operations.StartLRO(ctx, lroPrefix, lroMetadata, func() (proto.Message, error) {
+		lroMetadata.EndTime = timestamppb.New(time.Now())
+
+		return &emptypb.Empty{}, nil
+	})
+}
+
+type deliveryPipelineName struct {
+	Project            *projects.ProjectData
+	Location           string
+	DeliveryPipelineID string
+}
+
+func (n *deliveryPipelineName) String() string {
+	return fmt.Sprintf("projects/%s/locations/%s/deliveryPipelines/%s", n.Project.ID, n.Location, n.DeliveryPipelineID)
+}
+
+// parseDeliveryPipelineName parses a string into an deliveryPipelineName.
+// The expected form is `projects/*/locations/*/deliveryPipelines/*`.
+func (s *MockService) parseDeliveryPipelineName(name string) (*deliveryPipelineName, error) {
+	tokens := strings.Split(name, "/")
+
+	if len(tokens) == 6 && tokens[0] == "projects" && tokens[2] == "locations" && tokens[4] == "deliveryPipelines" {
+		project, err := s.Projects.GetProjectByID(tokens[1])
+		if err != nil {
+			return nil, err
+		}
+
+		validID := regexp.MustCompile(`^[a-z]([a-z0-9-]{0,61}[a-z0-9])?$`).MatchString
+		if !validID(tokens[5]) {
+			return nil, status.Errorf(codes.InvalidArgument, "DeliveryPipeline id %q is not a valid name", name)
+		}
+
+		name := &deliveryPipelineName{
+			Project:            project,
+			Location:           tokens[3],
+			DeliveryPipelineID: tokens[5],
+		}
+
+		return name, nil
+	}
+
+	return nil, status.Errorf(codes.InvalidArgument, "name %q is not valid", name)
+}
+
+type parentName struct {
+	Project  *projects.ProjectData
+	Location string
+}
+
+func (n *parentName) String() string {
+	return fmt.Sprintf("projects/%s/locations/%s", n.Project.ID, n.Location)
+}
+
+func (s *MockService) parseParentName(name string) (*parentName, error) {
+	tokens := strings.Split(name, "/")
+
+	if len(tokens) == 4 && tokens[0] == "projects" && tokens[2] == "locations" {
+		project, err := s.Projects.GetProjectByID(tokens[1])
+		if err != nil {
+			return nil, err
+		}
+
+		name := &parentName{
+			Project:  project,
+			Location: tokens[3],
+		}
+
+		return name, nil
+	}
+
+	return nil, status.Errorf(codes.InvalidArgument, "parent name %q is not valid", name)
+}
+
+