| // 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. |
| |
| #include "base/strings/stringprintf.h" |
| #include "chrome/browser/browser_process.h" |
| #include "chrome/browser/chromeos/policy/affiliation_test_helper.h" |
| #include "chrome/browser/chromeos/policy/browser_policy_connector_chromeos.h" |
| #include "chrome/browser/chromeos/policy/device_policy_cros_browser_test.h" |
| #include "chrome/browser/chromeos/policy/stub_enterprise_install_attributes.h" |
| #include "chrome/browser/extensions/extension_apitest.h" |
| #include "chrome/browser/net/url_request_mock_util.h" |
| #include "chrome/test/base/ui_test_utils.h" |
| #include "chromeos/dbus/fake_session_manager_client.h" |
| #include "chromeos/login/user_names.h" |
| #include "components/policy/core/common/mock_configuration_policy_provider.h" |
| #include "components/prefs/pref_service.h" |
| #include "components/signin/core/account_id/account_id.h" |
| #include "components/user_manager/user_manager.h" |
| #include "content/public/browser/notification_service.h" |
| #include "content/public/test/test_utils.h" |
| #include "extensions/browser/api_test_utils.h" |
| #include "extensions/browser/extension_registry.h" |
| #include "extensions/browser/test_extension_registry_observer.h" |
| #include "extensions/test/result_catcher.h" |
| #include "net/test/url_request/url_request_mock_http_job.h" |
| #include "policy/policy_constants.h" |
| |
| namespace { |
| |
| const char kDeviceId[] = "device_id"; |
| const base::FilePath::CharType kTestExtensionDir[] = |
| FILE_PATH_LITERAL("extensions/api_test/enterprise_device_attributes"); |
| const base::FilePath::CharType kUpdateManifestFileName[] = |
| FILE_PATH_LITERAL("update_manifest.xml"); |
| |
| const char kAffiliatedUserEmail[] = "user@example.com"; |
| const char kAffiliationID[] = "some-affiliation-id"; |
| const char kAnotherAffiliationID[] = "another-affiliation-id"; |
| |
| // The managed_storage extension has a key defined in its manifest, so that |
| // its extension ID is well-known and the policy system can push policies for |
| // the extension. |
| const char kTestExtensionID[] = "nbiliclbejdndfpchgkbmfoppjplbdok"; |
| |
| struct Params { |
| explicit Params(bool affiliated) : affiliated_(affiliated) {} |
| bool affiliated_; |
| }; |
| |
| } // namespace |
| |
| namespace extensions { |
| |
| class EnterpriseDeviceAttributesTest : |
| public ExtensionApiTest, |
| public ::testing::WithParamInterface<Params> { |
| public: |
| EnterpriseDeviceAttributesTest() { |
| set_exit_when_last_browser_closes(false); |
| set_chromeos_user_ = false; |
| } |
| |
| protected: |
| // ExtensionApiTest |
| void SetUpCommandLine(base::CommandLine* command_line) override { |
| ExtensionApiTest::SetUpCommandLine(command_line); |
| policy::affiliation_test_helper:: |
| AppendCommandLineSwitchesForLoginManager(command_line); |
| } |
| |
| void SetUpInProcessBrowserTestFixture() override { |
| ExtensionApiTest::SetUpInProcessBrowserTestFixture(); |
| |
| chromeos::FakeSessionManagerClient* fake_session_manager_client = |
| new chromeos::FakeSessionManagerClient; |
| chromeos::DBusThreadManager::GetSetterForTesting()->SetSessionManagerClient( |
| std::unique_ptr<chromeos::SessionManagerClient>( |
| fake_session_manager_client)); |
| |
| std::set<std::string> device_affiliation_ids; |
| device_affiliation_ids.insert(kAffiliationID); |
| policy::affiliation_test_helper::SetDeviceAffiliationID( |
| &test_helper_, fake_session_manager_client, device_affiliation_ids); |
| |
| std::set<std::string> user_affiliation_ids; |
| if (GetParam().affiliated_) { |
| user_affiliation_ids.insert(kAffiliationID); |
| } else { |
| user_affiliation_ids.insert(kAnotherAffiliationID); |
| } |
| policy::UserPolicyBuilder user_policy; |
| policy::affiliation_test_helper::SetUserAffiliationIDs( |
| &user_policy, fake_session_manager_client, |
| affiliated_account_id_.GetUserEmail(), user_affiliation_ids); |
| |
| // Set up fake install attributes. |
| std::unique_ptr<policy::StubEnterpriseInstallAttributes> attributes( |
| new policy::StubEnterpriseInstallAttributes()); |
| |
| attributes->SetRegistrationUser(affiliated_account_id_.GetUserEmail()); |
| policy::BrowserPolicyConnectorChromeOS::SetInstallAttributesForTesting( |
| attributes.release()); |
| |
| test_helper_.InstallOwnerKey(); |
| // Init the device policy. |
| policy::DevicePolicyBuilder* device_policy = test_helper_.device_policy(); |
| device_policy->SetDefaultSigningKey(); |
| device_policy->policy_data().set_directory_api_id(kDeviceId); |
| device_policy->Build(); |
| |
| fake_session_manager_client->set_device_policy(device_policy->GetBlob()); |
| fake_session_manager_client->OnPropertyChangeComplete(true); |
| |
| // Init the user policy provider. |
| EXPECT_CALL(policy_provider_, IsInitializationComplete(testing::_)) |
| .WillRepeatedly(testing::Return(true)); |
| policy_provider_.SetAutoRefresh(); |
| policy::BrowserPolicyConnector::SetPolicyProviderForTesting( |
| &policy_provider_); |
| } |
| |
| void SetUpOnMainThread() override { |
| const base::ListValue* users = |
| g_browser_process->local_state()->GetList("LoggedInUsers"); |
| if (!users->empty()) { |
| policy::affiliation_test_helper::LoginUser( |
| affiliated_account_id_.GetUserEmail()); |
| } |
| |
| ExtensionApiTest::SetUpOnMainThread(); |
| } |
| |
| void SetPolicy() { |
| // Extensions that are force-installed come from an update URL, which |
| // defaults to the webstore. Use a mock URL for this test with an update |
| // manifest that includes the crx file of the test extension. |
| base::FilePath update_manifest_path = |
| base::FilePath(kTestExtensionDir).Append(kUpdateManifestFileName); |
| GURL update_manifest_url(net::URLRequestMockHTTPJob::GetMockUrl( |
| update_manifest_path.MaybeAsASCII())); |
| |
| std::unique_ptr<base::ListValue> forcelist(new base::ListValue); |
| forcelist->AppendString(base::StringPrintf( |
| "%s;%s", kTestExtensionID, update_manifest_url.spec().c_str())); |
| |
| policy::PolicyMap policy; |
| policy.Set(policy::key::kExtensionInstallForcelist, |
| policy::POLICY_LEVEL_MANDATORY, policy::POLICY_SCOPE_MACHINE, |
| policy::POLICY_SOURCE_CLOUD, std::move(forcelist), nullptr); |
| |
| // Set the policy and wait until the extension is installed. |
| extensions::TestExtensionRegistryObserver observer( |
| ExtensionRegistry::Get(profile())); |
| policy_provider_.UpdateChromePolicy(policy); |
| observer.WaitForExtensionLoaded(); |
| } |
| |
| // Load |page_url| in |browser| and wait for PASSED or FAILED notification. |
| // The functionality of this function is reduced functionality of |
| // RunExtensionSubtest(), but we don't use it here because it requires |
| // function InProcessBrowserTest::browser() to return non-NULL pointer. |
| // Unfortunately it returns the value which is set in constructor and can't be |
| // modified. Because on login flow there is no browser, the function |
| // InProcessBrowserTest::browser() always returns NULL. Besides this we need |
| // only very little functionality from RunExtensionSubtest(). Thus so that |
| // don't make RunExtensionSubtest() to complex we just introduce a new |
| // function. |
| bool TestExtension(Browser* browser, const std::string& page_url) { |
| DCHECK(!page_url.empty()) << "page_url cannot be empty"; |
| |
| extensions::ResultCatcher catcher; |
| ui_test_utils::NavigateToURL(browser, GURL(page_url)); |
| |
| if (!catcher.GetNextResult()) { |
| message_ = catcher.message(); |
| return false; |
| } |
| return true; |
| } |
| |
| const AccountId affiliated_account_id_ = |
| AccountId::FromUserEmail(kAffiliatedUserEmail); |
| |
| private: |
| policy::MockConfigurationPolicyProvider policy_provider_; |
| policy::DevicePolicyCrosTestHelper test_helper_; |
| }; |
| |
| IN_PROC_BROWSER_TEST_P(EnterpriseDeviceAttributesTest, PRE_Success) { |
| policy::affiliation_test_helper::PreLoginUser( |
| affiliated_account_id_.GetUserEmail()); |
| } |
| |
| IN_PROC_BROWSER_TEST_P(EnterpriseDeviceAttributesTest, Success) { |
| content::BrowserThread::PostTask( |
| content::BrowserThread::IO, FROM_HERE, |
| base::Bind(chrome_browser_net::SetUrlRequestMocksEnabled, true)); |
| |
| SetPolicy(); |
| |
| EXPECT_EQ(GetParam().affiliated_, user_manager::UserManager::Get()-> |
| FindUser(affiliated_account_id_)->IsAffiliated()); |
| |
| // Device ID is available only for affiliated user. |
| std::string device_id = GetParam().affiliated_ ? kDeviceId : ""; |
| |
| // Pass the expected value (device_id) to test. |
| ASSERT_TRUE(TestExtension(CreateBrowser(profile()), |
| base::StringPrintf("chrome-extension://%s/basic.html?%s", |
| kTestExtensionID, device_id.c_str()))) |
| << message_; |
| } |
| |
| // Ensure that extensions that are not pre-installed by policy throw an install |
| // warning if they request the enterprise.deviceAttributes permission in the |
| // manifest and that such extensions don't see the |
| // chrome.enterprise.deviceAttributes namespace. |
| IN_PROC_BROWSER_TEST_F( |
| ExtensionApiTest, |
| EnterpriseDeviceAttributesIsRestrictedToPolicyExtension) { |
| ASSERT_TRUE(RunExtensionSubtest("enterprise_device_attributes", |
| "api_not_available.html", |
| kFlagIgnoreManifestWarnings)); |
| |
| base::FilePath extension_path = |
| test_data_dir_.AppendASCII("enterprise_device_attributes"); |
| extensions::ExtensionRegistry* registry = |
| extensions::ExtensionRegistry::Get(profile()); |
| const extensions::Extension* extension = |
| GetExtensionByPath(registry->enabled_extensions(), extension_path); |
| ASSERT_FALSE(extension->install_warnings().empty()); |
| EXPECT_EQ( |
| "'enterprise.deviceAttributes' is not allowed for specified install " |
| "location.", |
| extension->install_warnings()[0].message); |
| } |
| |
| // Both cases of affiliated and non-affiliated on the device user are tested. |
| INSTANTIATE_TEST_CASE_P(AffiliationCheck, |
| EnterpriseDeviceAttributesTest, |
| ::testing::Values(Params(true), Params(false))); |
| } // namespace extensions |