diff options
Diffstat (limited to 'internal/gitlabnet/accessverifier/client_test.go')
-rw-r--r-- | internal/gitlabnet/accessverifier/client_test.go | 10 |
1 files changed, 9 insertions, 1 deletions
diff --git a/internal/gitlabnet/accessverifier/client_test.go b/internal/gitlabnet/accessverifier/client_test.go index 6a187f2..f2c88a5 100644 --- a/internal/gitlabnet/accessverifier/client_test.go +++ b/internal/gitlabnet/accessverifier/client_test.go @@ -56,7 +56,7 @@ func buildExpectedResponse(who string) *Response { func TestSuccessfulResponses(t *testing.T) { okResponse := testResponse{body: responseBody(t, "allowed.json"), status: http.StatusOK} client := setup(t, - map[string]testResponse{"first": okResponse}, + map[string]testResponse{"first": okResponse, "test@TEST.TEST": okResponse}, map[string]testResponse{"1": okResponse}, ) @@ -73,6 +73,10 @@ func TestSuccessfulResponses(t *testing.T) { desc: "Provide username within the request", args: &commandargs.Shell{GitlabUsername: "first"}, who: "user-1", + }, { + desc: "Provide krb5principal within the request", + args: &commandargs.Shell{GitlabKrb5Principal: "test@TEST.TEST"}, + who: "user-1", }, } @@ -255,6 +259,10 @@ func setup(t *testing.T, userResponses, keyResponses map[string]testResponse) *C w.WriteHeader(tr.status) _, err := w.Write(tr.body) require.NoError(t, err) + } else if tr, ok := userResponses[requestBody.Krb5Principal]; ok { + w.WriteHeader(tr.status) + _, err := w.Write(tr.body) + require.NoError(t, err) } else if tr, ok := keyResponses[requestBody.KeyId]; ok { w.WriteHeader(tr.status) _, err := w.Write(tr.body) |