| Index: chrome/browser/chromeos/login/users/wallpaper/wallpaper_manager_policy_browsertest.cc
|
| diff --git a/chrome/browser/chromeos/login/users/wallpaper/wallpaper_manager_policy_browsertest.cc b/chrome/browser/chromeos/login/users/wallpaper/wallpaper_manager_policy_browsertest.cc
|
| index 11876dece636788f2032079ca2de334a349cd847..95e377c13ecd4798b6e9b35254eb0872c544bbe9 100644
|
| --- a/chrome/browser/chromeos/login/users/wallpaper/wallpaper_manager_policy_browsertest.cc
|
| +++ b/chrome/browser/chromeos/login/users/wallpaper/wallpaper_manager_policy_browsertest.cc
|
| @@ -134,18 +134,20 @@ class WallpaperManagerPolicyTest
|
| : LoginManagerTest(true),
|
| wallpaper_change_count_(0),
|
| fake_session_manager_client_(new FakeSessionManagerClient) {
|
| - testUsers_[0] = LoginManagerTest::kEnterpriseUser1;
|
| - testUsers_[1] = LoginManagerTest::kEnterpriseUser2;
|
| + testUsers_.push_back(
|
| + AccountId::FromUserEmail(LoginManagerTest::kEnterpriseUser1));
|
| + testUsers_.push_back(
|
| + AccountId::FromUserEmail(LoginManagerTest::kEnterpriseUser2));
|
| }
|
|
|
| scoped_ptr<policy::UserPolicyBuilder> GetUserPolicyBuilder(
|
| - const std::string& user_id) {
|
| + const AccountId& account_id) {
|
| scoped_ptr<policy::UserPolicyBuilder>
|
| user_policy_builder(new policy::UserPolicyBuilder());
|
| base::FilePath user_keys_dir;
|
| EXPECT_TRUE(PathService::Get(DIR_USER_POLICY_KEYS, &user_keys_dir));
|
| const std::string sanitized_user_id =
|
| - CryptohomeClient::GetStubSanitizedUsername(user_id);
|
| + CryptohomeClient::GetStubSanitizedUsername(account_id.GetUserEmail());
|
| const base::FilePath user_key_file =
|
| user_keys_dir.AppendASCII(sanitized_user_id)
|
| .AppendASCII("policy.pub");
|
| @@ -158,7 +160,7 @@ class WallpaperManagerPolicyTest
|
| reinterpret_cast<const char*>(user_key_bits.data()),
|
| user_key_bits.size()),
|
| static_cast<int>(user_key_bits.size()));
|
| - user_policy_builder->policy_data().set_username(user_id);
|
| + user_policy_builder->policy_data().set_username(account_id.GetUserEmail());
|
| return user_policy_builder.Pass();
|
| }
|
|
|
| @@ -233,7 +235,7 @@ class WallpaperManagerPolicyTest
|
| // empty |filename| to clear policy.
|
| void InjectPolicy(int user_number, const std::string& filename) {
|
| ASSERT_TRUE(user_number == 0 || user_number == 1);
|
| - const std::string user_id = testUsers_[user_number];
|
| + const AccountId& account_id = testUsers_[user_number];
|
| policy::UserPolicyBuilder* builder =
|
| user_policy_builders_[user_number].get();
|
| if (!filename.empty()) {
|
| @@ -243,9 +245,10 @@ class WallpaperManagerPolicyTest
|
| builder->payload().Clear();
|
| }
|
| builder->Build();
|
| - fake_session_manager_client_->set_user_policy(user_id, builder->GetBlob());
|
| - const user_manager::User* user = user_manager::UserManager::Get()->FindUser(
|
| - AccountId::FromUserEmail(user_id));
|
| + fake_session_manager_client_->set_user_policy(account_id.GetUserEmail(),
|
| + builder->GetBlob());
|
| + const user_manager::User* user =
|
| + user_manager::UserManager::Get()->FindUser(account_id);
|
| ASSERT_TRUE(user);
|
| policy::CloudPolicyStore* store = GetStoreForUser(user);
|
| ASSERT_TRUE(store);
|
| @@ -267,15 +270,15 @@ class WallpaperManagerPolicyTest
|
| int wallpaper_change_count_;
|
| scoped_ptr<policy::UserPolicyBuilder> user_policy_builders_[2];
|
| FakeSessionManagerClient* fake_session_manager_client_;
|
| - const char* testUsers_[2];
|
| + std::vector<AccountId> testUsers_;
|
|
|
| private:
|
| DISALLOW_COPY_AND_ASSIGN(WallpaperManagerPolicyTest);
|
| };
|
|
|
| IN_PROC_BROWSER_TEST_F(WallpaperManagerPolicyTest, PRE_SetResetClear) {
|
| - RegisterUser(testUsers_[0]);
|
| - RegisterUser(testUsers_[1]);
|
| + RegisterUser(testUsers_[0].GetUserEmail());
|
| + RegisterUser(testUsers_[1].GetUserEmail());
|
| StartupUtils::MarkOobeCompleted();
|
| }
|
|
|
| @@ -285,7 +288,7 @@ IN_PROC_BROWSER_TEST_F(WallpaperManagerPolicyTest, PRE_SetResetClear) {
|
| // reverts to default.
|
| IN_PROC_BROWSER_TEST_F(WallpaperManagerPolicyTest, SetResetClear) {
|
| wallpaper::WallpaperInfo info;
|
| - LoginUser(testUsers_[0]);
|
| + LoginUser(testUsers_[0].GetUserEmail());
|
| base::RunLoop().RunUntilIdle();
|
|
|
| // First user: Wait until default wallpaper has been loaded (happens
|
| @@ -326,14 +329,14 @@ IN_PROC_BROWSER_TEST_F(WallpaperManagerPolicyTest, SetResetClear) {
|
|
|
| IN_PROC_BROWSER_TEST_F(WallpaperManagerPolicyTest,
|
| DISABLED_PRE_PRE_PRE_WallpaperOnLoginScreen) {
|
| - RegisterUser(testUsers_[0]);
|
| - RegisterUser(testUsers_[1]);
|
| + RegisterUser(testUsers_[0].GetUserEmail());
|
| + RegisterUser(testUsers_[1].GetUserEmail());
|
| StartupUtils::MarkOobeCompleted();
|
| }
|
|
|
| IN_PROC_BROWSER_TEST_F(WallpaperManagerPolicyTest,
|
| DISABLED_PRE_PRE_WallpaperOnLoginScreen) {
|
| - LoginUser(testUsers_[0]);
|
| + LoginUser(testUsers_[0].GetUserEmail());
|
|
|
| // Wait until default wallpaper has been loaded.
|
| RunUntilWallpaperChangeCount(1);
|
| @@ -348,7 +351,7 @@ IN_PROC_BROWSER_TEST_F(WallpaperManagerPolicyTest,
|
|
|
| IN_PROC_BROWSER_TEST_F(WallpaperManagerPolicyTest,
|
| DISABLED_PRE_WallpaperOnLoginScreen) {
|
| - LoginUser(testUsers_[1]);
|
| + LoginUser(testUsers_[1].GetUserEmail());
|
|
|
| // Wait until default wallpaper has been loaded.
|
| RunUntilWallpaperChangeCount(1);
|
| @@ -379,12 +382,12 @@ IN_PROC_BROWSER_TEST_F(WallpaperManagerPolicyTest,
|
| }
|
|
|
| IN_PROC_BROWSER_TEST_F(WallpaperManagerPolicyTest, PRE_PRE_PersistOverLogout) {
|
| - RegisterUser(testUsers_[0]);
|
| + RegisterUser(testUsers_[0].GetUserEmail());
|
| StartupUtils::MarkOobeCompleted();
|
| }
|
|
|
| IN_PROC_BROWSER_TEST_F(WallpaperManagerPolicyTest, PRE_PersistOverLogout) {
|
| - LoginUser(testUsers_[0]);
|
| + LoginUser(testUsers_[0].GetUserEmail());
|
|
|
| // Wait until default wallpaper has been loaded.
|
| RunUntilWallpaperChangeCount(1);
|
| @@ -398,7 +401,7 @@ IN_PROC_BROWSER_TEST_F(WallpaperManagerPolicyTest, PRE_PersistOverLogout) {
|
| }
|
|
|
| IN_PROC_BROWSER_TEST_F(WallpaperManagerPolicyTest, PersistOverLogout) {
|
| - LoginUser(testUsers_[0]);
|
| + LoginUser(testUsers_[0].GetUserEmail());
|
|
|
| // Wait until wallpaper has been loaded.
|
| RunUntilWallpaperChangeCount(1);
|
|
|