15821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Copyright (c) 2012 The Chromium Authors. All rights reserved. 25821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Use of this source code is governed by a BSD-style license that can be 35821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// found in the LICENSE file. 45821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 52a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include "chrome/browser/chromeos/policy/user_cloud_policy_manager_chromeos.h" 65821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 7c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)#include "base/bind.h" 8c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)#include "base/bind_helpers.h" 97dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch#include "base/callback.h" 109ab5563a3196760eb381d102cbb2bc0f7abc6a50Ben Murdoch#include "base/message_loop/message_loop.h" 112a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include "base/prefs/pref_registry_simple.h" 122a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include "base/prefs/testing_pref_service.h" 13c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)#include "base/run_loop.h" 14d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles)#include "base/sequenced_task_runner.h" 155e3f23d412006dc4db4e659864679f29341e113fTorne (Richard Coles)#include "base/strings/string_util.h" 16868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)#include "base/strings/utf_string_conversions.h" 17d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles)#include "base/test/test_simple_task_runner.h" 18c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)#include "chrome/browser/chromeos/policy/user_cloud_policy_token_forwarder.h" 19c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)#include "chrome/browser/chromeos/profiles/profile_helper.h" 205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "chrome/browser/prefs/browser_prefs.h" 21eb525c5499e34cc9c4b825d6d9e75bb07cc06aceBen Murdoch#include "chrome/browser/prefs/pref_service_syncable.h" 22f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles)#include "chrome/browser/signin/fake_profile_oauth2_token_service.h" 23effb81e5f8246d0db0270817048dc992db66e9fbBen Murdoch#include "chrome/browser/signin/fake_profile_oauth2_token_service_builder.h" 24a36e5920737c6adbddd3e43b760e5de8431db6e0Torne (Richard Coles)#include "chrome/browser/signin/profile_oauth2_token_service_factory.h" 25a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles)#include "chrome/browser/signin/signin_manager_factory.h" 26c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)#include "chrome/common/chrome_constants.h" 27c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)#include "chrome/test/base/testing_browser_process.h" 28c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)#include "chrome/test/base/testing_profile.h" 29c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)#include "chrome/test/base/testing_profile_manager.h" 30a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles)#include "components/policy/core/common/cloud/cloud_external_data_manager.h" 31a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles)#include "components/policy/core/common/cloud/mock_cloud_external_data_manager.h" 32a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles)#include "components/policy/core/common/cloud/mock_cloud_policy_store.h" 33a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles)#include "components/policy/core/common/cloud/mock_device_management_service.h" 34f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles)#include "components/policy/core/common/external_data_fetcher.h" 35a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles)#include "components/policy/core/common/mock_configuration_policy_provider.h" 36f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles)#include "components/policy/core/common/schema_registry.h" 37effb81e5f8246d0db0270817048dc992db66e9fbBen Murdoch#include "components/signin/core/browser/profile_oauth2_token_service.h" 38e5d81f57cb97b3b6b7fccc9c5610d21eb81db09dBen Murdoch#include "components/signin/core/browser/signin_manager.h" 39a3f7b4e666c476898878fa745f637129375cd889Ben Murdoch#include "content/public/test/test_browser_thread_bundle.h" 40c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)#include "google_apis/gaia/gaia_auth_consumer.h" 41f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles)#include "google_apis/gaia/gaia_constants.h" 42c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)#include "google_apis/gaia/gaia_urls.h" 43c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)#include "net/url_request/test_url_fetcher_factory.h" 44c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)#include "net/url_request/url_fetcher_delegate.h" 452a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include "net/url_request/url_request_context_getter.h" 46c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)#include "net/url_request/url_request_status.h" 475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "policy/policy_constants.h" 48a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles)#include "policy/proto/device_management_backend.pb.h" 495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "testing/gmock/include/gmock/gmock.h" 505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "testing/gtest/include/gtest/gtest.h" 515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace em = enterprise_management; 535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)using testing::AnyNumber; 555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)using testing::AtLeast; 565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)using testing::Mock; 575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)using testing::_; 585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace policy { 60c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 616e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles)const char kOAuthCodeCookie[] = "oauth_code=1234; Secure; HttpOnly"; 62c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 63c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)const char kOAuth2TokenPairData[] = 64c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) "{" 65c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) " \"refresh_token\": \"1234\"," 66c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) " \"access_token\": \"5678\"," 67c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) " \"expires_in\": 3600" 68c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) "}"; 69c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 70c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)const char kOAuth2AccessTokenData[] = 71c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) "{" 72c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) " \"access_token\": \"5678\"," 73c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) " \"expires_in\": 3600" 74c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) "}"; 75c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 762a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)class UserCloudPolicyManagerChromeOSTest : public testing::Test { 775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) protected: 782a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) UserCloudPolicyManagerChromeOSTest() 79a3f7b4e666c476898878fa745f637129375cd889Ben Murdoch : store_(NULL), 80d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) external_data_manager_(NULL), 81d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) task_runner_(new base::TestSimpleTaskRunner()), 82c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) profile_(NULL), 83c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) signin_profile_(NULL) {} 845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual void SetUp() OVERRIDE { 86c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // The initialization path that blocks on the initial policy fetch requires 87c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // a signin Profile to use its URLRequestContext. 88c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) profile_manager_.reset( 89c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) new TestingProfileManager(TestingBrowserProcess::GetGlobal())); 90c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) ASSERT_TRUE(profile_manager_->SetUp()); 91f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) TestingProfile::TestingFactories factories; 92f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) factories.push_back( 93f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) std::make_pair(ProfileOAuth2TokenServiceFactory::GetInstance(), 94effb81e5f8246d0db0270817048dc992db66e9fbBen Murdoch BuildFakeProfileOAuth2TokenService)); 95c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) profile_ = profile_manager_->CreateTestingProfile( 961320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci chrome::kInitialProfile, 971320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci scoped_ptr<PrefServiceSyncable>(), 981320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci base::UTF8ToUTF16(""), 991320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci 0, 1001320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci std::string(), 1011320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci factories); 102c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // Usually the signin Profile and the main Profile are separate, but since 103c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // the signin Profile is an OTR Profile then for this test it suffices to 104c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // attach it to the main Profile. 1051320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci signin_profile_ = TestingProfile::Builder().BuildIncognito(profile_); 106c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) ASSERT_EQ(signin_profile_, chromeos::ProfileHelper::GetSigninProfile()); 107c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 1082a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) chrome::RegisterLocalState(prefs_.registry()); 1095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Set up a policy map for testing. 1110529e5d033099cbfc42635f6f6183833b09dff6eBen Murdoch policy_map_.Set(key::kHomepageLocation, 112c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) POLICY_LEVEL_MANDATORY, POLICY_SCOPE_USER, 113effb81e5f8246d0db0270817048dc992db66e9fbBen Murdoch new base::StringValue("http://chromium.org"), 1147dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch NULL); 1150529e5d033099cbfc42635f6f6183833b09dff6eBen Murdoch policy_map_.Set(key::kChromeOsMultiProfileUserBehavior, 1160529e5d033099cbfc42635f6f6183833b09dff6eBen Murdoch POLICY_LEVEL_MANDATORY, POLICY_SCOPE_USER, 1171320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci new base::StringValue("primary-only"), 1181320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci NULL); 1191320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci policy_map_.Set(key::kEasyUnlockAllowed, 1201320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci POLICY_LEVEL_MANDATORY, POLICY_SCOPE_USER, 1211320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci new base::FundamentalValue(false), 1220529e5d033099cbfc42635f6f6183833b09dff6eBen Murdoch NULL); 1232a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) expected_bundle_.Get(PolicyNamespace(POLICY_DOMAIN_CHROME, std::string())) 1242a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) .CopyFrom(policy_map_); 1255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 126c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // Create fake policy blobs to deliver to the client. 127c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) em::DeviceRegisterResponse* register_response = 128c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) register_blob_.mutable_register_response(); 129c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) register_response->set_device_management_token("dmtoken123"); 130c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 131c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) em::CloudPolicySettings policy_proto; 132c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) policy_proto.mutable_homepagelocation()->set_value("http://chromium.org"); 133c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) ASSERT_TRUE( 134c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) policy_proto.SerializeToString(policy_data_.mutable_policy_value())); 135c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) policy_data_.set_policy_type(dm_protocol::kChromeUserPolicyType); 136c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) policy_data_.set_request_token("dmtoken123"); 137c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) policy_data_.set_device_id("id987"); 1385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) em::PolicyFetchResponse* policy_response = 1395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) policy_blob_.mutable_policy_response()->add_response(); 140c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) ASSERT_TRUE(policy_data_.SerializeToString( 1415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) policy_response->mutable_policy_data())); 1425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_CALL(device_management_service_, StartJob(_, _, _, _, _, _, _)) 1445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) .Times(AnyNumber()); 1455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual void TearDown() OVERRIDE { 148a36e5920737c6adbddd3e43b760e5de8431db6e0Torne (Richard Coles) if (token_forwarder_) 149a36e5920737c6adbddd3e43b760e5de8431db6e0Torne (Richard Coles) token_forwarder_->Shutdown(); 1505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (manager_) { 1515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) manager_->RemoveObserver(&observer_); 1525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) manager_->Shutdown(); 1535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 154a3f7b4e666c476898878fa745f637129375cd889Ben Murdoch signin_profile_ = NULL; 155a3f7b4e666c476898878fa745f637129375cd889Ben Murdoch profile_ = NULL; 156a3f7b4e666c476898878fa745f637129375cd889Ben Murdoch profile_manager_->DeleteTestingProfile(chrome::kInitialProfile); 1575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 15958537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) void CreateManager(bool wait_for_fetch, int fetch_timeout) { 1605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) store_ = new MockCloudPolicyStore(); 161d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) external_data_manager_ = new MockCloudExternalDataManager; 162d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) external_data_manager_->SetPolicyStore(store_); 1635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_CALL(*store_, Load()); 1642a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) manager_.reset(new UserCloudPolicyManagerChromeOS( 1652a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) scoped_ptr<CloudPolicyStore>(store_), 166d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) scoped_ptr<CloudExternalDataManager>(external_data_manager_), 167f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) base::FilePath(), 16858537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) wait_for_fetch, 169f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) base::TimeDelta::FromSeconds(fetch_timeout), 170f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) task_runner_, 171f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) task_runner_, 172f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) task_runner_)); 173f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) manager_->Init(&schema_registry_); 1745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) manager_->AddObserver(&observer_); 1752a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) manager_->Connect(&prefs_, &device_management_service_, NULL, 1762a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) USER_AFFILIATION_NONE); 1775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) Mock::VerifyAndClearExpectations(store_); 1782a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) EXPECT_FALSE(manager_->IsInitializationComplete(POLICY_DOMAIN_CHROME)); 179c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) EXPECT_FALSE(manager_->core()->service()->IsInitializationComplete()); 180c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 181c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) if (!wait_for_fetch) { 182a36e5920737c6adbddd3e43b760e5de8431db6e0Torne (Richard Coles) // Create the UserCloudPolicyTokenForwarder, which fetches the access 183a36e5920737c6adbddd3e43b760e5de8431db6e0Torne (Richard Coles) // token using the OAuth2PolicyFetcher and forwards it to the 184a36e5920737c6adbddd3e43b760e5de8431db6e0Torne (Richard Coles) // UserCloudPolicyManagerChromeOS. This service is automatically created 185a36e5920737c6adbddd3e43b760e5de8431db6e0Torne (Richard Coles) // for regular Profiles but not for testing Profiles. 186a36e5920737c6adbddd3e43b760e5de8431db6e0Torne (Richard Coles) ProfileOAuth2TokenService* token_service = 187a36e5920737c6adbddd3e43b760e5de8431db6e0Torne (Richard Coles) ProfileOAuth2TokenServiceFactory::GetForProfile(profile_); 188c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) ASSERT_TRUE(token_service); 189a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) SigninManagerBase* signin_manager = 190a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) SigninManagerFactory::GetForProfile(profile_); 191a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) ASSERT_TRUE(signin_manager); 192c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) token_forwarder_.reset( 193a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) new UserCloudPolicyTokenForwarder(manager_.get(), token_service, 194a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) signin_manager)); 195c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) } 196c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) } 1975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 198c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // Expects a pending URLFetcher for the |expected_url|, and returns it with 199c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // prepared to deliver a response to its delegate. 200d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) net::TestURLFetcher* PrepareOAuthFetcher(const GURL& expected_url) { 201c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) net::TestURLFetcher* fetcher = test_url_fetcher_factory_.GetFetcherByID(0); 202c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) EXPECT_TRUE(fetcher); 203c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) if (!fetcher) 204c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) return NULL; 205c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) EXPECT_TRUE(fetcher->delegate()); 206c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) EXPECT_TRUE(StartsWithASCII(fetcher->GetOriginalURL().spec(), 207d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) expected_url.spec(), 208c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) true)); 209c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) fetcher->set_url(fetcher->GetOriginalURL()); 210c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) fetcher->set_response_code(200); 211c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) fetcher->set_status(net::URLRequestStatus()); 212c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) return fetcher; 213c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) } 214c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 215c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // Issues the OAuth2 tokens and returns the device management register job 216c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // if the flow succeeded. 217c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) MockDeviceManagementJob* IssueOAuthToken(bool has_request_token) { 218c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) EXPECT_FALSE(manager_->core()->client()->is_registered()); 219c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 220f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) // Issuing this token triggers the callback of the OAuth2PolicyFetcher, 221f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) // which triggers the registration request. 222f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) MockDeviceManagementJob* register_request = NULL; 223f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) EXPECT_CALL(device_management_service_, 224a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles) CreateJob(DeviceManagementRequestJob::TYPE_REGISTRATION, _)) 225f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) .WillOnce(device_management_service_.CreateAsyncJob(®ister_request)); 226c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 227c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) if (!has_request_token) { 228f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) GaiaUrls* gaia_urls = GaiaUrls::GetInstance(); 229f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) net::TestURLFetcher* fetcher = NULL; 230f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) 231c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // Issue the oauth_token cookie first. 232c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) fetcher = PrepareOAuthFetcher(gaia_urls->client_login_to_oauth2_url()); 233c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) if (!fetcher) 234c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) return NULL; 235c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) net::ResponseCookies cookies; 2366e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles) cookies.push_back(kOAuthCodeCookie); 2376e8cce623b6e4fe0c9e4af605d675dd9d0338c38Torne (Richard Coles) 238c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) fetcher->set_cookies(cookies); 239c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) fetcher->delegate()->OnURLFetchComplete(fetcher); 240c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 241c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // Issue the refresh token. 242c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) fetcher = PrepareOAuthFetcher(gaia_urls->oauth2_token_url()); 243c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) if (!fetcher) 244c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) return NULL; 245c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) fetcher->SetResponseString(kOAuth2TokenPairData); 246c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) fetcher->delegate()->OnURLFetchComplete(fetcher); 247c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 248f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) // Issue the access token. 249f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) fetcher = PrepareOAuthFetcher(gaia_urls->oauth2_token_url()); 250f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) if (!fetcher) 251f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) return NULL; 252f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) fetcher->SetResponseString(kOAuth2AccessTokenData); 253f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) fetcher->delegate()->OnURLFetchComplete(fetcher); 254f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) } else { 255f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) // Since the refresh token is available, OAuth2TokenService was used 256f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) // to request the access token and not UserCloudPolicyTokenForwarder. 257f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) // Issue the access token with the former. 258f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) FakeProfileOAuth2TokenService* token_service = 259f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) static_cast<FakeProfileOAuth2TokenService*>( 260f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) ProfileOAuth2TokenServiceFactory::GetForProfile(profile_)); 261f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) EXPECT_TRUE(token_service); 262f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) OAuth2TokenService::ScopeSet scopes; 263f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) scopes.insert(GaiaConstants::kDeviceManagementServiceOAuth); 264a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) scopes.insert(GaiaConstants::kOAuthWrapBridgeUserInfoScope); 265f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) token_service->IssueTokenForScope( 266f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) scopes, "5678", 267f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) base::Time::Now() + base::TimeDelta::FromSeconds(3600)); 268f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) } 269c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 270c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) EXPECT_TRUE(register_request); 271c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) EXPECT_FALSE(manager_->core()->client()->is_registered()); 272c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 273c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) Mock::VerifyAndClearExpectations(&device_management_service_); 274c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) EXPECT_CALL(device_management_service_, StartJob(_, _, _, _, _, _, _)) 275c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) .Times(AnyNumber()); 276c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 277c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) return register_request; 278c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) } 279c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 280c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // Expects a policy fetch request to be issued after invoking |trigger_fetch|. 281c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // This method replies to that fetch request and verifies that the manager 282c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // handled the response. 283c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) void FetchPolicy(const base::Closure& trigger_fetch) { 284c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) MockDeviceManagementJob* policy_request = NULL; 285c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) EXPECT_CALL(device_management_service_, 286a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles) CreateJob(DeviceManagementRequestJob::TYPE_POLICY_FETCH, _)) 287c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) .WillOnce(device_management_service_.CreateAsyncJob(&policy_request)); 288c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) trigger_fetch.Run(); 289c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) ASSERT_TRUE(policy_request); 290c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) EXPECT_TRUE(manager_->core()->service()->IsInitializationComplete()); 291c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) EXPECT_TRUE(manager_->core()->client()->is_registered()); 292c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 293c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) Mock::VerifyAndClearExpectations(&device_management_service_); 294c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) EXPECT_CALL(device_management_service_, StartJob(_, _, _, _, _, _, _)) 295c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) .Times(AnyNumber()); 296c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 297c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // Send the initial policy back. This completes the initialization flow. 298c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) EXPECT_CALL(*store_, Store(_)); 299c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) policy_request->SendResponse(DM_STATUS_SUCCESS, policy_blob_); 300c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) Mock::VerifyAndClearExpectations(store_); 301c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 302c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // Notifying that the store is has cached the fetched policy completes the 303c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // process, and initializes the manager. 304c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) EXPECT_CALL(observer_, OnUpdatePolicy(manager_.get())); 305c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) store_->policy_map_.CopyFrom(policy_map_); 3065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) store_->NotifyStoreLoaded(); 307c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) EXPECT_TRUE(manager_->IsInitializationComplete(POLICY_DOMAIN_CHROME)); 3085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) Mock::VerifyAndClearExpectations(&observer_); 309c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) EXPECT_TRUE(manager_->policies().Equals(expected_bundle_)); 3105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 3115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 312a3f7b4e666c476898878fa745f637129375cd889Ben Murdoch // Required by the refresh scheduler that's created by the manager and 313a3f7b4e666c476898878fa745f637129375cd889Ben Murdoch // for the cleanup of URLRequestContextGetter in the |signin_profile_|. 314a3f7b4e666c476898878fa745f637129375cd889Ben Murdoch content::TestBrowserThreadBundle thread_bundle_; 3155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Convenience policy objects. 317c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) em::PolicyData policy_data_; 318c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) em::DeviceManagementResponse register_blob_; 3195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) em::DeviceManagementResponse policy_blob_; 3205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) PolicyMap policy_map_; 3215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) PolicyBundle expected_bundle_; 3225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Policy infrastructure. 324c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) net::TestURLFetcherFactory test_url_fetcher_factory_; 3252a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) TestingPrefServiceSimple prefs_; 3265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) MockConfigurationPolicyObserver observer_; 3275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) MockDeviceManagementService device_management_service_; 328d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) MockCloudPolicyStore* store_; // Not owned. 329d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) MockCloudExternalDataManager* external_data_manager_; // Not owned. 330d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) scoped_refptr<base::TestSimpleTaskRunner> task_runner_; 331f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) SchemaRegistry schema_registry_; 3322a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) scoped_ptr<UserCloudPolicyManagerChromeOS> manager_; 333c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) scoped_ptr<UserCloudPolicyTokenForwarder> token_forwarder_; 334c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 335c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // Required by ProfileHelper to get the signin Profile context. 336c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) scoped_ptr<TestingProfileManager> profile_manager_; 337c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) TestingProfile* profile_; 338c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) TestingProfile* signin_profile_; 3395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 340a3f7b4e666c476898878fa745f637129375cd889Ben Murdoch static const char kSigninProfile[]; 341a3f7b4e666c476898878fa745f637129375cd889Ben Murdoch 3425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) private: 3432a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) DISALLOW_COPY_AND_ASSIGN(UserCloudPolicyManagerChromeOSTest); 3445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}; 3455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 346a3f7b4e666c476898878fa745f637129375cd889Ben Murdochconst char UserCloudPolicyManagerChromeOSTest::kSigninProfile[] = 347a3f7b4e666c476898878fa745f637129375cd889Ben Murdoch "signin_profile"; 348a3f7b4e666c476898878fa745f637129375cd889Ben Murdoch 349c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)TEST_F(UserCloudPolicyManagerChromeOSTest, BlockingFirstFetch) { 350c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // Tests the initialization of a manager whose Profile is waiting for the 351c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // initial fetch, when the policy cache is empty. 35258537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) ASSERT_NO_FATAL_FAILURE(CreateManager(true, 1000)); 3535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 354c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // Initialize the CloudPolicyService without any stored data. 355c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) EXPECT_FALSE(manager_->core()->service()->IsInitializationComplete()); 356c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) store_->NotifyStoreLoaded(); 357c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) EXPECT_TRUE(manager_->core()->service()->IsInitializationComplete()); 358c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) EXPECT_FALSE(manager_->core()->client()->is_registered()); 3595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 360c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // This starts the OAuth2 policy token fetcher using the signin Profile. 361c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // The manager will then issue the registration request. 362c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) MockDeviceManagementJob* register_request = IssueOAuthToken(false); 363c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) ASSERT_TRUE(register_request); 364c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 365c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // Reply with a valid registration response. This triggers the initial policy 366c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // fetch. 367c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) FetchPolicy(base::Bind(&MockDeviceManagementJob::SendResponse, 368c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) base::Unretained(register_request), 369c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) DM_STATUS_SUCCESS, register_blob_)); 370c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)} 371c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 372c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)TEST_F(UserCloudPolicyManagerChromeOSTest, BlockingRefreshFetch) { 373c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // Tests the initialization of a manager whose Profile is waiting for the 374c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // initial fetch, when a previously cached policy and DMToken already exist. 37558537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) ASSERT_NO_FATAL_FAILURE(CreateManager(true, 1000)); 376c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 377c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // Set the initially cached data and initialize the CloudPolicyService. 378c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // The initial policy fetch is issued using the cached DMToken. 379c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) store_->policy_.reset(new em::PolicyData(policy_data_)); 380c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) FetchPolicy(base::Bind(&MockCloudPolicyStore::NotifyStoreLoaded, 381c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) base::Unretained(store_))); 382c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)} 383c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 384c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)TEST_F(UserCloudPolicyManagerChromeOSTest, BlockingFetchStoreError) { 385c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // Tests the initialization of a manager whose Profile is waiting for the 386c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // initial fetch, when the initial store load fails. 38758537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) ASSERT_NO_FATAL_FAILURE(CreateManager(true, 1000)); 388c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 389c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // Initialize the CloudPolicyService without any stored data. 390c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) EXPECT_FALSE(manager_->core()->service()->IsInitializationComplete()); 391c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) store_->NotifyStoreError(); 392c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) EXPECT_TRUE(manager_->core()->service()->IsInitializationComplete()); 393c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) EXPECT_FALSE(manager_->core()->client()->is_registered()); 394c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 395c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // This starts the OAuth2 policy token fetcher using the signin Profile. 396c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // The manager will then issue the registration request. 397c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) MockDeviceManagementJob* register_request = IssueOAuthToken(false); 398c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) ASSERT_TRUE(register_request); 399c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 400c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // Reply with a valid registration response. This triggers the initial policy 401c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // fetch. 402c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) FetchPolicy(base::Bind(&MockDeviceManagementJob::SendResponse, 403c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) base::Unretained(register_request), 404c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) DM_STATUS_SUCCESS, register_blob_)); 405c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)} 406c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 407c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)TEST_F(UserCloudPolicyManagerChromeOSTest, BlockingFetchOAuthError) { 408c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // Tests the initialization of a manager whose Profile is waiting for the 409c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // initial fetch, when the OAuth2 token fetch fails. 41058537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) ASSERT_NO_FATAL_FAILURE(CreateManager(true, 1000)); 411c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 412c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // Initialize the CloudPolicyService without any stored data. 413c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) EXPECT_FALSE(manager_->core()->service()->IsInitializationComplete()); 414c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) store_->NotifyStoreLoaded(); 415c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) EXPECT_TRUE(manager_->core()->service()->IsInitializationComplete()); 416c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) EXPECT_FALSE(manager_->core()->client()->is_registered()); 417c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 418c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // This starts the OAuth2 policy token fetcher using the signin Profile. 419c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // The manager will initialize with no policy after the token fetcher fails. 420c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) EXPECT_CALL(observer_, OnUpdatePolicy(manager_.get())); 421c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 422c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // The PolicyOAuth2TokenFetcher posts delayed retries on some errors. This 423c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // data will make it fail immediately. 424c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) net::TestURLFetcher* fetcher = PrepareOAuthFetcher( 425c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) GaiaUrls::GetInstance()->client_login_to_oauth2_url()); 426c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) ASSERT_TRUE(fetcher); 427c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) fetcher->set_response_code(400); 428c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) fetcher->SetResponseString("Error=BadAuthentication"); 4292a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) EXPECT_FALSE(manager_->IsInitializationComplete(POLICY_DOMAIN_CHROME)); 430c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) fetcher->delegate()->OnURLFetchComplete(fetcher); 431c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) EXPECT_TRUE(manager_->IsInitializationComplete(POLICY_DOMAIN_CHROME)); 432c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) EXPECT_TRUE(PolicyBundle().Equals(manager_->policies())); 4335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) Mock::VerifyAndClearExpectations(&observer_); 434c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)} 435c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 436c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)TEST_F(UserCloudPolicyManagerChromeOSTest, BlockingFetchRegisterError) { 437c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // Tests the initialization of a manager whose Profile is waiting for the 438c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // initial fetch, when the device management registration fails. 43958537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) ASSERT_NO_FATAL_FAILURE(CreateManager(true, 1000)); 440c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 441c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // Initialize the CloudPolicyService without any stored data. 442c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) EXPECT_FALSE(manager_->core()->service()->IsInitializationComplete()); 443c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) store_->NotifyStoreError(); 444c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) EXPECT_TRUE(manager_->core()->service()->IsInitializationComplete()); 445c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) EXPECT_FALSE(manager_->core()->client()->is_registered()); 446c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 447c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // This starts the OAuth2 policy token fetcher using the signin Profile. 448c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // The manager will then issue the registration request. 449c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) MockDeviceManagementJob* register_request = IssueOAuthToken(false); 450c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) ASSERT_TRUE(register_request); 4515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 452c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // Now make it fail. 453c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) EXPECT_FALSE(manager_->IsInitializationComplete(POLICY_DOMAIN_CHROME)); 4545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_CALL(observer_, OnUpdatePolicy(manager_.get())); 455c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) register_request->SendResponse(DM_STATUS_TEMPORARY_UNAVAILABLE, 456c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) em::DeviceManagementResponse()); 4572a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) EXPECT_TRUE(manager_->IsInitializationComplete(POLICY_DOMAIN_CHROME)); 458c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) EXPECT_TRUE(PolicyBundle().Equals(manager_->policies())); 4595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) Mock::VerifyAndClearExpectations(&observer_); 4605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 4615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 462c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)TEST_F(UserCloudPolicyManagerChromeOSTest, BlockingFetchPolicyFetchError) { 463c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // Tests the initialization of a manager whose Profile is waiting for the 464c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // initial fetch, when the policy fetch request fails. 46558537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) ASSERT_NO_FATAL_FAILURE(CreateManager(true, 1000)); 466c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 467c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // Initialize the CloudPolicyService without any stored data. 468c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) EXPECT_FALSE(manager_->core()->service()->IsInitializationComplete()); 469c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) store_->NotifyStoreLoaded(); 470c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) EXPECT_TRUE(manager_->core()->service()->IsInitializationComplete()); 471c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) EXPECT_FALSE(manager_->core()->client()->is_registered()); 472c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 473c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // This starts the OAuth2 policy token fetcher using the signin Profile. 474c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // The manager will then issue the registration request. 475c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) MockDeviceManagementJob* register_request = IssueOAuthToken(false); 476c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) ASSERT_TRUE(register_request); 4775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 478c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // Reply with a valid registration response. This triggers the initial policy 479c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // fetch. 480c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) MockDeviceManagementJob* policy_request = NULL; 4815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_CALL(device_management_service_, 482a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles) CreateJob(DeviceManagementRequestJob::TYPE_POLICY_FETCH, _)) 483c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) .WillOnce(device_management_service_.CreateAsyncJob(&policy_request)); 484c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) register_request->SendResponse(DM_STATUS_SUCCESS, register_blob_); 485c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) Mock::VerifyAndClearExpectations(&device_management_service_); 486c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) ASSERT_TRUE(policy_request); 487c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) EXPECT_TRUE(manager_->core()->service()->IsInitializationComplete()); 488c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) EXPECT_TRUE(manager_->core()->client()->is_registered()); 4895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 490c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // Make the policy fetch fail. The observer gets 2 notifications: one from the 491c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // RefreshPolicies callback, and another from the OnClientError callback. 492c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // A single notification suffices for this edge case, but this behavior is 493c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // also correct and makes the implementation simpler. 4945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_CALL(observer_, OnUpdatePolicy(manager_.get())).Times(AtLeast(1)); 495c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) EXPECT_FALSE(manager_->IsInitializationComplete(POLICY_DOMAIN_CHROME)); 496c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) policy_request->SendResponse(DM_STATUS_TEMPORARY_UNAVAILABLE, 497c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) em::DeviceManagementResponse()); 4985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) Mock::VerifyAndClearExpectations(&observer_); 499c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) EXPECT_TRUE(manager_->IsInitializationComplete(POLICY_DOMAIN_CHROME)); 500c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) EXPECT_TRUE(PolicyBundle().Equals(manager_->policies())); 5015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 5025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 50358537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles)TEST_F(UserCloudPolicyManagerChromeOSTest, BlockingFetchTimeout) { 50458537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) // The blocking fetch should be abandoned after the timeout. 50558537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) ASSERT_NO_FATAL_FAILURE(CreateManager(true, 0)); 50658537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) 50758537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) // Initialize the CloudPolicyService without any stored data. 50858537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) EXPECT_FALSE(manager_->core()->service()->IsInitializationComplete()); 50958537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) store_->NotifyStoreLoaded(); 51058537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) EXPECT_TRUE(manager_->core()->service()->IsInitializationComplete()); 51158537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) EXPECT_FALSE(manager_->core()->client()->is_registered()); 51258537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) 51358537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) // Running the message loop should trigger the timeout. 51458537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) EXPECT_CALL(observer_, OnUpdatePolicy(manager_.get())).Times(AtLeast(1)); 51558537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) EXPECT_FALSE(manager_->IsInitializationComplete(POLICY_DOMAIN_CHROME)); 51658537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) base::RunLoop().RunUntilIdle(); 51758537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) Mock::VerifyAndClearExpectations(&observer_); 51858537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) EXPECT_TRUE(manager_->IsInitializationComplete(POLICY_DOMAIN_CHROME)); 51958537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) EXPECT_TRUE(PolicyBundle().Equals(manager_->policies())); 52058537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles)} 52158537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) 52258537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) 523c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)TEST_F(UserCloudPolicyManagerChromeOSTest, NonBlockingFirstFetch) { 524c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // Tests the first policy fetch request by a Profile that isn't managed. 52558537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) ASSERT_NO_FATAL_FAILURE(CreateManager(false, 1000)); 5265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 527c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // Initialize the CloudPolicyService without any stored data. Since the 528c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // manager is not waiting for the initial fetch, it will become initialized 529c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // once the store is ready. 530c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) EXPECT_FALSE(manager_->core()->service()->IsInitializationComplete()); 531c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) EXPECT_FALSE(manager_->IsInitializationComplete(POLICY_DOMAIN_CHROME)); 5325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_CALL(observer_, OnUpdatePolicy(manager_.get())); 533c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) store_->NotifyStoreLoaded(); 534c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) Mock::VerifyAndClearExpectations(&observer_); 535c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) EXPECT_TRUE(manager_->core()->service()->IsInitializationComplete()); 5362a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) EXPECT_TRUE(manager_->IsInitializationComplete(POLICY_DOMAIN_CHROME)); 537c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) EXPECT_FALSE(manager_->core()->client()->is_registered()); 538c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 539c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // The manager is waiting for the refresh token, and hasn't started any 540c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // fetchers. 541c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) EXPECT_FALSE(test_url_fetcher_factory_.GetFetcherByID(0)); 542c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 543f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) // Set a fake refresh token at the OAuth2TokenService. 544f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) FakeProfileOAuth2TokenService* token_service = 545f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) static_cast<FakeProfileOAuth2TokenService*>( 546f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) ProfileOAuth2TokenServiceFactory::GetForProfile(profile_)); 547c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) ASSERT_TRUE(token_service); 548a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) SigninManagerBase* signin_manager = 549a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) SigninManagerFactory::GetForProfile(profile_); 550a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) ASSERT_TRUE(signin_manager); 551a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) const std::string& account_id = signin_manager->GetAuthenticatedAccountId(); 5525d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) EXPECT_FALSE(token_service->RefreshTokenIsAvailable(account_id)); 5535d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) token_service->UpdateCredentials(account_id, "refresh_token"); 5545d1f7b1de12d16ceb2c938c56701a3e8bfa558f7Torne (Richard Coles) EXPECT_TRUE(token_service->RefreshTokenIsAvailable(account_id)); 555c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 556c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // That should have notified the manager, which now issues the request for the 557c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // policy oauth token. 558c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) MockDeviceManagementJob* register_request = IssueOAuthToken(true); 559c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) ASSERT_TRUE(register_request); 560c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) register_request->SendResponse(DM_STATUS_SUCCESS, register_blob_); 561c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 562c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // The refresh scheduler takes care of the initial fetch for unmanaged users. 563d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) // Running the task runner issues the initial fetch. 564c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) FetchPolicy( 565d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) base::Bind(&base::TestSimpleTaskRunner::RunUntilIdle, task_runner_)); 566c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)} 567c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 568c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)TEST_F(UserCloudPolicyManagerChromeOSTest, NonBlockingRefreshFetch) { 569c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // Tests a non-blocking initial policy fetch for a Profile that already has 570c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // a cached DMToken. 57158537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) ASSERT_NO_FATAL_FAILURE(CreateManager(false, 1000)); 572c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 573c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // Set the initially cached data and initialize the CloudPolicyService. 574c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // The initial policy fetch is issued using the cached DMToken. 575c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) EXPECT_FALSE(manager_->core()->service()->IsInitializationComplete()); 576c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) EXPECT_FALSE(manager_->IsInitializationComplete(POLICY_DOMAIN_CHROME)); 577c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) EXPECT_CALL(observer_, OnUpdatePolicy(manager_.get())); 578c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) store_->policy_.reset(new em::PolicyData(policy_data_)); 579c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) store_->NotifyStoreLoaded(); 5805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) Mock::VerifyAndClearExpectations(&observer_); 581c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) EXPECT_TRUE(manager_->core()->service()->IsInitializationComplete()); 582c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) EXPECT_TRUE(manager_->IsInitializationComplete(POLICY_DOMAIN_CHROME)); 583c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) EXPECT_TRUE(manager_->core()->client()->is_registered()); 584c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 585c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // The refresh scheduler takes care of the initial fetch for unmanaged users. 586d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) // Running the task runner issues the initial fetch. 587c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) FetchPolicy( 588d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) base::Bind(&base::TestSimpleTaskRunner::RunUntilIdle, task_runner_)); 5895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 5905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} // namespace policy 592