user_cloud_policy_manager_chromeos_unittest.cc revision a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7
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" 23a36e5920737c6adbddd3e43b760e5de8431db6e0Torne (Richard Coles)#include "chrome/browser/signin/profile_oauth2_token_service.h" 24a36e5920737c6adbddd3e43b760e5de8431db6e0Torne (Richard Coles)#include "chrome/browser/signin/profile_oauth2_token_service_factory.h" 25c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)#include "chrome/common/chrome_constants.h" 26c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)#include "chrome/test/base/testing_browser_process.h" 27c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)#include "chrome/test/base/testing_profile.h" 28c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)#include "chrome/test/base/testing_profile_manager.h" 29a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles)#include "components/policy/core/common/cloud/cloud_external_data_manager.h" 30a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles)#include "components/policy/core/common/cloud/mock_cloud_external_data_manager.h" 31a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles)#include "components/policy/core/common/cloud/mock_cloud_policy_store.h" 32a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles)#include "components/policy/core/common/cloud/mock_device_management_service.h" 33f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles)#include "components/policy/core/common/external_data_fetcher.h" 34a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles)#include "components/policy/core/common/mock_configuration_policy_provider.h" 35f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles)#include "components/policy/core/common/schema_registry.h" 36a3f7b4e666c476898878fa745f637129375cd889Ben Murdoch#include "content/public/test/test_browser_thread_bundle.h" 37c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)#include "google_apis/gaia/gaia_auth_consumer.h" 38f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles)#include "google_apis/gaia/gaia_constants.h" 39c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)#include "google_apis/gaia/gaia_urls.h" 40c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)#include "net/url_request/test_url_fetcher_factory.h" 41c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)#include "net/url_request/url_fetcher_delegate.h" 422a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include "net/url_request/url_request_context_getter.h" 43c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)#include "net/url_request/url_request_status.h" 445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "policy/policy_constants.h" 45a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles)#include "policy/proto/device_management_backend.pb.h" 465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "testing/gmock/include/gmock/gmock.h" 475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "testing/gtest/include/gtest/gtest.h" 485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace em = enterprise_management; 505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)using testing::AnyNumber; 525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)using testing::AtLeast; 535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)using testing::Mock; 545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)using testing::_; 555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace policy { 57c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace { 595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 60c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)const char kOAuthTokenCookie[] = "oauth_token=1234"; 61f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles)const char kTestAccountId[] = "user@gmail.com"; 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) 76c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)} // namespace 77c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 782a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)class UserCloudPolicyManagerChromeOSTest : public testing::Test { 795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) protected: 802a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) UserCloudPolicyManagerChromeOSTest() 81a3f7b4e666c476898878fa745f637129375cd889Ben Murdoch : store_(NULL), 82d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) external_data_manager_(NULL), 83d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) task_runner_(new base::TestSimpleTaskRunner()), 84c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) profile_(NULL), 85c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) signin_profile_(NULL) {} 865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual void SetUp() OVERRIDE { 88c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // The initialization path that blocks on the initial policy fetch requires 89c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // a signin Profile to use its URLRequestContext. 90c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) profile_manager_.reset( 91c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) new TestingProfileManager(TestingBrowserProcess::GetGlobal())); 92c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) ASSERT_TRUE(profile_manager_->SetUp()); 93f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) TestingProfile::TestingFactories factories; 94f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) factories.push_back( 95f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) std::make_pair(ProfileOAuth2TokenServiceFactory::GetInstance(), 96f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) FakeProfileOAuth2TokenService::Build)); 97c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) profile_ = profile_manager_->CreateTestingProfile( 98eb525c5499e34cc9c4b825d6d9e75bb07cc06aceBen Murdoch chrome::kInitialProfile, scoped_ptr<PrefServiceSyncable>(), 99f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) UTF8ToUTF16("testing_profile"), 0, std::string(), factories); 100a3f7b4e666c476898878fa745f637129375cd889Ben Murdoch signin_profile_ = profile_manager_->CreateTestingProfile(kSigninProfile); 1013551c9c881056c480085172ff9840cab31610854Torne (Richard Coles) signin_profile_->ForceIncognito(true); 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. 1053551c9c881056c480085172ff9840cab31610854Torne (Richard Coles) profile_->SetOffTheRecordProfile(scoped_ptr<Profile>(signin_profile_)); 106c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) signin_profile_->SetOriginalProfile(profile_); 107c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) ASSERT_EQ(signin_profile_, chromeos::ProfileHelper::GetSigninProfile()); 108c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 1092a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) chrome::RegisterLocalState(prefs_.registry()); 1105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Set up a policy map for testing. 112c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) policy_map_.Set("HomepageLocation", 113c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) POLICY_LEVEL_MANDATORY, POLICY_SCOPE_USER, 1147dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch base::Value::CreateStringValue("http://chromium.org"), 1157dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch NULL); 1162a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) expected_bundle_.Get(PolicyNamespace(POLICY_DOMAIN_CHROME, std::string())) 1172a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) .CopyFrom(policy_map_); 1185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 119c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // Create fake policy blobs to deliver to the client. 120c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) em::DeviceRegisterResponse* register_response = 121c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) register_blob_.mutable_register_response(); 122c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) register_response->set_device_management_token("dmtoken123"); 123c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 124c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) em::CloudPolicySettings policy_proto; 125c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) policy_proto.mutable_homepagelocation()->set_value("http://chromium.org"); 126c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) ASSERT_TRUE( 127c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) policy_proto.SerializeToString(policy_data_.mutable_policy_value())); 128c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) policy_data_.set_policy_type(dm_protocol::kChromeUserPolicyType); 129c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) policy_data_.set_request_token("dmtoken123"); 130c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) policy_data_.set_device_id("id987"); 1315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) em::PolicyFetchResponse* policy_response = 1325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) policy_blob_.mutable_policy_response()->add_response(); 133c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) ASSERT_TRUE(policy_data_.SerializeToString( 1345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) policy_response->mutable_policy_data())); 1355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_CALL(device_management_service_, StartJob(_, _, _, _, _, _, _)) 1375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) .Times(AnyNumber()); 1385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual void TearDown() OVERRIDE { 141a36e5920737c6adbddd3e43b760e5de8431db6e0Torne (Richard Coles) if (token_forwarder_) 142a36e5920737c6adbddd3e43b760e5de8431db6e0Torne (Richard Coles) token_forwarder_->Shutdown(); 1435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (manager_) { 1445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) manager_->RemoveObserver(&observer_); 1455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) manager_->Shutdown(); 1465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 147a3f7b4e666c476898878fa745f637129375cd889Ben Murdoch signin_profile_ = NULL; 148a3f7b4e666c476898878fa745f637129375cd889Ben Murdoch profile_ = NULL; 149a3f7b4e666c476898878fa745f637129375cd889Ben Murdoch profile_manager_->DeleteTestingProfile(kSigninProfile); 150a3f7b4e666c476898878fa745f637129375cd889Ben Murdoch profile_manager_->DeleteTestingProfile(chrome::kInitialProfile); 1515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 15358537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) void CreateManager(bool wait_for_fetch, int fetch_timeout) { 1545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) store_ = new MockCloudPolicyStore(); 155d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) external_data_manager_ = new MockCloudExternalDataManager; 156d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) external_data_manager_->SetPolicyStore(store_); 1575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_CALL(*store_, Load()); 1582a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) manager_.reset(new UserCloudPolicyManagerChromeOS( 1592a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) scoped_ptr<CloudPolicyStore>(store_), 160d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) scoped_ptr<CloudExternalDataManager>(external_data_manager_), 161f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) base::FilePath(), 16258537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) wait_for_fetch, 163f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) base::TimeDelta::FromSeconds(fetch_timeout), 164f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) task_runner_, 165f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) task_runner_, 166f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) task_runner_)); 167f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) manager_->Init(&schema_registry_); 1685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) manager_->AddObserver(&observer_); 1692a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) manager_->Connect(&prefs_, &device_management_service_, NULL, 1702a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) USER_AFFILIATION_NONE); 1715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) Mock::VerifyAndClearExpectations(store_); 1722a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) EXPECT_FALSE(manager_->IsInitializationComplete(POLICY_DOMAIN_CHROME)); 173c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) EXPECT_FALSE(manager_->core()->service()->IsInitializationComplete()); 174c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 175c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) if (!wait_for_fetch) { 176a36e5920737c6adbddd3e43b760e5de8431db6e0Torne (Richard Coles) // Create the UserCloudPolicyTokenForwarder, which fetches the access 177a36e5920737c6adbddd3e43b760e5de8431db6e0Torne (Richard Coles) // token using the OAuth2PolicyFetcher and forwards it to the 178a36e5920737c6adbddd3e43b760e5de8431db6e0Torne (Richard Coles) // UserCloudPolicyManagerChromeOS. This service is automatically created 179a36e5920737c6adbddd3e43b760e5de8431db6e0Torne (Richard Coles) // for regular Profiles but not for testing Profiles. 180a36e5920737c6adbddd3e43b760e5de8431db6e0Torne (Richard Coles) ProfileOAuth2TokenService* token_service = 181a36e5920737c6adbddd3e43b760e5de8431db6e0Torne (Richard Coles) ProfileOAuth2TokenServiceFactory::GetForProfile(profile_); 182c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) ASSERT_TRUE(token_service); 183c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) token_forwarder_.reset( 184c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) new UserCloudPolicyTokenForwarder(manager_.get(), token_service)); 185c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) } 186c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) } 1875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 188c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // Expects a pending URLFetcher for the |expected_url|, and returns it with 189c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // prepared to deliver a response to its delegate. 190d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) net::TestURLFetcher* PrepareOAuthFetcher(const GURL& expected_url) { 191c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) net::TestURLFetcher* fetcher = test_url_fetcher_factory_.GetFetcherByID(0); 192c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) EXPECT_TRUE(fetcher); 193c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) if (!fetcher) 194c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) return NULL; 195c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) EXPECT_TRUE(fetcher->delegate()); 196c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) EXPECT_TRUE(StartsWithASCII(fetcher->GetOriginalURL().spec(), 197d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) expected_url.spec(), 198c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) true)); 199c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) fetcher->set_url(fetcher->GetOriginalURL()); 200c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) fetcher->set_response_code(200); 201c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) fetcher->set_status(net::URLRequestStatus()); 202c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) return fetcher; 203c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) } 204c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 205c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // Issues the OAuth2 tokens and returns the device management register job 206c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // if the flow succeeded. 207c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) MockDeviceManagementJob* IssueOAuthToken(bool has_request_token) { 208c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) EXPECT_FALSE(manager_->core()->client()->is_registered()); 209c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 210f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) // Issuing this token triggers the callback of the OAuth2PolicyFetcher, 211f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) // which triggers the registration request. 212f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) MockDeviceManagementJob* register_request = NULL; 213f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) EXPECT_CALL(device_management_service_, 214a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles) CreateJob(DeviceManagementRequestJob::TYPE_REGISTRATION, _)) 215f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) .WillOnce(device_management_service_.CreateAsyncJob(®ister_request)); 216c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 217c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) if (!has_request_token) { 218f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) GaiaUrls* gaia_urls = GaiaUrls::GetInstance(); 219f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) net::TestURLFetcher* fetcher = NULL; 220f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) 221c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // Issue the oauth_token cookie first. 222c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) fetcher = PrepareOAuthFetcher(gaia_urls->client_login_to_oauth2_url()); 223c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) if (!fetcher) 224c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) return NULL; 225c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) net::ResponseCookies cookies; 226c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) cookies.push_back(kOAuthTokenCookie); 227c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) fetcher->set_cookies(cookies); 228c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) fetcher->delegate()->OnURLFetchComplete(fetcher); 229c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 230c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // Issue the refresh token. 231c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) fetcher = PrepareOAuthFetcher(gaia_urls->oauth2_token_url()); 232c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) if (!fetcher) 233c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) return NULL; 234c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) fetcher->SetResponseString(kOAuth2TokenPairData); 235c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) fetcher->delegate()->OnURLFetchComplete(fetcher); 236c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 237f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) // Issue the access token. 238f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) fetcher = PrepareOAuthFetcher(gaia_urls->oauth2_token_url()); 239f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) if (!fetcher) 240f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) return NULL; 241f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) fetcher->SetResponseString(kOAuth2AccessTokenData); 242f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) fetcher->delegate()->OnURLFetchComplete(fetcher); 243f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) } else { 244f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) // Since the refresh token is available, OAuth2TokenService was used 245f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) // to request the access token and not UserCloudPolicyTokenForwarder. 246f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) // Issue the access token with the former. 247f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) FakeProfileOAuth2TokenService* token_service = 248f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) static_cast<FakeProfileOAuth2TokenService*>( 249f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) ProfileOAuth2TokenServiceFactory::GetForProfile(profile_)); 250f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) EXPECT_TRUE(token_service); 251f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) OAuth2TokenService::ScopeSet scopes; 252f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) scopes.insert(GaiaConstants::kDeviceManagementServiceOAuth); 253f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) token_service->IssueTokenForScope( 254f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) scopes, "5678", 255f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) base::Time::Now() + base::TimeDelta::FromSeconds(3600)); 256f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) } 257c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 258c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) EXPECT_TRUE(register_request); 259c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) EXPECT_FALSE(manager_->core()->client()->is_registered()); 260c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 261c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) Mock::VerifyAndClearExpectations(&device_management_service_); 262c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) EXPECT_CALL(device_management_service_, StartJob(_, _, _, _, _, _, _)) 263c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) .Times(AnyNumber()); 264c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 265c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) return register_request; 266c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) } 267c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 268c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // Expects a policy fetch request to be issued after invoking |trigger_fetch|. 269c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // This method replies to that fetch request and verifies that the manager 270c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // handled the response. 271c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) void FetchPolicy(const base::Closure& trigger_fetch) { 272c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) MockDeviceManagementJob* policy_request = NULL; 273c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) EXPECT_CALL(device_management_service_, 274a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles) CreateJob(DeviceManagementRequestJob::TYPE_POLICY_FETCH, _)) 275c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) .WillOnce(device_management_service_.CreateAsyncJob(&policy_request)); 276c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) trigger_fetch.Run(); 277c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) ASSERT_TRUE(policy_request); 278c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) EXPECT_TRUE(manager_->core()->service()->IsInitializationComplete()); 279c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) EXPECT_TRUE(manager_->core()->client()->is_registered()); 280c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 281c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) Mock::VerifyAndClearExpectations(&device_management_service_); 282c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) EXPECT_CALL(device_management_service_, StartJob(_, _, _, _, _, _, _)) 283c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) .Times(AnyNumber()); 284c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 285c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // Send the initial policy back. This completes the initialization flow. 286c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) EXPECT_CALL(*store_, Store(_)); 287c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) policy_request->SendResponse(DM_STATUS_SUCCESS, policy_blob_); 288c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) Mock::VerifyAndClearExpectations(store_); 289c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 290c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // Notifying that the store is has cached the fetched policy completes the 291c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // process, and initializes the manager. 292c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) EXPECT_CALL(observer_, OnUpdatePolicy(manager_.get())); 293c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) store_->policy_map_.CopyFrom(policy_map_); 2945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) store_->NotifyStoreLoaded(); 295c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) EXPECT_TRUE(manager_->IsInitializationComplete(POLICY_DOMAIN_CHROME)); 2965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) Mock::VerifyAndClearExpectations(&observer_); 297c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) EXPECT_TRUE(manager_->policies().Equals(expected_bundle_)); 2985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 2995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 300a3f7b4e666c476898878fa745f637129375cd889Ben Murdoch // Required by the refresh scheduler that's created by the manager and 301a3f7b4e666c476898878fa745f637129375cd889Ben Murdoch // for the cleanup of URLRequestContextGetter in the |signin_profile_|. 302a3f7b4e666c476898878fa745f637129375cd889Ben Murdoch content::TestBrowserThreadBundle thread_bundle_; 3035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Convenience policy objects. 305c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) em::PolicyData policy_data_; 306c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) em::DeviceManagementResponse register_blob_; 3075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) em::DeviceManagementResponse policy_blob_; 3085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) PolicyMap policy_map_; 3095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) PolicyBundle expected_bundle_; 3105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Policy infrastructure. 312c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) net::TestURLFetcherFactory test_url_fetcher_factory_; 3132a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) TestingPrefServiceSimple prefs_; 3145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) MockConfigurationPolicyObserver observer_; 3155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) MockDeviceManagementService device_management_service_; 316d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) MockCloudPolicyStore* store_; // Not owned. 317d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) MockCloudExternalDataManager* external_data_manager_; // Not owned. 318d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) scoped_refptr<base::TestSimpleTaskRunner> task_runner_; 319f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) SchemaRegistry schema_registry_; 3202a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) scoped_ptr<UserCloudPolicyManagerChromeOS> manager_; 321c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) scoped_ptr<UserCloudPolicyTokenForwarder> token_forwarder_; 322c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 323c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // Required by ProfileHelper to get the signin Profile context. 324c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) scoped_ptr<TestingProfileManager> profile_manager_; 325c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) TestingProfile* profile_; 326c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) TestingProfile* signin_profile_; 3275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 328a3f7b4e666c476898878fa745f637129375cd889Ben Murdoch static const char kSigninProfile[]; 329a3f7b4e666c476898878fa745f637129375cd889Ben Murdoch 3305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) private: 3312a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) DISALLOW_COPY_AND_ASSIGN(UserCloudPolicyManagerChromeOSTest); 3325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}; 3335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 334a3f7b4e666c476898878fa745f637129375cd889Ben Murdochconst char UserCloudPolicyManagerChromeOSTest::kSigninProfile[] = 335a3f7b4e666c476898878fa745f637129375cd889Ben Murdoch "signin_profile"; 336a3f7b4e666c476898878fa745f637129375cd889Ben Murdoch 337c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)TEST_F(UserCloudPolicyManagerChromeOSTest, BlockingFirstFetch) { 338c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // Tests the initialization of a manager whose Profile is waiting for the 339c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // initial fetch, when the policy cache is empty. 34058537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) ASSERT_NO_FATAL_FAILURE(CreateManager(true, 1000)); 3415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 342c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // Initialize the CloudPolicyService without any stored data. 343c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) EXPECT_FALSE(manager_->core()->service()->IsInitializationComplete()); 344c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) store_->NotifyStoreLoaded(); 345c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) EXPECT_TRUE(manager_->core()->service()->IsInitializationComplete()); 346c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) EXPECT_FALSE(manager_->core()->client()->is_registered()); 3475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 348c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // This starts the OAuth2 policy token fetcher using the signin Profile. 349c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // The manager will then issue the registration request. 350c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) MockDeviceManagementJob* register_request = IssueOAuthToken(false); 351c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) ASSERT_TRUE(register_request); 352c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 353c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // Reply with a valid registration response. This triggers the initial policy 354c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // fetch. 355c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) FetchPolicy(base::Bind(&MockDeviceManagementJob::SendResponse, 356c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) base::Unretained(register_request), 357c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) DM_STATUS_SUCCESS, register_blob_)); 358c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)} 359c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 360c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)TEST_F(UserCloudPolicyManagerChromeOSTest, BlockingRefreshFetch) { 361c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // Tests the initialization of a manager whose Profile is waiting for the 362c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // initial fetch, when a previously cached policy and DMToken already exist. 36358537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) ASSERT_NO_FATAL_FAILURE(CreateManager(true, 1000)); 364c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 365c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // Set the initially cached data and initialize the CloudPolicyService. 366c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // The initial policy fetch is issued using the cached DMToken. 367c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) store_->policy_.reset(new em::PolicyData(policy_data_)); 368c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) FetchPolicy(base::Bind(&MockCloudPolicyStore::NotifyStoreLoaded, 369c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) base::Unretained(store_))); 370c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)} 371c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 372c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)TEST_F(UserCloudPolicyManagerChromeOSTest, BlockingFetchStoreError) { 373c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // Tests the initialization of a manager whose Profile is waiting for the 374c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // initial fetch, when the initial store load fails. 37558537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) ASSERT_NO_FATAL_FAILURE(CreateManager(true, 1000)); 376c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 377c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // Initialize the CloudPolicyService without any stored data. 378c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) EXPECT_FALSE(manager_->core()->service()->IsInitializationComplete()); 379c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) store_->NotifyStoreError(); 380c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) EXPECT_TRUE(manager_->core()->service()->IsInitializationComplete()); 381c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) EXPECT_FALSE(manager_->core()->client()->is_registered()); 382c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 383c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // This starts the OAuth2 policy token fetcher using the signin Profile. 384c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // The manager will then issue the registration request. 385c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) MockDeviceManagementJob* register_request = IssueOAuthToken(false); 386c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) ASSERT_TRUE(register_request); 387c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 388c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // Reply with a valid registration response. This triggers the initial policy 389c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // fetch. 390c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) FetchPolicy(base::Bind(&MockDeviceManagementJob::SendResponse, 391c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) base::Unretained(register_request), 392c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) DM_STATUS_SUCCESS, register_blob_)); 393c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)} 394c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 395c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)TEST_F(UserCloudPolicyManagerChromeOSTest, BlockingFetchOAuthError) { 396c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // Tests the initialization of a manager whose Profile is waiting for the 397c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // initial fetch, when the OAuth2 token fetch fails. 39858537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) ASSERT_NO_FATAL_FAILURE(CreateManager(true, 1000)); 399c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 400c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // Initialize the CloudPolicyService without any stored data. 401c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) EXPECT_FALSE(manager_->core()->service()->IsInitializationComplete()); 402c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) store_->NotifyStoreLoaded(); 403c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) EXPECT_TRUE(manager_->core()->service()->IsInitializationComplete()); 404c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) EXPECT_FALSE(manager_->core()->client()->is_registered()); 405c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 406c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // This starts the OAuth2 policy token fetcher using the signin Profile. 407c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // The manager will initialize with no policy after the token fetcher fails. 408c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) EXPECT_CALL(observer_, OnUpdatePolicy(manager_.get())); 409c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 410c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // The PolicyOAuth2TokenFetcher posts delayed retries on some errors. This 411c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // data will make it fail immediately. 412c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) net::TestURLFetcher* fetcher = PrepareOAuthFetcher( 413c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) GaiaUrls::GetInstance()->client_login_to_oauth2_url()); 414c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) ASSERT_TRUE(fetcher); 415c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) fetcher->set_response_code(400); 416c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) fetcher->SetResponseString("Error=BadAuthentication"); 4172a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) EXPECT_FALSE(manager_->IsInitializationComplete(POLICY_DOMAIN_CHROME)); 418c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) fetcher->delegate()->OnURLFetchComplete(fetcher); 419c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) EXPECT_TRUE(manager_->IsInitializationComplete(POLICY_DOMAIN_CHROME)); 420c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) EXPECT_TRUE(PolicyBundle().Equals(manager_->policies())); 4215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) Mock::VerifyAndClearExpectations(&observer_); 422c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)} 423c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 424c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)TEST_F(UserCloudPolicyManagerChromeOSTest, BlockingFetchRegisterError) { 425c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // Tests the initialization of a manager whose Profile is waiting for the 426c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // initial fetch, when the device management registration fails. 42758537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) ASSERT_NO_FATAL_FAILURE(CreateManager(true, 1000)); 428c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 429c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // Initialize the CloudPolicyService without any stored data. 430c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) EXPECT_FALSE(manager_->core()->service()->IsInitializationComplete()); 431c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) store_->NotifyStoreError(); 432c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) EXPECT_TRUE(manager_->core()->service()->IsInitializationComplete()); 433c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) EXPECT_FALSE(manager_->core()->client()->is_registered()); 434c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 435c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // This starts the OAuth2 policy token fetcher using the signin Profile. 436c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // The manager will then issue the registration request. 437c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) MockDeviceManagementJob* register_request = IssueOAuthToken(false); 438c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) ASSERT_TRUE(register_request); 4395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 440c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // Now make it fail. 441c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) EXPECT_FALSE(manager_->IsInitializationComplete(POLICY_DOMAIN_CHROME)); 4425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_CALL(observer_, OnUpdatePolicy(manager_.get())); 443c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) register_request->SendResponse(DM_STATUS_TEMPORARY_UNAVAILABLE, 444c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) em::DeviceManagementResponse()); 4452a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) EXPECT_TRUE(manager_->IsInitializationComplete(POLICY_DOMAIN_CHROME)); 446c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) EXPECT_TRUE(PolicyBundle().Equals(manager_->policies())); 4475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) Mock::VerifyAndClearExpectations(&observer_); 4485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 4495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 450c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)TEST_F(UserCloudPolicyManagerChromeOSTest, BlockingFetchPolicyFetchError) { 451c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // Tests the initialization of a manager whose Profile is waiting for the 452c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // initial fetch, when the policy fetch request fails. 45358537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) ASSERT_NO_FATAL_FAILURE(CreateManager(true, 1000)); 454c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 455c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // Initialize the CloudPolicyService without any stored data. 456c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) EXPECT_FALSE(manager_->core()->service()->IsInitializationComplete()); 457c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) store_->NotifyStoreLoaded(); 458c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) EXPECT_TRUE(manager_->core()->service()->IsInitializationComplete()); 459c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) EXPECT_FALSE(manager_->core()->client()->is_registered()); 460c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 461c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // This starts the OAuth2 policy token fetcher using the signin Profile. 462c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // The manager will then issue the registration request. 463c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) MockDeviceManagementJob* register_request = IssueOAuthToken(false); 464c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) ASSERT_TRUE(register_request); 4655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 466c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // Reply with a valid registration response. This triggers the initial policy 467c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // fetch. 468c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) MockDeviceManagementJob* policy_request = NULL; 4695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_CALL(device_management_service_, 470a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles) CreateJob(DeviceManagementRequestJob::TYPE_POLICY_FETCH, _)) 471c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) .WillOnce(device_management_service_.CreateAsyncJob(&policy_request)); 472c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) register_request->SendResponse(DM_STATUS_SUCCESS, register_blob_); 473c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) Mock::VerifyAndClearExpectations(&device_management_service_); 474c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) ASSERT_TRUE(policy_request); 475c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) EXPECT_TRUE(manager_->core()->service()->IsInitializationComplete()); 476c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) EXPECT_TRUE(manager_->core()->client()->is_registered()); 4775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 478c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // Make the policy fetch fail. The observer gets 2 notifications: one from the 479c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // RefreshPolicies callback, and another from the OnClientError callback. 480c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // A single notification suffices for this edge case, but this behavior is 481c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // also correct and makes the implementation simpler. 4825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_CALL(observer_, OnUpdatePolicy(manager_.get())).Times(AtLeast(1)); 483c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) EXPECT_FALSE(manager_->IsInitializationComplete(POLICY_DOMAIN_CHROME)); 484c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) policy_request->SendResponse(DM_STATUS_TEMPORARY_UNAVAILABLE, 485c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) em::DeviceManagementResponse()); 4865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) Mock::VerifyAndClearExpectations(&observer_); 487c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) EXPECT_TRUE(manager_->IsInitializationComplete(POLICY_DOMAIN_CHROME)); 488c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) EXPECT_TRUE(PolicyBundle().Equals(manager_->policies())); 4895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 4905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 49158537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles)TEST_F(UserCloudPolicyManagerChromeOSTest, BlockingFetchTimeout) { 49258537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) // The blocking fetch should be abandoned after the timeout. 49358537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) ASSERT_NO_FATAL_FAILURE(CreateManager(true, 0)); 49458537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) 49558537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) // Initialize the CloudPolicyService without any stored data. 49658537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) EXPECT_FALSE(manager_->core()->service()->IsInitializationComplete()); 49758537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) store_->NotifyStoreLoaded(); 49858537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) EXPECT_TRUE(manager_->core()->service()->IsInitializationComplete()); 49958537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) EXPECT_FALSE(manager_->core()->client()->is_registered()); 50058537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) 50158537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) // Running the message loop should trigger the timeout. 50258537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) EXPECT_CALL(observer_, OnUpdatePolicy(manager_.get())).Times(AtLeast(1)); 50358537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) EXPECT_FALSE(manager_->IsInitializationComplete(POLICY_DOMAIN_CHROME)); 50458537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) base::RunLoop().RunUntilIdle(); 50558537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) Mock::VerifyAndClearExpectations(&observer_); 50658537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) EXPECT_TRUE(manager_->IsInitializationComplete(POLICY_DOMAIN_CHROME)); 50758537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) EXPECT_TRUE(PolicyBundle().Equals(manager_->policies())); 50858537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles)} 50958537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) 51058537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) 511c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)TEST_F(UserCloudPolicyManagerChromeOSTest, NonBlockingFirstFetch) { 512c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // Tests the first policy fetch request by a Profile that isn't managed. 51358537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) ASSERT_NO_FATAL_FAILURE(CreateManager(false, 1000)); 5145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 515c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // Initialize the CloudPolicyService without any stored data. Since the 516c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // manager is not waiting for the initial fetch, it will become initialized 517c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // once the store is ready. 518c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) EXPECT_FALSE(manager_->core()->service()->IsInitializationComplete()); 519c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) EXPECT_FALSE(manager_->IsInitializationComplete(POLICY_DOMAIN_CHROME)); 5205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_CALL(observer_, OnUpdatePolicy(manager_.get())); 521c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) store_->NotifyStoreLoaded(); 522c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) Mock::VerifyAndClearExpectations(&observer_); 523c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) EXPECT_TRUE(manager_->core()->service()->IsInitializationComplete()); 5242a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) EXPECT_TRUE(manager_->IsInitializationComplete(POLICY_DOMAIN_CHROME)); 525c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) EXPECT_FALSE(manager_->core()->client()->is_registered()); 526c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 527c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // The manager is waiting for the refresh token, and hasn't started any 528c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // fetchers. 529c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) EXPECT_FALSE(test_url_fetcher_factory_.GetFetcherByID(0)); 530c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 531f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) // Set a fake refresh token at the OAuth2TokenService. 532f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) FakeProfileOAuth2TokenService* token_service = 533f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) static_cast<FakeProfileOAuth2TokenService*>( 534f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) ProfileOAuth2TokenServiceFactory::GetForProfile(profile_)); 535c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) ASSERT_TRUE(token_service); 536f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) EXPECT_FALSE(token_service->RefreshTokenIsAvailable(kTestAccountId)); 537f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) token_service->IssueRefreshToken(kTestAccountId); 538f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles) EXPECT_TRUE(token_service->RefreshTokenIsAvailable(kTestAccountId)); 539c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 540c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // That should have notified the manager, which now issues the request for the 541c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // policy oauth token. 542c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) MockDeviceManagementJob* register_request = IssueOAuthToken(true); 543c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) ASSERT_TRUE(register_request); 544c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) register_request->SendResponse(DM_STATUS_SUCCESS, register_blob_); 545c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 546c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // The refresh scheduler takes care of the initial fetch for unmanaged users. 547d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) // Running the task runner issues the initial fetch. 548c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) FetchPolicy( 549d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) base::Bind(&base::TestSimpleTaskRunner::RunUntilIdle, task_runner_)); 550c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)} 551c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 552c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)TEST_F(UserCloudPolicyManagerChromeOSTest, NonBlockingRefreshFetch) { 553c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // Tests a non-blocking initial policy fetch for a Profile that already has 554c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // a cached DMToken. 55558537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles) ASSERT_NO_FATAL_FAILURE(CreateManager(false, 1000)); 556c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 557c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // Set the initially cached data and initialize the CloudPolicyService. 558c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // The initial policy fetch is issued using the cached DMToken. 559c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) EXPECT_FALSE(manager_->core()->service()->IsInitializationComplete()); 560c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) EXPECT_FALSE(manager_->IsInitializationComplete(POLICY_DOMAIN_CHROME)); 561c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) EXPECT_CALL(observer_, OnUpdatePolicy(manager_.get())); 562c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) store_->policy_.reset(new em::PolicyData(policy_data_)); 563c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) store_->NotifyStoreLoaded(); 5645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) Mock::VerifyAndClearExpectations(&observer_); 565c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) EXPECT_TRUE(manager_->core()->service()->IsInitializationComplete()); 566c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) EXPECT_TRUE(manager_->IsInitializationComplete(POLICY_DOMAIN_CHROME)); 567c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) EXPECT_TRUE(manager_->core()->client()->is_registered()); 568c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 569c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // The refresh scheduler takes care of the initial fetch for unmanaged users. 570d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) // Running the task runner issues the initial fetch. 571c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) FetchPolicy( 572d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) base::Bind(&base::TestSimpleTaskRunner::RunUntilIdle, task_runner_)); 5735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 5745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} // namespace policy 576