| // Copyright 2013 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. |
| |
| #include "base/memory/ptr_util.h" |
| #include "base/values.h" |
| #include "build/build_config.h" |
| #include "chrome/browser/ui/tabs/tab_strip_model.h" |
| #include "chrome/browser/ui/webui/options/options_ui_browsertest.h" |
| #include "components/policy/core/browser/browser_policy_connector.h" |
| #include "components/policy/core/common/external_data_fetcher.h" |
| #include "components/policy/core/common/mock_configuration_policy_provider.h" |
| #include "components/policy/core/common/policy_map.h" |
| #include "components/policy/core/common/policy_types.h" |
| #include "content/public/browser/render_frame_host.h" |
| #include "content/public/browser/web_contents.h" |
| #include "content/public/test/browser_test_utils.h" |
| #include "content/public/test/test_utils.h" |
| #include "policy/policy_constants.h" |
| #include "testing/gmock/include/gmock/gmock.h" |
| #include "testing/gtest/include/gtest/gtest.h" |
| |
| #if defined(OS_CHROMEOS) |
| #include "chrome/browser/chromeos/policy/device_policy_cros_browser_test.h" |
| #include "chromeos/network/onc/onc_test_utils.h" |
| #endif |
| |
| using testing::Return; |
| using testing::_; |
| |
| class CertificateManagerBrowserTest : public options::OptionsUIBrowserTest { |
| public: |
| CertificateManagerBrowserTest() {} |
| ~CertificateManagerBrowserTest() override {} |
| |
| protected: |
| void SetUpInProcessBrowserTestFixture() override { |
| #if defined(OS_CHROMEOS) |
| device_policy_test_helper_.MarkAsEnterpriseOwned(); |
| #endif |
| // Setup the policy provider for injecting certs through ONC policy. |
| EXPECT_CALL(provider_, IsInitializationComplete(_)) |
| .WillRepeatedly(Return(true)); |
| policy::BrowserPolicyConnector::SetPolicyProviderForTesting(&provider_); |
| } |
| |
| #if defined(OS_CHROMEOS) |
| void LoadONCPolicy(const std::string& filename) { |
| const std::string& user_policy_blob = |
| chromeos::onc::test_utils::ReadTestData(filename); |
| policy::PolicyMap policy; |
| policy.Set( |
| policy::key::kOpenNetworkConfiguration, policy::POLICY_LEVEL_MANDATORY, |
| policy::POLICY_SCOPE_USER, policy::POLICY_SOURCE_CLOUD, |
| base::WrapUnique(new base::StringValue(user_policy_blob)), nullptr); |
| provider_.UpdateChromePolicy(policy); |
| content::RunAllPendingInMessageLoop(); |
| } |
| #endif |
| |
| void ClickElement(const std::string& selector) { |
| EXPECT_TRUE(content::ExecuteScript( |
| GetSettingsFrame(), |
| "document.querySelector(\"" + selector + "\").click()")); |
| } |
| |
| bool HasElement(const std::string& selector) { |
| bool result; |
| EXPECT_TRUE(content::ExecuteScriptAndExtractBool( |
| GetSettingsFrame(), |
| "window.domAutomationController.send(" |
| " !!document.querySelector('" + selector + "'));", |
| &result)); |
| return result; |
| } |
| |
| policy::MockConfigurationPolicyProvider provider_; |
| #if defined(OS_CHROMEOS) |
| policy::DevicePolicyCrosTestHelper device_policy_test_helper_; |
| #endif |
| }; |
| |
| #if defined(OS_CHROMEOS) |
| // Ensure policy-installed certificates without web trust do not display |
| // the managed setting indicator (only on Chrome OS). |
| IN_PROC_BROWSER_TEST_F(CertificateManagerBrowserTest, |
| PolicyCertificateWithoutWebTrustHasNoIndicator) { |
| LoadONCPolicy("certificate-authority.onc"); |
| NavigateToSettings(); |
| ClickElement("#certificatesManageButton"); |
| ClickElement("#ca-certs-nav-tab"); |
| EXPECT_FALSE(HasElement(".cert-policy")); |
| } |
| #endif |
| |
| #if defined(OS_CHROMEOS) |
| // Ensure policy-installed certificates with web trust display the |
| // managed setting indicator (only on Chrome OS). |
| IN_PROC_BROWSER_TEST_F(CertificateManagerBrowserTest, |
| PolicyCertificateWithWebTrustHasIndicator) { |
| LoadONCPolicy("certificate-web-authority.onc"); |
| NavigateToSettings(); |
| ClickElement("#certificatesManageButton"); |
| ClickElement("#ca-certs-nav-tab"); |
| EXPECT_TRUE(HasElement(".cert-policy")); |
| } |
| #endif |