From ccd9713e4c70188024d2e96ec2085079ab637ee0 Mon Sep 17 00:00:00 2001 From: Miguel Crespo Date: Fri, 4 Oct 2024 16:27:29 +0100 Subject: [PATCH] rollout tests --- .../pkg/versions/versions_test.go | 22 ++++++++++++++----- 1 file changed, 16 insertions(+), 6 deletions(-) diff --git a/services/rollout-service/pkg/versions/versions_test.go b/services/rollout-service/pkg/versions/versions_test.go index 5b7c87904..26f7a7dbe 100644 --- a/services/rollout-service/pkg/versions/versions_test.go +++ b/services/rollout-service/pkg/versions/versions_test.go @@ -64,12 +64,13 @@ type mockOverviewStreamMessage struct { type mockOverviewClient struct { grpc.ClientStream - Responses map[string]*api.GetOverviewResponse - LastMetadata metadata.MD - StartStep chan struct{} - Steps chan step - savedStep *step - current int + Responses map[string]*api.GetOverviewResponse + AppDetailsResponses map[string]*api.GetAppDetailsResponse + LastMetadata metadata.MD + StartStep chan struct{} + Steps chan step + savedStep *step + current int } // GetOverview implements api.OverviewServiceClient @@ -81,6 +82,15 @@ func (m *mockOverviewClient) GetOverview(ctx context.Context, in *api.GetOvervie return nil, status.Error(codes.Unknown, "no") } +// GetOverview implements api.GetAppDetails +func (m *mockOverviewClient) GetAppDetails(ctx context.Context, in *api.GetAppDetailsRequest, opts ...grpc.CallOption) (*api.GetAppDetailsResponse, error) { + m.LastMetadata, _ = metadata.FromOutgoingContext(ctx) + if resp := m.AppDetailsResponses[in.AppName]; resp != nil { + return resp, nil + } + return nil, status.Error(codes.Unknown, "no") +} + // StreamOverview implements api.OverviewServiceClient func (m *mockOverviewClient) StreamOverview(ctx context.Context, in *api.GetOverviewRequest, opts ...grpc.CallOption) (api.OverviewService_StreamOverviewClient, error) { m.StartStep <- struct{}{}