Simple Android integration tests for PaymentRequest UI

chrome/android/javatests/src/org/chromium/chrome/browser/payments/PaymentRequestTestBase.java

Index: chrome/android/javatests/src/org/chromium/chrome/browser/payments/PaymentRequestTestBase.java
diff --git a/chrome/android/javatests/src/org/chromium/chrome/browser/payments/PaymentRequestTestBase.java b/chrome/android/javatests/src/org/chromium/chrome/browser/payments/PaymentRequestTestBase.java
new file mode 100644
index 0000000000000000000000000000000000000000..cad06f58b1129f999a5931474b066d7b93834a3b
--- /dev/null
+++ b/chrome/android/javatests/src/org/chromium/chrome/browser/payments/PaymentRequestTestBase.java
@@ -0,0 +1,140 @@
+// 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.payments;
+
+import android.view.View;
+
+import org.chromium.base.ThreadUtils;
+import org.chromium.base.test.util.CommandLineFlags;
+import org.chromium.base.test.util.UrlUtils;
+import org.chromium.chrome.R;
+import org.chromium.chrome.browser.ChromeActivity;
+import org.chromium.chrome.browser.ChromeSwitches;
+import org.chromium.chrome.browser.payments.ui.PaymentRequestUI;
+import org.chromium.chrome.test.ChromeActivityTestCaseBase;
+import org.chromium.content.browser.ContentViewCore;
+import org.chromium.content.browser.test.util.Criteria;
+import org.chromium.content.browser.test.util.CriteriaHelper;
+import org.chromium.content.browser.test.util.DOMUtils;
+import org.chromium.content_public.browser.WebContents;
+
+import java.util.concurrent.ExecutionException;
+import java.util.concurrent.TimeoutException;
+import java.util.concurrent.atomic.AtomicReference;
+
+/**
+ * A base integration test for payments.
+ */
+@CommandLineFlags.Add({ChromeSwitches.EXPERIMENTAL_WEB_PLAFTORM_FEATURES})
+abstract class PaymentRequestTestBase extends ChromeActivityTestCaseBase<ChromeActivity> {
+    private final AtomicReference<ContentViewCore> mViewCoreRef =
+            new AtomicReference<ContentViewCore>();
+    private final AtomicReference<WebContents> mWebContentsRef = new AtomicReference<WebContents>();
+    private final String mTestFilePath;
+
+    protected PaymentRequestTestBase(String testFileName) {
+        super(ChromeActivity.class);
+        mTestFilePath = UrlUtils.getIsolatedTestFilePath(
+                String.format("chrome/test/data/android/payments/%s", testFileName));
+    }
+
+    @Override
+    public void startMainActivity() throws InterruptedException {}
+
+    protected abstract void onMainActivityStarted()
+            throws InterruptedException, ExecutionException, TimeoutException;
+
+    protected void triggerPaymentUI()
+            throws InterruptedException, ExecutionException, TimeoutException {
+        startMainActivityWithURL(mTestFilePath);
+        onMainActivityStarted();
+        ThreadUtils.runOnUiThreadBlocking(new Runnable() {
+            @Override
+            public void run() {
+                mViewCoreRef.set(getActivity().getCurrentContentViewCore());
+                mWebContentsRef.set(mViewCoreRef.get().getWebContents());
+            }
+        });
+        assertWaitForPageScaleFactorMatch(1);
+        DOMUtils.waitForNonZeroNodeBounds(mWebContentsRef.get(), "buy");
+        DOMUtils.clickNode(this, mViewCoreRef.get(), "buy");
+    }
+
+    protected void clickClosePaymentUIButton() throws InterruptedException {
+        clickPaymentUIButton("close", R.id.close_button);
+    }
+
+    protected void clickSecondaryPaymentUIButton() throws InterruptedException {
+        clickPaymentUIButton("secondary", R.id.button_secondary);
+    }
+
+    private void clickPaymentUIButton(final String buttonName, final int resourceId)
+            throws InterruptedException {
+        CriteriaHelper.pollUiThread(new Criteria() {
+            @Override
+            public boolean isSatisfied() {
+                PaymentRequestUI ui = PaymentRequestUI.getCurrentUIForTest();
+                if (ui == null) {
+                    updateFailureReason("Payment UI was not shown");
+                    return false;
+                }
+
+                View button = ui.getDialogForTest().findViewById(resourceId);
+                if (button == null) {
+                    updateFailureReason(
+                            String.format("Cannot find the %s button on payment UI", buttonName));
+                    return false;
+                }
+
+                if (!button.isEnabled()) {
+                    updateFailureReason(
+                            String.format("The %s button on payment UI is disabled", buttonName));
+                    return false;
+                }
+
+                button.performClick();
+                return true;
+            }
+        });
+    }
+
+    protected void waitForPaymentUIHidden() throws InterruptedException {
+        CriteriaHelper.pollUiThread(new Criteria("Payment UI should be hidden") {
+            @Override
+            public boolean isSatisfied() {
+                return PaymentRequestUI.getCurrentUIForTest() == null;
+            }
+        });
+    }
+
+    protected void expectResultContains(final String[] contents) throws InterruptedException {
+        CriteriaHelper.pollInstrumentationThread(new Criteria() {
+            @Override
+            public boolean isSatisfied() {
+                try {
+                    String result = DOMUtils.getNodeContents(mWebContentsRef.get(), "result");
+                    if (result == null) {
+                        updateFailureReason("Cannot find 'result' node on test page");
+                        return false;
+                    }
+                    for (int i = 0; i < contents.length; i++) {
+                        if (!result.contains(contents[i])) {
+                            updateFailureReason(
+                                    String.format("Result should contain '%s'", contents[i]));
+                            return false;
+                        }
+                    }
+                    return true;
+                } catch (InterruptedException e1) {
+                    updateFailureReason(e1.getMessage());
+                    return false;
+                } catch (TimeoutException e2) {
+                    updateFailureReason(e2.getMessage());
+                    return false;
+                }
+            }
+        });
+    }
+}