diff --git a/common/src/test/java/com/microsoft/identity/common/internal/ui/webview/AzureActiveDirectoryWebViewClientTest.java b/common/src/test/java/com/microsoft/identity/common/internal/ui/webview/AzureActiveDirectoryWebViewClientTest.java index b267d2b829..01d6aee8c6 100644 --- a/common/src/test/java/com/microsoft/identity/common/internal/ui/webview/AzureActiveDirectoryWebViewClientTest.java +++ b/common/src/test/java/com/microsoft/identity/common/internal/ui/webview/AzureActiveDirectoryWebViewClientTest.java @@ -62,6 +62,7 @@ import com.microsoft.identity.common.java.ui.webview.authorization.IAuthorizationCompletionCallback; import com.microsoft.identity.common.shadows.ShadowProcessUtil; +import org.junit.After; import org.junit.Assert; import org.junit.Before; import org.junit.Test; @@ -159,6 +160,11 @@ public void onPageLoaded(final String url) { } } + @After + public void cleanUp(){ + CommonFlightsManager.INSTANCE.resetFlightsManager(); + } + @Test(expected = IllegalArgumentException.class) public void testUrlOverrideHandlesEmptyString() { assertTrue(mWebViewClient.shouldOverrideUrlLoading(mMockWebView, "")); @@ -489,6 +495,11 @@ public void testOnReceivedSslError_Legacy() { final SslError mockError = Mockito.mock(android.net.http.SslError.class); final IAuthorizationCompletionCallback mockCallback = Mockito.mock(IAuthorizationCompletionCallback.class); when(mockError.getUrl()).thenReturn("https://example.com"); + final IFlightsManager mockFlightsManager = Mockito.mock(IFlightsManager.class); + final IFlightsProvider mockFlightsProvider = Mockito.mock(IFlightsProvider.class); + when(mockFlightsProvider.isFlightEnabled(eq(CommonFlight.SHOULD_PRESERVE_WEBVIEW_FLOW_ON_SSL_ERROR))).thenReturn(false); + when(mockFlightsManager.getFlightsProvider(anyLong())).thenReturn(mockFlightsProvider); + CommonFlightsManager.INSTANCE.initializeCommonFlightsManager(mockFlightsManager); final AzureActiveDirectoryWebViewClient mockWebViewClient = new AzureActiveDirectoryWebViewClient( mActivity, mockCallback,