Add with login url.

Author: chrishalbert

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:
 //

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)
+	})
+}