[Storage] Android - ManageSpace UI, Important Origins, and CBD Dialog

chrome/android/javatests/src/org/chromium/chrome/browser/preferences/website/ManageSpaceActivityTest.java

Index: chrome/android/javatests/src/org/chromium/chrome/browser/preferences/website/ManageSpaceActivityTest.java
diff --git a/chrome/android/javatests/src/org/chromium/chrome/browser/preferences/website/ManageSpaceActivityTest.java b/chrome/android/javatests/src/org/chromium/chrome/browser/preferences/website/ManageSpaceActivityTest.java
new file mode 100644
index 0000000000000000000000000000000000000000..20d5bc64799ee7c3d34f28dd3877ed4a22c3b512
--- /dev/null
+++ b/chrome/android/javatests/src/org/chromium/chrome/browser/preferences/website/ManageSpaceActivityTest.java
@@ -0,0 +1,157 @@
+// 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.preferences.website;
+
+import android.annotation.TargetApi;
+import android.content.Intent;
+import android.os.Build;
+import android.os.Environment;
+import android.support.v7.app.AlertDialog;
+import android.test.suitebuilder.annotation.MediumTest;
+import android.test.suitebuilder.annotation.SmallTest;
+
+import org.chromium.base.ThreadUtils;
+import org.chromium.base.test.util.CommandLineFlags;
+import org.chromium.chrome.browser.ChromeActivity;
+import org.chromium.chrome.browser.preferences.PrefServiceBridge;
+import org.chromium.chrome.test.ChromeActivityTestCaseBase;
+import org.chromium.content.browser.test.util.Criteria;
+import org.chromium.content.browser.test.util.CriteriaHelper;
+import org.chromium.net.test.EmbeddedTestServer;
+
+/**
+ * Tests for ManageSpaceActivity.
+ */
+@TargetApi(Build.VERSION_CODES.KITKAT)
+@CommandLineFlags.Add({"enable-site-engagement"})
+public class ManageSpaceActivityTest extends ChromeActivityTestCaseBase<ChromeActivity> {
+    private EmbeddedTestServer mTestServer;
+
+    public ManageSpaceActivityTest() {
+        super(ChromeActivity.class);
+    }
+
+    @Override
+    protected void setUp() throws Exception {
+        super.setUp();
+        mTestServer = EmbeddedTestServer.createAndStartFileServer(
+                getInstrumentation().getContext(), Environment.getExternalStorageDirectory());
+    }
+
+    @Override
+    protected void tearDown() throws Exception {
+        mTestServer.stopAndDestroyServer();
+        super.tearDown();
+    }
+
+    @Override
+    public void startMainActivity() throws InterruptedException {
+        if (getName().equals("testClearUnimporantWithoutChromeStart")) {
+            return;
+        }
+        startMainActivityOnBlankPage();
+    }
+
+    private ManageSpaceActivity startManageSpaceActivity() {
+        Intent intent =
+                new Intent(getInstrumentation().getTargetContext(), ManageSpaceActivity.class);
+        intent.addFlags(Intent.FLAG_ACTIVITY_NEW_TASK);
+        intent.addFlags(Intent.FLAG_ACTIVITY_CLEAR_TOP);
+        return (ManageSpaceActivity) getInstrumentation().startActivitySync(intent);
+    }
+
+    public void waitForClearButtonEnabled(final ManageSpaceActivity activity) throws Exception {
+        CriteriaHelper.pollUiThread(new Criteria() {
+            @Override
+            public boolean isSatisfied() {
+                return activity.getClearUnimportantButton().isEnabled();
+            }
+        });
+    }
+
+    public Runnable getClickClearRunnable(final ManageSpaceActivity activity) {
+        return new Runnable() {
+            @Override
+            public void run() {
+                activity.onClick(activity.getClearUnimportantButton());
+            }
+        };
+    }
+
+    public void waitForDialogShowing(final ManageSpaceActivity activity) throws Exception {
+        CriteriaHelper.pollUiThread(new Criteria() {
+            @Override
+            public boolean isSatisfied() {
+                return activity.getUnimportantConfirmDialog().isShowing();
+            }
+        });
+    }
+
+    public Runnable getPressClearRunnable(final AlertDialog dialog) {
+        return new Runnable() {
+            @Override
+            public void run() {
+                dialog.getButton(AlertDialog.BUTTON_POSITIVE).performClick();
+            }
+        };
+    }
+
+    @SmallTest
+    public void testLaunchActivity() {
+        startManageSpaceActivity();
+    }
+
+    @MediumTest
+    public void testClearUnimportantOnly() throws Exception {
+        final String cookiesUrl =
+                mTestServer.getURL("/chrome/test/data/android/storage_persistance.html");
+        final String serverOrigin = mTestServer.getURL("/");
+
+        loadUrl(cookiesUrl + "#clear");
+        assertEquals("false", runJavaScriptCodeInCurrentTab("hasAllStorage()"));
+        runJavaScriptCodeInCurrentTab("setStorage()");
+        assertEquals("true", runJavaScriptCodeInCurrentTab("hasAllStorage()"));
+        loadUrl("about:blank");
+
+        // Now we set the origin as important, and check that we don't clear it.
+        ThreadUtils.runOnUiThreadBlocking(new Runnable() {
+            @Override
+            public void run() {
+                PrefServiceBridge.markOriginAsImportantForTesting(serverOrigin);
+            }
+        });
+
+        ManageSpaceActivity manageSpaceActivity = startManageSpaceActivity();
+        // Click 'clear' in the CBD screen.
+        waitForClearButtonEnabled(manageSpaceActivity);
+        ThreadUtils.runOnUiThreadBlocking(getClickClearRunnable(manageSpaceActivity));
+        // Press 'clear' in our dialog.
+        waitForDialogShowing(manageSpaceActivity);
+        ThreadUtils.runOnUiThreadBlocking(
+                getPressClearRunnable(manageSpaceActivity.getUnimportantConfirmDialog()));
+        waitForClearButtonEnabled(manageSpaceActivity);
+        manageSpaceActivity.finish();
+
+        loadUrl(cookiesUrl);
+        assertEquals("true", runJavaScriptCodeInCurrentTab("hasAllStorage()"));
+    }
+
+    @MediumTest
+    public void testClearUnimporantWithoutChromeStart() throws Exception {
+        ManageSpaceActivity manageSpaceActivity = startManageSpaceActivity();
+        // Click 'clear' in the CBD screen.
+        waitForClearButtonEnabled(manageSpaceActivity);
+        ThreadUtils.runOnUiThreadBlocking(getClickClearRunnable(manageSpaceActivity));
+        // Press 'clear' in our dialog.
+        waitForDialogShowing(manageSpaceActivity);
+        ThreadUtils.runOnUiThreadBlocking(
+                getPressClearRunnable(manageSpaceActivity.getUnimportantConfirmDialog()));
+        waitForClearButtonEnabled(manageSpaceActivity);
+        manageSpaceActivity.finish();
+    }
+
+    // TODO(dmurph): Test the other buttons. One should go to the site storage list, and the other
+    //               should reset all app data.
+}