Add with login url.

zalando · Jan 16, 2024 · be30822 · be30822
1 parent a860885
commit be30822
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 0 deletions.
diff --git a/google/google.go b/google/google.go
@@ -116,6 +116,10 @@ func GetLoginURL(state string) string {
 	return conf.AuthCodeURL(state)
 }
 
+func WithLoginURL(url string) {
+	loginURL = url
+}
+
 // Auth is the google authorization middleware. You can use them to protect a routergroup.
 // Example:
 //

diff --git a/google/google_test.go b/google/google_test.go
@@ -17,3 +17,13 @@ func TestSetupFromString(t *testing.T) {
 		assert.Equal(t, conf.ClientSecret, "clientsecret")
 	})
 }
+
+func TestWithLoginURL(t *testing.T) {
+	t.Run("should assign the login url", func(t *testing.T) {
+		loginURL = ""
+		url := "http://fake.fake"
+		WithLoginURL(url)
+		assert.NotEmpty(t, url)
+		assert.Equal(t, url, loginURL)
+	})
+}