Fixed tests that were broken due to a bad merge.

Fixes: 112436631

Test: atest WebViewActivityTest

Change-Id: Ia310a5cc387b3005b81e35bf9dd50d7f7fa48c6d
diff --git a/tests/autofillservice/src/android/autofillservice/cts/WebViewActivity.java b/tests/autofillservice/src/android/autofillservice/cts/WebViewActivity.java
index 08d4c70..e51bbfc 100644
--- a/tests/autofillservice/src/android/autofillservice/cts/WebViewActivity.java
+++ b/tests/autofillservice/src/android/autofillservice/cts/WebViewActivity.java
@@ -17,6 +17,8 @@
 
 import static android.autofillservice.cts.Timeouts.WEBVIEW_TIMEOUT;
 
+import static com.google.common.truth.Truth.assertThat;
+
 import android.content.Context;
 import android.os.Bundle;
 import android.os.SystemClock;
@@ -112,6 +114,7 @@
 
             });
             mWebView.loadUrl(FAKE_URL);
+            assertThat(mWebView.isAutofillEnabled()).isTrue();
         });
 
         // Wait until it's loaded.
diff --git a/tests/autofillservice/src/android/autofillservice/cts/WebViewActivityTest.java b/tests/autofillservice/src/android/autofillservice/cts/WebViewActivityTest.java
index 57bca35..7149899 100644
--- a/tests/autofillservice/src/android/autofillservice/cts/WebViewActivityTest.java
+++ b/tests/autofillservice/src/android/autofillservice/cts/WebViewActivityTest.java
@@ -71,7 +71,6 @@
     @Before
     public void setActivity() {
         mActivity = mActivityRule.getActivity();
-        assertAutofillEnabledOnWebView();
     }
 
     @BeforeClass
@@ -611,8 +610,4 @@
                 ID_OUTSIDE2);
         Helper.assertTextAndValue(outside2SaveNode, "SWEETER");
     }
-
-    private void assertAutofillEnabledOnWebView() {
-        assertThat(mActivity.mWebView.isAutofillEnabled()).isTrue();
-    }
 }