x

flyteorg · Apr 10, 2024 · 1425610 · 1425610
1 parent bb76066
commit 1425610
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/flyteadmin/auth/auth_context_test.go b/flyteadmin/auth/auth_context_test.go
@@ -30,6 +30,16 @@ func TestOAuth2ClientConfig(t *testing.T) {
 			url:                 "https://flyte.com/custom-subpath/console",
 			expectedRedirectURL: "https://flyte.com/custom-subpath/callback",
 		},
+		{
+			name:                "complex publicUrl",
+			url:                 "https://flyte.com/login?redirect_url=https://flyte.com/console/select-project",
+			expectedRedirectURL: "https://flyte.com/callback",
+		},
+		{
+			name:                "complex publicUrl with custom subpath",
+			url:                 "https://flyte.com/custom-subpath/login?redirect_url=https://flyte.com/custom-subpath/console/select-project",
+			expectedRedirectURL: "https://flyte.com/custom-subpath/callback",
+		},
 	}
 
 	for _, tt := range tests {