Adding <keygen> Content Setting (Android UI)

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

 // Copyright 2015 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.content.Intent;
 import android.os.Bundle;
 import android.test.suitebuilder.annotation.SmallTest;
 
 import org.chromium.base.ThreadUtils;
 import org.chromium.base.test.util.CommandLineFlags;
 import org.chromium.base.test.util.Feature;
 import org.chromium.chrome.browser.ChromeActivity;
 import org.chromium.chrome.browser.ChromeApplication;
 import org.chromium.chrome.browser.ChromeSwitches;
 import org.chromium.chrome.browser.infobar.InfoBarContainer;
 import org.chromium.chrome.browser.preferences.ChromeBaseCheckBoxPreference;
+import org.chromium.chrome.browser.preferences.ChromeBaseListPreference;
 import org.chromium.chrome.browser.preferences.ChromeSwitchPreference;
 import org.chromium.chrome.browser.preferences.LocationSettings;
 import org.chromium.chrome.browser.preferences.PrefServiceBridge;
 import org.chromium.chrome.browser.preferences.Preferences;
 import org.chromium.chrome.browser.preferences.PreferencesLauncher;
 import org.chromium.chrome.browser.preferences.PreferencesTest;
 import org.chromium.chrome.browser.util.FeatureUtilities;
 import org.chromium.chrome.test.ChromeActivityTestCaseBase;
 import org.chromium.chrome.test.util.ChromeTabUtils;
 import org.chromium.chrome.test.util.InfoBarTestAnimationListener;
@@ skipping 87 matching lines @@
 
     private Preferences startSiteSettingsCategory(String category) {
         Bundle fragmentArgs = new Bundle();
         fragmentArgs.putString(SingleCategoryPreferences.EXTRA_CATEGORY, category);
         Intent intent = PreferencesLauncher.createIntentForSettingsPage(
                 getInstrumentation().getTargetContext(), SingleCategoryPreferences.class.getName());
         intent.putExtra(Preferences.EXTRA_SHOW_FRAGMENT_ARGUMENTS, fragmentArgs);
         return (Preferences) getInstrumentation().startActivitySync(intent);
     }
 
+    private Preferences startSingleWebsitePreferences(Website site) {
+        Bundle fragmentArgs = new Bundle();
+        fragmentArgs.putSerializable(SingleWebsitePreferences.EXTRA_SITE, site);
+        Intent intent = PreferencesLauncher.createIntentForSettingsPage(
+                getInstrumentation().getTargetContext(), SingleWebsitePreferences.class.getName());
+        intent.putExtra(Preferences.EXTRA_SHOW_FRAGMENT_ARGUMENTS, fragmentArgs);
+        return (Preferences) getInstrumentation().startActivitySync(intent);
+    }
+
     private void setCookiesEnabled(final Preferences preferenceActivity, final boolean enabled) {
         ThreadUtils.runOnUiThreadBlocking(new Runnable() {
             @Override
             public void run() {
                 final SingleCategoryPreferences websitePreferences =
                         (SingleCategoryPreferences) preferenceActivity.getFragmentForTest();
                 final ChromeSwitchPreference cookies =
                         (ChromeSwitchPreference) websitePreferences.findPreference(
                                 SingleCategoryPreferences.READ_WRITE_TOGGLE_KEY);
                 final ChromeBaseCheckBoxPreference thirdPartyCookies =
@@ skipping 111 matching lines @@
                     PreferencesTest.clickPreference(languagePreferences, checkbox);
                 }
                 assertEquals("Auto detect encoding should be " + (enabled ? "enabled" : "disabled"),
                         enabled, PrefServiceBridge.getInstance().isAutoDetectEncodingEnabled());
             }
         });
 
         preferenceActivity.finish();
     }
 
+    private void setEnableKeygen(final String origin, final boolean enabled) {
+        Website website = new Website(WebsiteAddress.create(origin));
+        website.setKeygenInfo(new KeygenInfo(origin, origin, false));
+        final Preferences preferenceActivity = startSingleWebsitePreferences(website);
+
+        ThreadUtils.runOnUiThreadBlocking(new Runnable() {
+            @Override
+            public void run() {
+                SingleWebsitePreferences websitePreferences =
+                        (SingleWebsitePreferences) preferenceActivity.getFragmentForTest();
+                ChromeBaseListPreference keygen =
+                        (ChromeBaseListPreference) websitePreferences.findPreference(
+                                SingleWebsitePreferences.PREF_KEYGEN_PERMISSION);
+                websitePreferences.onPreferenceChange(keygen, enabled
+                                ? ContentSetting.ALLOW.toString()
+                                : ContentSetting.BLOCK.toString());
+            }
+        });
+        preferenceActivity.finish();
+    }
+
     /**
      * Tests that disabling cookies turns off the third-party cookie toggle.
      * @throws Exception
      */
     @SmallTest
     @Feature({"Preferences"})
     public void testThirdPartyCookieToggleGetsDisabled() throws Exception {
         Preferences preferenceActivity =
                 startSiteSettingsCategory(SiteSettingsPreferences.COOKIES_KEY);
         setCookiesEnabled(preferenceActivity, true);
@@ skipping 77 matching lines @@
         setEnablePopups(true);
 
         // Test that a popup opens.
         loadUrl(TestHttpServerClient.getUrl("chrome/test/data/android/popup.html"));
         getInstrumentation().waitForIdleSync();
 
         assertEquals(2, getTabCount());
     }
 
     /**
+     * Sets Allow Keygen Enabled to be false and make sure it is set correctly.
+     * @throws Exception
+     */
+    @SmallTest
+    @Feature({"Preferences"})
+    public void testKeygenBlocked() throws Exception {
+        final String origin = "http://example.com/";
+        setEnableKeygen(origin, false);
+
+        ThreadUtils.runOnUiThreadBlocking(new Runnable() {
+            @Override
+            public void run() {
+                Website site = new Website(WebsiteAddress.create(origin));
+                site.setKeygenInfo(new KeygenInfo(origin, origin, false));
+                assertEquals(site.getKeygenPermission(), ContentSetting.BLOCK);
+            }
+        });
+    }
+
+    /**
+     * Sets Allow Keygen Enabled to be true and make sure it is set correctly.
+     * @throws Exception
+     */
+    @SmallTest
+    @Feature({"Preferences"})
+    public void testKeygenNotBlocked() throws Exception {
+        final String origin = "http://example.com/";
+        setEnableKeygen(origin, true);
+
+        ThreadUtils.runOnUiThreadBlocking(new Runnable() {
+            @Override
+            public void run() {
+                Website site = new Website(WebsiteAddress.create(origin));
+                site.setKeygenInfo(new KeygenInfo(origin, origin, false));
+                assertEquals(site.getKeygenPermission(), ContentSetting.ALLOW);
+            }
+        });
+    }
+
+    /**
      * Sets Allow Camera Enabled to be false and make sure it is set correctly.
      * @throws Exception
      */
     @SmallTest
     @Feature({"Preferences"})
     @CommandLineFlags.Add(ChromeSwitches.USE_FAKE_DEVICE_FOR_MEDIA_STREAM)
     public void testCameraBlocked() throws Exception {
         setEnableCamera(false);
 
         // Test that the camera permission doesn't get requested.
@@ skipping 96 matching lines @@
             public Integer call() throws Exception {
                 if (FeatureUtilities.isDocumentMode(getInstrumentation().getTargetContext())) {
                     return ChromeApplication.getDocumentTabModelSelector().getTotalTabCount();
                 } else {
                     return getActivity().getTabModelSelector().getTotalTabCount();
                 }
             }
         });
     }
 }