From 2f96dac5e648194dd4b8c4e32fe13fa754f86adb Mon Sep 17 00:00:00 2001 From: Brenna Hewer-Darroch Date: Mon, 2 Mar 2020 17:37:53 -0800 Subject: [PATCH] fix adminmgmt_test post-rebase Signed-off-by: Brenna Hewer-Darroch --- .../pkg/adminmgmt/adminmgmt_test.go | 28 ++++++++----------- 1 file changed, 12 insertions(+), 16 deletions(-) diff --git a/components/automate-cli/pkg/adminmgmt/adminmgmt_test.go b/components/automate-cli/pkg/adminmgmt/adminmgmt_test.go index 62c34f9e2e28..6da2d35674e1 100644 --- a/components/automate-cli/pkg/adminmgmt/adminmgmt_test.go +++ b/components/automate-cli/pkg/adminmgmt/adminmgmt_test.go @@ -176,6 +176,11 @@ func TestEnsureTeam(t *testing.T) { }, nil } + serverMocks.TeamsMock.GetTeamFunc = func( + _ context.Context, req *iam_req.GetTeamReq) (*iam_resp.GetTeamResp, error) { + return nil, status.Error(codes.NotFound, "Not Found") + } + found, err := adminmgmt.CreateAdminTeamIfMissing(ctx, apiClient, false) require.NoError(t, err) assert.False(t, found) @@ -191,22 +196,13 @@ func TestEnsureTeam(t *testing.T) { }) t.Run("when the admins team exists already", func(t *testing.T) { - mockedAdminsID := "mocked-admin-id" - serverMocks.TeamsMock.ListTeamsFunc = func( - context.Context, *iam_req.ListTeamsReq) (*iam_resp.ListTeamsResp, error) { - - return &iam_resp.ListTeamsResp{ - Teams: []*iam_common.Team{ - { - Id: "mocked-not-admin-id", - Name: "not-admin", - Projects: []string{}, - }, - { - Id: mockedAdminsID, - Name: "admins", - Projects: []string{}, - }, + serverMocks.TeamsMock.GetTeamFunc = func( + _ context.Context, req *iam_req.GetTeamReq) (*iam_resp.GetTeamResp, error) { + return &iam_resp.GetTeamResp{ + Team: &iam_common.Team{ + Id: "admins", + Name: "admins", + Projects: []string{}, }, }, nil }