rollout tests

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{}{}