test: move mocks to their own package

9 files changed, 35 insertions, 35 deletions

diff --git a/internal/auth/auth_test.go b/internal/auth/auth_test.go
index d55c5a46..47e424c1 100644
--- a/internal/auth/auth_test.go
+++ b/internal/auth/auth_test.go
@@ -15,8 +15,8 @@ import (
 	"github.com/gorilla/sessions"
 	"github.com/stretchr/testify/require"
 
-	"gitlab.com/gitlab-org/gitlab-pages/internal/mocks"
 	"gitlab.com/gitlab-org/gitlab-pages/internal/request"
+	"gitlab.com/gitlab-org/gitlab-pages/internal/source/mock"
 )
 
 func createTestAuth(t *testing.T, internalServer string, publicServer string) *Auth {
@@ -86,7 +86,7 @@ func TestTryAuthenticate(t *testing.T) {
 
 	mockCtrl := gomock.NewController(t)
 
-	mockSource := mocks.NewMockSource(mockCtrl)
+	mockSource := mock.NewMockSource(mockCtrl)
 	require.False(t, auth.TryAuthenticate(result, r, mockSource))
 }
 
@@ -102,7 +102,7 @@ func TestTryAuthenticateWithError(t *testing.T) {
 
 	mockCtrl := gomock.NewController(t)
 
-	mockSource := mocks.NewMockSource(mockCtrl)
+	mockSource := mock.NewMockSource(mockCtrl)
 	require.True(t, auth.TryAuthenticate(result, r, mockSource))
 	require.Equal(t, http.StatusUnauthorized, result.Code)
 }
@@ -124,7 +124,7 @@ func TestTryAuthenticateWithCodeButInvalidState(t *testing.T) {
 
 	mockCtrl := gomock.NewController(t)
 
-	mockSource := mocks.NewMockSource(mockCtrl)
+	mockSource := mock.NewMockSource(mockCtrl)
 	require.True(t, auth.TryAuthenticate(result, r, mockSource))
 	require.Equal(t, http.StatusUnauthorized, result.Code)
 }
@@ -149,7 +149,7 @@ func TestTryAuthenticateRemoveTokenFromRedirect(t *testing.T) {
 
 	mockCtrl := gomock.NewController(t)
 
-	mockSource := mocks.NewMockSource(mockCtrl)
+	mockSource := mock.NewMockSource(mockCtrl)
 	require.True(t, auth.TryAuthenticate(result, r, mockSource))
 	require.Equal(t, http.StatusFound, result.Code)
 
@@ -168,7 +168,7 @@ func TestTryAuthenticateWithDomainAndState(t *testing.T) {
 
 	mockCtrl := gomock.NewController(t)
 
-	mockSource := mocks.NewMockSource(mockCtrl)
+	mockSource := mock.NewMockSource(mockCtrl)
 	require.True(t, auth.TryAuthenticate(result, r, mockSource))
 	require.Equal(t, http.StatusFound, result.Code)
 	redirect, err := url.Parse(result.Header().Get("Location"))
@@ -228,7 +228,7 @@ func testTryAuthenticateWithCodeAndState(t *testing.T, https bool) {
 
 	mockCtrl := gomock.NewController(t)
 
-	mockSource := mocks.NewMockSource(mockCtrl)
+	mockSource := mock.NewMockSource(mockCtrl)
 	require.True(t, auth.TryAuthenticate(result, r, mockSource))
 
 	res := result.Result()
@@ -518,7 +518,7 @@ func TestCheckResponseForInvalidTokenWhenNotInvalidToken(t *testing.T) {
 func TestDomainAllowed(t *testing.T) {
 	auth := createTestAuth(t, "", "")
 	mockCtrl := gomock.NewController(t)
-	mockSource := mocks.NewMockSource(mockCtrl)
+	mockSource := mock.NewMockSource(mockCtrl)
 
 	testCases := []struct {
 		name     string
diff --git a/internal/domain/domain_test.go b/internal/domain/domain_test.go
index 408ddf38..3496a6da 100644
--- a/internal/domain/domain_test.go
+++ b/internal/domain/domain_test.go
@@ -9,8 +9,8 @@ import (
 	"github.com/stretchr/testify/require"
 
 	"gitlab.com/gitlab-org/gitlab-pages/internal/domain"
+	"gitlab.com/gitlab-org/gitlab-pages/internal/domain/mock"
 	"gitlab.com/gitlab-org/gitlab-pages/internal/fixture"
-	"gitlab.com/gitlab-org/gitlab-pages/internal/mocks"
 	"gitlab.com/gitlab-org/gitlab-pages/internal/serving"
 	"gitlab.com/gitlab-org/gitlab-pages/internal/serving/disk/local"
 	"gitlab.com/gitlab-org/gitlab-pages/internal/testhelpers"
@@ -208,7 +208,7 @@ func TestServeNamespaceNotFound(t *testing.T) {
 func mockResolver(t *testing.T, project *serving.LookupPath, subpath string, err error) domain.Resolver {
 	mockCtrl := gomock.NewController(t)
 
-	mockResolver := mocks.NewMockResolver(mockCtrl)
+	mockResolver := mock.NewMockResolver(mockCtrl)
 
 	mockResolver.EXPECT().
 		Resolve(gomock.Any()).
diff --git a/internal/mocks/resolver.go b/internal/domain/mock/resolver_mock.go
index 14b86a9c..8231b2b4 100644
--- a/internal/mocks/resolver.go
+++ b/internal/domain/mock/resolver_mock.go
@@ -1,8 +1,8 @@
 // Code generated by MockGen. DO NOT EDIT.
 // Source: internal/domain/resolver.go
 
-// Package mocks is a generated GoMock package.
-package mocks
+// Package mock is a generated GoMock package.
+package mock
 
 import (
 	http "net/http"
diff --git a/internal/handlers/handlers_test.go b/internal/handlers/handlers_test.go
index dd940251..22027ad0 100644
--- a/internal/handlers/handlers_test.go
+++ b/internal/handlers/handlers_test.go
@@ -10,19 +10,19 @@ import (
 	"github.com/golang/mock/gomock"
 	"github.com/stretchr/testify/require"
 
-	"gitlab.com/gitlab-org/gitlab-pages/internal/mocks"
+	"gitlab.com/gitlab-org/gitlab-pages/internal/handlers/mock"
 )
 
 func TestNotHandleArtifactRequestReturnsFalse(t *testing.T) {
 	mockCtrl := gomock.NewController(t)
 
-	mockArtifact := mocks.NewMockArtifact(mockCtrl)
+	mockArtifact := mock.NewMockArtifact(mockCtrl)
 	mockArtifact.EXPECT().
 		TryMakeRequest(gomock.Any(), gomock.Any(), gomock.Any(), "", gomock.Any()).
 		Return(false).
 		Times(1)
 
-	mockAuth := mocks.NewMockAuth(mockCtrl)
+	mockAuth := mock.NewMockAuth(mockCtrl)
 	mockAuth.EXPECT().
 		GetTokenIfExists(gomock.Any(), gomock.Any()).
 		Return("", nil).
@@ -41,13 +41,13 @@ func TestNotHandleArtifactRequestReturnsFalse(t *testing.T) {
 func TestHandleArtifactRequestedReturnsTrue(t *testing.T) {
 	mockCtrl := gomock.NewController(t)
 
-	mockArtifact := mocks.NewMockArtifact(mockCtrl)
+	mockArtifact := mock.NewMockArtifact(mockCtrl)
 	mockArtifact.EXPECT().
 		TryMakeRequest(gomock.Any(), gomock.Any(), gomock.Any(), "", gomock.Any()).
 		Return(true).
 		Times(1)
 
-	mockAuth := mocks.NewMockAuth(mockCtrl)
+	mockAuth := mock.NewMockAuth(mockCtrl)
 	mockAuth.EXPECT().
 		GetTokenIfExists(gomock.Any(), gomock.Any()).
 		Return("", nil).
@@ -64,7 +64,7 @@ func TestHandleArtifactRequestedReturnsTrue(t *testing.T) {
 func TestNotFoundWithTokenIsNotHandled(t *testing.T) {
 	mockCtrl := gomock.NewController(t)
 
-	mockAuth := mocks.NewMockAuth(mockCtrl)
+	mockAuth := mock.NewMockAuth(mockCtrl)
 	mockAuth.EXPECT().CheckResponseForInvalidToken(gomock.Any(), gomock.Any(), gomock.Any()).
 		Return(false)
 
@@ -101,7 +101,7 @@ func TestForbiddenWithTokenIsNotHandled(t *testing.T) {
 		t.Run(tn, func(t *testing.T) {
 			mockCtrl := gomock.NewController(t)
 
-			mockAuth := mocks.NewMockAuth(mockCtrl)
+			mockAuth := mock.NewMockAuth(mockCtrl)
 			if tc.Token == "" {
 				mockAuth.EXPECT().IsAuthSupported().Return(true)
 				mockAuth.EXPECT().RequireAuth(gomock.Any(), gomock.Any()).Return(true)
@@ -125,7 +125,7 @@ func TestForbiddenWithTokenIsNotHandled(t *testing.T) {
 func TestNotFoundWithoutTokenIsNotHandledWhenNotAuthSupport(t *testing.T) {
 	mockCtrl := gomock.NewController(t)
 
-	mockAuth := mocks.NewMockAuth(mockCtrl)
+	mockAuth := mock.NewMockAuth(mockCtrl)
 	mockAuth.EXPECT().IsAuthSupported().Return(false)
 
 	handlers := New(mockAuth, nil)
@@ -140,7 +140,7 @@ func TestNotFoundWithoutTokenIsNotHandledWhenNotAuthSupport(t *testing.T) {
 func TestNotFoundWithoutTokenIsHandled(t *testing.T) {
 	mockCtrl := gomock.NewController(t)
 
-	mockAuth := mocks.NewMockAuth(mockCtrl)
+	mockAuth := mock.NewMockAuth(mockCtrl)
 	mockAuth.EXPECT().IsAuthSupported().Return(true)
 	mockAuth.EXPECT().RequireAuth(gomock.Any(), gomock.Any()).Times(1).Return(true)
 
@@ -156,7 +156,7 @@ func TestNotFoundWithoutTokenIsHandled(t *testing.T) {
 func TestInvalidTokenResponseIsHandled(t *testing.T) {
 	mockCtrl := gomock.NewController(t)
 
-	mockAuth := mocks.NewMockAuth(mockCtrl)
+	mockAuth := mock.NewMockAuth(mockCtrl)
 	mockAuth.EXPECT().CheckResponseForInvalidToken(gomock.Any(), gomock.Any(), gomock.Any()).
 		Return(true)
 
@@ -173,12 +173,12 @@ func TestInvalidTokenResponseIsHandled(t *testing.T) {
 func TestHandleArtifactRequestButGetTokenFails(t *testing.T) {
 	mockCtrl := gomock.NewController(t)
 
-	mockArtifact := mocks.NewMockArtifact(mockCtrl)
+	mockArtifact := mock.NewMockArtifact(mockCtrl)
 	mockArtifact.EXPECT().
 		TryMakeRequest(gomock.Any(), gomock.Any(), gomock.Any(), "", gomock.Any()).
 		Times(0)
 
-	mockAuth := mocks.NewMockAuth(mockCtrl)
+	mockAuth := mock.NewMockAuth(mockCtrl)
 	mockAuth.EXPECT().GetTokenIfExists(gomock.Any(), gomock.Any()).Return("", errors.New("error when retrieving token"))
 
 	handlers := New(mockAuth, mockArtifact)
diff --git a/internal/mocks/mocks.go b/internal/handlers/mock/handler_mock.go
index b18bede4..11548221 100644
--- a/internal/mocks/mocks.go
+++ b/internal/handlers/mock/handler_mock.go
@@ -1,8 +1,8 @@
 // Code generated by MockGen. DO NOT EDIT.
 // Source: internal/interface.go
 
-// Package mocks is a generated GoMock package.
-package mocks
+// Package mock is a generated GoMock package.
+package mock
 
 import (
 	http "net/http"
diff --git a/internal/source/gitlab/gitlab_test.go b/internal/source/gitlab/gitlab_test.go
index d7fbf454..e7f80387 100644
--- a/internal/source/gitlab/gitlab_test.go
+++ b/internal/source/gitlab/gitlab_test.go
@@ -12,9 +12,9 @@ import (
 	"github.com/golang/mock/gomock"
 	"github.com/stretchr/testify/require"
 
-	"gitlab.com/gitlab-org/gitlab-pages/internal/mocks"
 	"gitlab.com/gitlab-org/gitlab-pages/internal/source/gitlab/api"
 	"gitlab.com/gitlab-org/gitlab-pages/internal/source/gitlab/client"
+	"gitlab.com/gitlab-org/gitlab-pages/internal/source/gitlab/mock"
 )
 
 func TestGetDomain(t *testing.T) {
@@ -171,10 +171,10 @@ func TestResolveLookupPathsOrderDoesNotMatter(t *testing.T) {
 	}
 }
 
-func NewMockClient(t *testing.T, file string, mockedLookup *api.Lookup) *mocks.MockClientStub {
+func NewMockClient(t *testing.T, file string, mockedLookup *api.Lookup) *mock.MockClientStub {
 	mockCtrl := gomock.NewController(t)
 
-	mockClient := mocks.NewMockClientStub(mockCtrl)
+	mockClient := mock.NewMockClientStub(mockCtrl)
 	mockClient.EXPECT().
 		Resolve(gomock.Any(), gomock.Any()).
 		DoAndReturn(func(ctx context.Context, domain string) *api.Lookup {
diff --git a/internal/mocks/client.go b/internal/source/gitlab/mock/client_mock.go
index 22b0757e..6f78b01b 100644
--- a/internal/mocks/client.go
+++ b/internal/source/gitlab/mock/client_mock.go
@@ -1,8 +1,8 @@
 // Code generated by MockGen. DO NOT EDIT.
-// Source: internal/mocks/api/client_stub.go
+// Source: internal/source/gitlab/mock/client_stub.go
 
-// Package mocks is a generated GoMock package.
-package mocks
+// Package mock is a generated GoMock package.
+package mock
 
 import (
 	context "context"
diff --git a/internal/mocks/api/client_stub.go b/internal/source/gitlab/mock/client_stub.go
index f1a20754..5b201da2 100644
--- a/internal/mocks/api/client_stub.go
+++ b/internal/source/gitlab/mock/client_stub.go
@@ -1,4 +1,4 @@
-package mocks
+package mock
 
 import "gitlab.com/gitlab-org/gitlab-pages/internal/source/gitlab/api"
 
diff --git a/internal/mocks/source.go b/internal/source/mock/source_mock.go
index 5412310f..6ec884f1 100644
--- a/internal/mocks/source.go
+++ b/internal/source/mock/source_mock.go
@@ -1,8 +1,8 @@
 // Code generated by MockGen. DO NOT EDIT.
 // Source: internal/source/source.go
 
-// Package mocks is a generated GoMock package.
-package mocks
+// Package mock is a generated GoMock package.
+package mock
 
 import (
 	context "context"