Fixup browser after hidden API refactor

Interdepends on https://android-git.corp.google.com/g/#/c/198193

Change-Id: I9f44c0f4f84674965b7afc652420986314e2fc8e

1 files changed, 7 insertions, 2 deletions

diff --git a/tests/src/com/android/browser/TestWebViewClient.java b/tests/src/com/android/browser/TestWebViewClient.java
index 208a82249..6ab40dd59 100644
--- a/tests/src/com/android/browser/TestWebViewClient.java
+++ b/tests/src/com/android/browser/TestWebViewClient.java
@@ -25,6 +25,7 @@ import android.webkit.HttpAuthHandler;
 import android.webkit.SslErrorHandler;
 import android.webkit.WebView;
 import android.webkit.WebViewClient;
+import android.webkit.WebViewClientClassicExt;
 
 /**
  *
@@ -33,7 +34,7 @@ import android.webkit.WebViewClient;
  * Wraps around existing client so that specific methods can be overridden if needed.
  *
  */
-abstract class TestWebViewClient extends WebViewClient {
+abstract class TestWebViewClient extends WebViewClientClassicExt {
 
   private WebViewClient mWrappedClient;
 
@@ -105,7 +106,11 @@ abstract class TestWebViewClient extends WebViewClient {
   @Override
   public void onReceivedClientCertRequest(WebView view, ClientCertRequestHandler handler,
           String host_and_port) {
-      mWrappedClient.onReceivedClientCertRequest(view, handler, host_and_port);
+    if (mWrappedClient instanceof WebViewClientClassicExt) {
+      ((WebViewClientClassicExt) mWrappedClient).onReceivedClientCertRequest(view, handler, host_and_port);
+    } else {
+      super.onReceivedClientCertRequest(view, handler, host_and_port);
+    }
   }
 
   /** {@inheritDoc} */