From a068bab7e98339de131038551e7a84060b5c222f Mon Sep 17 00:00:00 2001 From: ddl-rliu Date: Wed, 10 Apr 2024 14:05:05 -0700 Subject: [PATCH] x Signed-off-by: ddl-rliu --- flyteadmin/auth/auth_context_test.go | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/flyteadmin/auth/auth_context_test.go b/flyteadmin/auth/auth_context_test.go index 4a3105ad4b..ed34d92026 100644 --- 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 {