/* * Copyright 2020 The Android Open Source Project * * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. * You may obtain a copy of the License at * * http://www.apache.org/licenses/LICENSE-2.0 * * Unless required by applicable law or agreed to in writing, software * distributed under the License is distributed on an "AS IS" BASIS, * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. * See the License for the specific language governing permissions and * limitations under the License. */ package androidx.webkit; import androidx.test.ext.junit.runners.AndroidJUnit4; import androidx.test.filters.MediumTest; import androidx.webkit.test.common.TestWebMessageListener; import androidx.webkit.test.common.WebViewOnUiThread; import androidx.webkit.test.common.WebkitUtils; import org.junit.After; import org.junit.Assert; import org.junit.Before; import org.junit.Test; import org.junit.runner.RunWith; import java.util.Collections; import java.util.HashSet; import java.util.Set; /** * Test * {@link WebViewCompat#addDocumentStartJavaScript} * * Test in Chromium JsJavaInteractionTest.java for these APIs are more comprehensive. */ @MediumTest @RunWith(AndroidJUnit4.class) public class WebViewDocumentStartJavaScriptTest { private static final String BASE_URI = "http://www.example.com"; private static final String JS_OBJECT_NAME = "myObject"; private static final String BASIC_USAGE = ""; private static final String BASIC_SCRIPT = "myObject.postMessage('hello');"; private static final Set MATCH_EXAMPLE_COM = new HashSet<>( Collections.singletonList(BASE_URI)); private WebViewOnUiThread mWebViewOnUiThread; private final TestWebMessageListener mListener = new TestWebMessageListener(); @Before public void setUp() { WebkitUtils.checkFeature(WebViewFeature.WEB_MESSAGE_LISTENER); WebkitUtils.checkFeature(WebViewFeature.DOCUMENT_START_SCRIPT); mWebViewOnUiThread = new WebViewOnUiThread(); mWebViewOnUiThread.getSettings().setJavaScriptEnabled(true); } @After public void tearDown() { if (mWebViewOnUiThread != null) { mWebViewOnUiThread.cleanUp(); } } @Test @SuppressWarnings("deprecation") // To be removed in 1.9.0 public void testAddDocumentStartJavaScriptBasicUsage() throws Exception { mWebViewOnUiThread.addWebMessageListener(JS_OBJECT_NAME, MATCH_EXAMPLE_COM, mListener); mWebViewOnUiThread.addDocumentStartJavaScript(BASIC_SCRIPT, MATCH_EXAMPLE_COM); loadHtmlSync(BASIC_USAGE); TestWebMessageListener.Data data = mListener.waitForOnPostMessage(); Assert.assertEquals("hello", data.mMessage.getData()); Assert.assertTrue("No more message at this point.", mListener.hasNoMoreOnPostMessage()); } @Test @SuppressWarnings("deprecation") // To be removed in 1.9.0 public void testAddDocumentStartJavaScriptRemoveScript() throws Exception { mWebViewOnUiThread.addWebMessageListener(JS_OBJECT_NAME, MATCH_EXAMPLE_COM, mListener); ScriptHandler scriptHandler = mWebViewOnUiThread.addDocumentStartJavaScript(BASIC_SCRIPT, MATCH_EXAMPLE_COM); loadHtmlSync(BASIC_USAGE); TestWebMessageListener.Data data = mListener.waitForOnPostMessage(); Assert.assertEquals("hello", data.mMessage.getData()); WebkitUtils.onMainThread(scriptHandler::remove); loadHtmlSync(BASIC_USAGE); Assert.assertTrue("No more message at this point.", mListener.hasNoMoreOnPostMessage()); } private void loadHtmlSync(String html) { mWebViewOnUiThread.loadDataWithBaseURLAndWaitForCompletion( BASE_URI, html, "text/html", null, null); } }