one more test

Signed-off-by: Haytham Abuelfutuh <[email protected]>
flyteorg · Feb 27, 2021 · bb04640 · bb04640
1 parent 150bc21
commit bb04640
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/pkg/auth/handlers_test.go b/pkg/auth/handlers_test.go
@@ -99,11 +99,11 @@ func TestGetMetadataEndpointRedirectHandler(t *testing.T) {
 	ctx := context.Background()
 	baseURL, err := url.Parse("http://www.google.com")
 	assert.NoError(t, err)
-	metadataPath, err := url.Parse(OIdCMetadataEndpoint)
+	metadataPath, err := url.Parse(OAuth2MetadataEndpoint)
 	assert.NoError(t, err)
 	mockAuthCtx := mocks.AuthenticationContext{}
-	mockAuthCtx.On("GetBaseURL").Return(baseURL)
-	mockAuthCtx.On("GetMetadataURL").Return(metadataPath)
+	mockAuthCtx.OnGetBaseURL().Return(baseURL)
+	mockAuthCtx.OnGetOAuth2MetadataURL().Return(metadataPath)
 	handler := GetOAuth2MetadataEndpointRedirectHandler(ctx, &mockAuthCtx)
 	req, err := http.NewRequest("GET", "/xyz", nil)
 	assert.NoError(t, err)