diff options
Diffstat (limited to 'internal/gitlabnet/twofactorrecover/client_test.go')
| -rw-r--r-- | internal/gitlabnet/twofactorrecover/client_test.go | 9 |
1 files changed, 5 insertions, 4 deletions
diff --git a/internal/gitlabnet/twofactorrecover/client_test.go b/internal/gitlabnet/twofactorrecover/client_test.go index 372afec..46291aa 100644 --- a/internal/gitlabnet/twofactorrecover/client_test.go +++ b/internal/gitlabnet/twofactorrecover/client_test.go @@ -1,6 +1,7 @@ package twofactorrecover import ( + "context" "encoding/json" "io/ioutil" "net/http" @@ -85,7 +86,7 @@ func TestGetRecoveryCodesByKeyId(t *testing.T) { defer cleanup() args := &commandargs.Shell{GitlabKeyId: "0"} - result, err := client.GetRecoveryCodes(args) + result, err := client.GetRecoveryCodes(context.Background(), args) assert.NoError(t, err) assert.Equal(t, []string{"recovery 1", "codes 1"}, result) } @@ -95,7 +96,7 @@ func TestGetRecoveryCodesByUsername(t *testing.T) { defer cleanup() args := &commandargs.Shell{GitlabUsername: "jane-doe"} - result, err := client.GetRecoveryCodes(args) + result, err := client.GetRecoveryCodes(context.Background(), args) assert.NoError(t, err) assert.Equal(t, []string{"recovery 2", "codes 2"}, result) } @@ -105,7 +106,7 @@ func TestMissingUser(t *testing.T) { defer cleanup() args := &commandargs.Shell{GitlabKeyId: "1"} - _, err := client.GetRecoveryCodes(args) + _, err := client.GetRecoveryCodes(context.Background(), args) assert.Equal(t, "missing user", err.Error()) } @@ -138,7 +139,7 @@ func TestErrorResponses(t *testing.T) { for _, tc := range testCases { t.Run(tc.desc, func(t *testing.T) { args := &commandargs.Shell{GitlabKeyId: tc.fakeId} - resp, err := client.GetRecoveryCodes(args) + resp, err := client.GetRecoveryCodes(context.Background(), args) assert.EqualError(t, err, tc.expectedError) assert.Nil(t, resp) |