diff options
Diffstat (limited to 'internal/source/gitlab/client/client_test.go')
-rw-r--r-- | internal/source/gitlab/client/client_test.go | 10 |
1 files changed, 4 insertions, 6 deletions
diff --git a/internal/source/gitlab/client/client_test.go b/internal/source/gitlab/client/client_test.go index 9b7632bd..5f6be9ca 100644 --- a/internal/source/gitlab/client/client_test.go +++ b/internal/source/gitlab/client/client_test.go @@ -10,8 +10,6 @@ import ( jwt "github.com/dgrijalva/jwt-go" "github.com/stretchr/testify/require" - - "gitlab.com/gitlab-org/gitlab-pages/internal/source/gitlab/api" ) var ( @@ -48,8 +46,8 @@ func TestNewInvalidBaseURL(t *testing.T) { func TestLookupForErrorResponses(t *testing.T) { tests := map[int]string{ - http.StatusUnauthorized: "Unauthorized", - http.StatusNotFound: "Not Found", + http.StatusUnauthorized: "HTTP status: 401", + http.StatusNotFound: "HTTP status: 404", } for statusCode, expectedError := range tests { @@ -70,7 +68,7 @@ func TestLookupForErrorResponses(t *testing.T) { lookup := client.GetLookup(context.Background(), "group.gitlab.io") require.EqualError(t, lookup.Error, expectedError) - require.Equal(t, lookup.Domain, api.VirtualDomain{}) + require.Nil(t, lookup.Domain) }) } } @@ -91,7 +89,7 @@ func TestMissingDomain(t *testing.T) { lookup := client.GetLookup(context.Background(), "group.gitlab.io") require.NoError(t, lookup.Error) - require.Equal(t, lookup.Domain, api.VirtualDomain{}) + require.Nil(t, lookup.Domain) } func TestGetVirtualDomainAuthenticatedRequest(t *testing.T) { |