Fix cancelling permission requests on Android when the PermissionRequestManager is enabled

chrome/android/javatests/src/org/chromium/chrome/browser/permissions/PermissionTestCaseBase.java

 // Copyright 2016 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 package org.chromium.chrome.browser.permissions;
 
 import android.content.DialogInterface;
 import android.support.v7.app.AlertDialog;
 import android.support.v7.widget.SwitchCompat;
 
@@ skipping 118 matching lines @@
         container.addAnimationListener(mListener);
         mTestServer = EmbeddedTestServer.createAndStartServer(getInstrumentation().getContext());
     }
 
     @Override
     protected void tearDown() throws Exception {
         mTestServer.stopAndDestroyServer();
         super.tearDown();
     }
 
+    protected void setUpUrl(final String url) throws InterruptedException {
+        final String test_url = mTestServer.getURL(url);
+        loadUrl(test_url);
+    }
+
+    /**
+     * Checks permission status via the navigator.permissions.query API
+     */
+    protected void checkPermissionStatus(final String permission, final String expected)
+            throws Exception {
+        final CallbackHelper callbackHelper = new CallbackHelper();
+        EmptyTabObserver updateWaiter = new EmptyTabObserver() {
+            @Override
+            public void onTitleUpdated(Tab tab) {
+                String title = getActivity().getActivityTab().getTitle();
+                String prefix = permission + " status: ";
+                if (!title.startsWith(prefix)) return;
+                String actual = title.substring(prefix.length());
+                assertEquals(expected, actual);
+                callbackHelper.notifyCalled();
+            }
+        };
+
+        Tab tab = getActivity().getActivityTab();
+        tab.addObserver(updateWaiter);
+        runJavaScriptCodeInCurrentTab("navigator.permissions.query({ name: '" + permission
+                + "' }).then(({state}) => { window.document.title = '" + permission
+                + " status: ' + state; });");
+        callbackHelper.waitForCallback(0);
+        tab.removeObserver(updateWaiter);
+        runJavaScriptCodeInCurrentTab("window.document.title = 'banana';");
+    }
+
     /**
      * Simulates clicking a button on an AlertDialog.
      */
     private void clickButton(final AlertDialog dialog, final int button) {
         ThreadUtils.runOnUiThreadBlocking(new Runnable() {
             @Override
             public void run() {
                 dialog.getButton(button).performClick();
             }
         });
     }
 
     /**
+     * Waits for an infobar to have been added since to the last call to this function.
+     */
+    protected void waitForInfoBar() throws Exception {
+        mListener.addInfoBarAnimationFinished("InfoBar not added.");
+    }
+
+    /**
      * Runs a permission prompt test that grants the permission and expects the page title to be
      * updated in response.
      * @param updateWaiter  The update waiter to wait for callbacks. Should be added as an observer
      *                      to the current tab prior to calling this method.
      * @param javascript    The JS function to run in the current tab to execute the test and update
      *                      the page title.
      * @param nUpdates      How many updates of the page title to wait for.
      * @param withGeature   True if we require a user gesture to trigger the prompt.
      * @param isDialog      True if we are expecting a permission dialog, false for an infobar.
      * @param hasSwitch     True if we are expecting a persistence switch, false otherwise.
      * @param toggleSwitch  True if we should toggle the switch off, false otherwise.
      * @throws Exception
      */
     protected void runAllowTest(PermissionUpdateWaiter updateWaiter, final String url,
             String javascript, int nUpdates, boolean withGesture, boolean isDialog,
             boolean hasSwitch, boolean toggleSwitch) throws Exception {
-        final String test_url = mTestServer.getURL(url);
-        loadUrl(test_url);
+        setUpUrl(url);
 
         if (withGesture) {
             runJavaScriptCodeInCurrentTab("functionToRun = '" + javascript + "'");
             singleClickView(getActivity().getActivityTab().getView());
         } else {
             runJavaScriptCodeInCurrentTab(javascript);
         }
 
         if (isDialog) {
             DialogShownCriteria criteria = new DialogShownCriteria("Dialog not shown");
             CriteriaHelper.pollUiThread(criteria);
-            replyToDialogAndWaitForUpdates(
-                    updateWaiter, criteria.getDialog(), nUpdates, true, hasSwitch, toggleSwitch);
+            replyToDialog(criteria.getDialog(), true, hasSwitch, toggleSwitch);
         } else {
-            replyToInfoBarAndWaitForUpdates(updateWaiter, nUpdates, true, hasSwitch, toggleSwitch);
+            waitForInfoBar();
+            replyToInfoBar(true, hasSwitch, toggleSwitch);
         }
+        updateWaiter.waitForNumUpdates(nUpdates);
     }
 
     /**
      * Replies to an infobar permission prompt, optionally checking for the presence of a
-     * persistence switch and toggling it. Waits for a provided number of updates to the page title
-     * in response.
+     * persistence switch and toggling it.
      */
-    private void replyToInfoBarAndWaitForUpdates(PermissionUpdateWaiter updateWaiter, int nUpdates,
-            boolean allow, boolean hasSwitch, boolean toggleSwitch) throws Exception {
-        mListener.addInfoBarAnimationFinished("InfoBar not added.");
+    protected void replyToInfoBar(boolean allow, boolean hasSwitch, boolean toggleSwitch)
+            throws Exception {
         InfoBar infobar = getInfoBars().get(0);
         assertNotNull(infobar);
 
         if (hasSwitch) {
             SwitchCompat persistSwitch = (SwitchCompat) infobar.getView().findViewById(
                     R.id.permission_infobar_persist_toggle);
             checkAndToggleSwitch(persistSwitch, toggleSwitch);
         }
 
         if (allow) {
             assertTrue("Allow button wasn't found", InfoBarUtil.clickPrimaryButton(infobar));
         } else {
             assertTrue("Block button wasn't found", InfoBarUtil.clickSecondaryButton(infobar));
         }
-        updateWaiter.waitForNumUpdates(nUpdates);
     }
 
     /**
      * Replies to a dialog permission prompt, optionally checking for the presence of a
-     * persistence switch and toggling it. Waits for a provided number of updates to the page title
-     * in response.
+     * persistence switch and toggling it.
      */
-    private void replyToDialogAndWaitForUpdates(PermissionUpdateWaiter updateWaiter,
-            AlertDialog dialog, int nUpdates, boolean allow, boolean hasSwitch,
+    private void replyToDialog(AlertDialog dialog, boolean allow, boolean hasSwitch,
             boolean toggleSwitch) throws Exception {
         if (hasSwitch) {
             SwitchCompat persistSwitch =
                     (SwitchCompat) dialog.findViewById(R.id.permission_dialog_persist_toggle);
             checkAndToggleSwitch(persistSwitch, toggleSwitch);
         }
 
         if (allow) {
             clickButton(dialog, DialogInterface.BUTTON_POSITIVE);
         } else {
             clickButton(dialog, DialogInterface.BUTTON_NEGATIVE);
         }
-        updateWaiter.waitForNumUpdates(nUpdates);
     }
 
     private void checkAndToggleSwitch(final SwitchCompat persistSwitch, boolean toggleSwitch) {
         assertNotNull(persistSwitch);
         assertTrue(persistSwitch.isChecked());
         if (toggleSwitch) {
             ThreadUtils.runOnUiThreadBlocking(new Runnable() {
                 @Override
                 public void run() {
                     persistSwitch.toggle();
                 }
             });
         }
     }
 
     @Override
     public void startMainActivity() throws InterruptedException {
         startMainActivityOnBlankPage();
     }
 }