src/chrome/browser/ui/webui/options/certificate_manager_browsertest.cc

   1 // Copyright 2013 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "base/callback.h"
   6 #include "base/values.h"
   7 #include "chrome/browser/browser_process.h"
   8 #include "chrome/browser/policy/browser_policy_connector.h"
   9 #include "chrome/browser/policy/external_data_fetcher.h"
  10 #include "chrome/browser/policy/mock_configuration_policy_provider.h"
  11 #include "chrome/browser/policy/policy_map.h"
  12 #include "chrome/browser/policy/policy_types.h"
  13 #include "chrome/browser/ui/browser.h"
  14 #include "chrome/browser/ui/tabs/tab_strip_model.h"
  15 #include "chrome/browser/ui/webui/options/options_ui_browsertest.h"
  16 #include "content/public/browser/web_contents.h"
  17 #include "content/public/test/browser_test_utils.h"
  18 #include "content/public/test/test_utils.h"
  19 #include "policy/policy_constants.h"
  20 #include "testing/gmock/include/gmock/gmock.h"
  21 #include "testing/gtest/include/gtest/gtest.h"
  22
  23 #if defined(OS_CHROMEOS)
  24 #include "chrome/browser/chromeos/policy/device_policy_cros_browser_test.h"
  25 #include "chrome/browser/chromeos/policy/network_configuration_updater.h"
  26 #include "chromeos/network/onc/onc_test_utils.h"
  27 #include "crypto/nss_util.h"
  28 #endif
  29
  30 using testing::AnyNumber;
  31 using testing::Return;
  32 using testing::_;
  33
  34 class CertificateManagerBrowserTest : public options::OptionsUIBrowserTest {
  35  public:
  36   CertificateManagerBrowserTest() {}
  37   virtual ~CertificateManagerBrowserTest() {}
  38
  39  protected:
  40   virtual void SetUpInProcessBrowserTestFixture() OVERRIDE {
  41 #if defined(OS_CHROMEOS)
  42     device_policy_test_helper_.MarkAsEnterpriseOwned();
  43 #endif
  44     // Setup the policy provider for injecting certs through ONC policy.
  45     EXPECT_CALL(provider_, IsInitializationComplete(_))
  46         .WillRepeatedly(Return(true));
  47     EXPECT_CALL(provider_, RegisterPolicyDomain(_)).Times(AnyNumber());
  48     policy::BrowserPolicyConnector::SetPolicyProviderForTesting(&provider_);
  49   }
  50
  51   virtual void SetUpOnMainThread() OVERRIDE {
  52     content::RunAllPendingInMessageLoop();
  53   }
  54
  55 #if defined(OS_CHROMEOS)
  56   void LoadONCPolicy(const std::string& filename) {
  57     const std::string& user_policy_blob =
  58         chromeos::onc::test_utils::ReadTestData(filename);
  59     policy::PolicyMap policy;
  60     policy.Set(policy::key::kOpenNetworkConfiguration,
  61                policy::POLICY_LEVEL_MANDATORY,
  62                policy::POLICY_SCOPE_USER,
  63                Value::CreateStringValue(user_policy_blob),
  64                NULL);
  65     provider_.UpdateChromePolicy(policy);
  66     content::RunAllPendingInMessageLoop();
  67   }
  68 #endif
  69
  70   void ClickElement(const std::string& selector) {
  71     EXPECT_TRUE(content::ExecuteScriptInFrame(
  72         browser()->tab_strip_model()->GetActiveWebContents(),
  73         "//div[@id='settings']/iframe",
  74         "document.querySelector(\"" + selector + "\").click()"));
  75   }
  76
  77   bool HasElement(const std::string& selector) {
  78     bool result;
  79     EXPECT_TRUE(content::ExecuteScriptInFrameAndExtractBool(
  80         browser()->tab_strip_model()->GetActiveWebContents(),
  81         "//div[@id='settings']/iframe",
  82         "window.domAutomationController.send("
  83         "    !!document.querySelector('" + selector + "'));",
  84         &result));
  85     return result;
  86   }
  87
  88   policy::MockConfigurationPolicyProvider provider_;
  89 #if defined(OS_CHROMEOS)
  90   policy::DevicePolicyCrosTestHelper device_policy_test_helper_;
  91   crypto::ScopedTestNSSDB test_nssdb_;
  92 #endif
  93 };
  94
  95 #if defined(OS_CHROMEOS)
  96 // Ensure policy-installed certificates without web trust do not display
  97 // the managed setting indicator (only on Chrome OS).
  98 IN_PROC_BROWSER_TEST_F(CertificateManagerBrowserTest,
  99                        PolicyCertificateWithoutWebTrustHasNoIndicator) {
 100   LoadONCPolicy("certificate-authority.onc");
 101   NavigateToSettings();
 102   ClickElement("#certificatesManageButton");
 103   ClickElement("#ca-certs-nav-tab");
 104   EXPECT_FALSE(HasElement(".cert-policy"));
 105 }
 106 #endif
 107
 108 #if defined(OS_CHROMEOS)
 109 // Ensure policy-installed certificates with web trust display the
 110 // managed setting indicator (only on Chrome OS).
 111 IN_PROC_BROWSER_TEST_F(CertificateManagerBrowserTest,
 112                        PolicyCertificateWithWebTrustHasIndicator) {
 113   LoadONCPolicy("certificate-web-authority.onc");
 114   NavigateToSettings();
 115   ClickElement("#certificatesManageButton");
 116   ClickElement("#ca-certs-nav-tab");
 117   EXPECT_TRUE(HasElement(".cert-policy"));
 118 }
 119 #endif