| // Copyright (c) 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/command_line.h" |
| #include "base/memory/scoped_ptr.h" |
| #include "base/run_loop.h" |
| #include "chrome/browser/browser_process.h" |
| #include "chrome/browser/policy/browser_policy_connector.h" |
| #include "chrome/browser/policy/cloud/test_request_interceptor.h" |
| #include "chrome/browser/profiles/profile.h" |
| #include "chrome/browser/ui/browser.h" |
| #include "chrome/common/chrome_switches.h" |
| #include "chrome/test/base/in_process_browser_test.h" |
| #include "components/policy/core/common/cloud/cloud_policy_client.h" |
| #include "components/policy/core/common/cloud/mock_cloud_policy_client.h" |
| #include "components/policy/core/common/policy_test_utils.h" |
| #include "content/public/browser/browser_thread.h" |
| #include "net/base/net_errors.h" |
| #include "net/url_request/url_request_context_getter.h" |
| #include "policy/proto/device_management_backend.pb.h" |
| #include "testing/gmock/include/gmock/gmock.h" |
| #include "testing/gtest/include/gtest/gtest.h" |
| |
| #if defined(OS_CHROMEOS) |
| #include "chrome/browser/chromeos/policy/user_cloud_policy_manager_chromeos.h" |
| #include "chrome/browser/chromeos/policy/user_cloud_policy_manager_factory_chromeos.h" |
| #else |
| #include "chrome/browser/policy/cloud/user_cloud_policy_manager.h" |
| #include "chrome/browser/policy/cloud/user_cloud_policy_manager_factory.h" |
| #include "chrome/browser/signin/signin_manager.h" |
| #include "chrome/browser/signin/signin_manager_factory.h" |
| #endif |
| |
| using content::BrowserThread; |
| using testing::AnyNumber; |
| using testing::InvokeWithoutArgs; |
| using testing::Mock; |
| using testing::_; |
| |
| namespace em = enterprise_management; |
| |
| namespace policy { |
| |
| // Tests the cloud policy stack using a URLRequestJobFactory::ProtocolHandler |
| // to intercept requests and produce canned responses. |
| class CloudPolicyManagerTest : public InProcessBrowserTest { |
| protected: |
| CloudPolicyManagerTest() {} |
| virtual ~CloudPolicyManagerTest() {} |
| |
| virtual void SetUpInProcessBrowserTestFixture() OVERRIDE { |
| CommandLine* command_line = CommandLine::ForCurrentProcess(); |
| command_line->AppendSwitchASCII(switches::kDeviceManagementUrl, |
| "http://localhost"); |
| } |
| |
| virtual void SetUpOnMainThread() OVERRIDE { |
| ASSERT_TRUE(PolicyServiceIsEmpty(g_browser_process->policy_service())) |
| << "Pre-existing policies in this machine will make this test fail."; |
| |
| interceptor_.reset(new TestRequestInterceptor( |
| "localhost", |
| BrowserThread::GetMessageLoopProxyForThread(BrowserThread::IO))); |
| |
| BrowserPolicyConnector* connector = |
| g_browser_process->browser_policy_connector(); |
| connector->ScheduleServiceInitialization(0); |
| |
| #if !defined(OS_CHROMEOS) |
| // Mock a signed-in user. This is used by the UserCloudPolicyStore to pass |
| // the username to the UserCloudPolicyValidator. |
| SigninManager* signin_manager = |
| SigninManagerFactory::GetForProfile(browser()->profile()); |
| ASSERT_TRUE(signin_manager); |
| signin_manager->SetAuthenticatedUsername("user@example.com"); |
| |
| ASSERT_TRUE(policy_manager()); |
| policy_manager()->Connect( |
| g_browser_process->local_state(), |
| g_browser_process->system_request_context(), |
| UserCloudPolicyManager::CreateCloudPolicyClient( |
| connector->device_management_service(), |
| g_browser_process->system_request_context()).Pass()); |
| #endif |
| } |
| |
| virtual void CleanUpOnMainThread() OVERRIDE { |
| // Verify that all the expected requests were handled. |
| EXPECT_EQ(0u, interceptor_->GetPendingSize()); |
| |
| interceptor_.reset(); |
| } |
| |
| #if defined(OS_CHROMEOS) |
| UserCloudPolicyManagerChromeOS* policy_manager() { |
| return UserCloudPolicyManagerFactoryChromeOS::GetForProfile( |
| browser()->profile()); |
| } |
| #else |
| UserCloudPolicyManager* policy_manager() { |
| return UserCloudPolicyManagerFactory::GetForBrowserContext( |
| browser()->profile()); |
| } |
| #endif // defined(OS_CHROMEOS) |
| |
| // Register the client of the policy_manager() using a bogus auth token, and |
| // returns once the registration gets a result back. |
| void Register() { |
| ASSERT_TRUE(policy_manager()); |
| ASSERT_TRUE(policy_manager()->core()->client()); |
| |
| base::RunLoop run_loop; |
| MockCloudPolicyClientObserver observer; |
| EXPECT_CALL(observer, OnRegistrationStateChanged(_)) |
| .Times(AnyNumber()) |
| .WillRepeatedly(InvokeWithoutArgs(&run_loop, &base::RunLoop::Quit)); |
| EXPECT_CALL(observer, OnClientError(_)) |
| .Times(AnyNumber()) |
| .WillRepeatedly(InvokeWithoutArgs(&run_loop, &base::RunLoop::Quit)); |
| policy_manager()->core()->client()->AddObserver(&observer); |
| |
| // Give a bogus OAuth token to the |policy_manager|. This should make its |
| // CloudPolicyClient fetch the DMToken. |
| em::DeviceRegisterRequest::Type registration_type = |
| #if defined(OS_CHROMEOS) |
| em::DeviceRegisterRequest::USER; |
| #else |
| em::DeviceRegisterRequest::BROWSER; |
| #endif |
| policy_manager()->core()->client()->Register( |
| registration_type, "bogus", std::string(), false, std::string()); |
| run_loop.Run(); |
| Mock::VerifyAndClearExpectations(&observer); |
| policy_manager()->core()->client()->RemoveObserver(&observer); |
| } |
| |
| scoped_ptr<TestRequestInterceptor> interceptor_; |
| }; |
| |
| IN_PROC_BROWSER_TEST_F(CloudPolicyManagerTest, Register) { |
| // Accept one register request. The initial request should not include the |
| // reregister flag. |
| em::DeviceRegisterRequest::Type expected_type = |
| #if defined(OS_CHROMEOS) |
| em::DeviceRegisterRequest::USER; |
| #else |
| em::DeviceRegisterRequest::BROWSER; |
| #endif |
| const bool expect_reregister = false; |
| interceptor_->PushJobCallback( |
| TestRequestInterceptor::RegisterJob(expected_type, expect_reregister)); |
| |
| EXPECT_FALSE(policy_manager()->core()->client()->is_registered()); |
| ASSERT_NO_FATAL_FAILURE(Register()); |
| EXPECT_TRUE(policy_manager()->core()->client()->is_registered()); |
| } |
| |
| IN_PROC_BROWSER_TEST_F(CloudPolicyManagerTest, RegisterFails) { |
| // The interceptor makes all requests fail by default; this will trigger |
| // an OnClientError() call on the observer. |
| EXPECT_FALSE(policy_manager()->core()->client()->is_registered()); |
| ASSERT_NO_FATAL_FAILURE(Register()); |
| EXPECT_FALSE(policy_manager()->core()->client()->is_registered()); |
| } |
| |
| IN_PROC_BROWSER_TEST_F(CloudPolicyManagerTest, RegisterFailsWithRetries) { |
| // Fail 4 times with ERR_NETWORK_CHANGED; the first 3 will trigger a retry, |
| // the last one will forward the error to the client and unblock the |
| // register process. |
| for (int i = 0; i < 4; ++i) { |
| interceptor_->PushJobCallback( |
| TestRequestInterceptor::ErrorJob(net::ERR_NETWORK_CHANGED)); |
| } |
| |
| EXPECT_FALSE(policy_manager()->core()->client()->is_registered()); |
| ASSERT_NO_FATAL_FAILURE(Register()); |
| EXPECT_FALSE(policy_manager()->core()->client()->is_registered()); |
| } |
| |
| IN_PROC_BROWSER_TEST_F(CloudPolicyManagerTest, RegisterWithRetry) { |
| // Accept one register request after failing once. The retry request should |
| // set the reregister flag. |
| interceptor_->PushJobCallback( |
| TestRequestInterceptor::ErrorJob(net::ERR_NETWORK_CHANGED)); |
| em::DeviceRegisterRequest::Type expected_type = |
| #if defined(OS_CHROMEOS) |
| em::DeviceRegisterRequest::USER; |
| #else |
| em::DeviceRegisterRequest::BROWSER; |
| #endif |
| const bool expect_reregister = true; |
| interceptor_->PushJobCallback( |
| TestRequestInterceptor::RegisterJob(expected_type, expect_reregister)); |
| |
| EXPECT_FALSE(policy_manager()->core()->client()->is_registered()); |
| ASSERT_NO_FATAL_FAILURE(Register()); |
| EXPECT_TRUE(policy_manager()->core()->client()->is_registered()); |
| } |
| |
| } // namespace policy |