Work on tests

1 files changed, 8 insertions, 5 deletions

diff --git a/internal/auth/auth_test.go b/internal/auth/auth_test.go
index ed130caf..01208997 100644
--- a/internal/auth/auth_test.go
+++ b/internal/auth/auth_test.go
@@ -5,7 +5,6 @@ import (
 	"net/http"
 	"net/http/httptest"
 	"net/url"
-	"sync"
 	"testing"
 
 	"github.com/gorilla/sessions"
@@ -16,6 +15,10 @@ import (
 	"gitlab.com/gitlab-org/gitlab-pages/internal/domain"
 )
 
+func findDomain(host string) *domain.D {
+	return nil
+}
+
 func createAuth(t *testing.T) *auth.Auth {
 	return auth.New("pages.gitlab-example.com",
 		"something-very-secret",
@@ -33,7 +36,7 @@ func TestTryAuthenticate(t *testing.T) {
 	require.NoError(t, err)
 	r := &http.Request{URL: reqURL}
 
-	assert.Equal(t, false, auth.TryAuthenticate(result, r, make(domain.Map), &sync.RWMutex{}))
+	assert.Equal(t, false, auth.TryAuthenticate(result, r, findDomain))
 }
 
 func TestTryAuthenticateWithError(t *testing.T) {
@@ -44,7 +47,7 @@ func TestTryAuthenticateWithError(t *testing.T) {
 	require.NoError(t, err)
 	r := &http.Request{URL: reqURL}
 
-	assert.Equal(t, true, auth.TryAuthenticate(result, r, make(domain.Map), &sync.RWMutex{}))
+	assert.Equal(t, true, auth.TryAuthenticate(result, r, findDomain))
 	assert.Equal(t, 401, result.Code)
 }
 
@@ -61,7 +64,7 @@ func TestTryAuthenticateWithCodeButInvalidState(t *testing.T) {
 	session.Values["state"] = "state"
 	session.Save(r, result)
 
-	assert.Equal(t, true, auth.TryAuthenticate(result, r, make(domain.Map), &sync.RWMutex{}))
+	assert.Equal(t, true, auth.TryAuthenticate(result, r, findDomain))
 	assert.Equal(t, 401, result.Code)
 }
 
@@ -103,7 +106,7 @@ func TestTryAuthenticateWithCodeAndState(t *testing.T) {
 	session.Values["state"] = "state"
 	session.Save(r, result)
 
-	assert.Equal(t, true, auth.TryAuthenticate(result, r, make(domain.Map), &sync.RWMutex{}))
+	assert.Equal(t, true, auth.TryAuthenticate(result, r, findDomain))
 	assert.Equal(t, 302, result.Code)
 	assert.Equal(t, "http://pages.gitlab-example.com/project/", result.Header().Get("Location"))
 }