| OLD | NEW |
| 1 // Copyright 2014 The Chromium Authors. All rights reserved. | 1 // Copyright 2014 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "chrome/browser/chromeos/login/users/fake_user_manager.h" | 5 #include "chrome/browser/chromeos/login/users/fake_user_manager.h" |
| 6 | 6 |
| 7 #include "base/task_runner.h" | 7 #include "base/task_runner.h" |
| 8 #include "chrome/browser/chromeos/login/users/fake_supervised_user_manager.h" | 8 #include "chrome/browser/chromeos/login/users/fake_supervised_user_manager.h" |
| 9 #include "chrome/browser/chromeos/profiles/profile_helper.h" | 9 #include "chrome/browser/chromeos/profiles/profile_helper.h" |
| 10 #include "chrome/grit/theme_resources.h" | 10 #include "chrome/grit/theme_resources.h" |
| 11 #include "components/user_manager/user_image/user_image.h" | 11 #include "components/user_manager/user_image/user_image.h" |
| 12 #include "components/user_manager/user_type.h" | 12 #include "components/user_manager/user_type.h" |
| 13 #include "ui/base/resource/resource_bundle.h" | 13 #include "ui/base/resource/resource_bundle.h" |
| 14 | 14 |
| 15 namespace { | 15 namespace { |
| 16 | 16 |
| 17 // As defined in /chromeos/dbus/cryptohome_client.cc. | |
| 18 static const char kUserIdHashSuffix[] = "-hash"; | |
| 19 | |
| 20 class FakeTaskRunner : public base::TaskRunner { | 17 class FakeTaskRunner : public base::TaskRunner { |
| 21 public: | 18 public: |
| 22 virtual bool PostDelayedTask(const tracked_objects::Location& from_here, | 19 virtual bool PostDelayedTask(const tracked_objects::Location& from_here, |
| 23 const base::Closure& task, | 20 const base::Closure& task, |
| 24 base::TimeDelta delay) OVERRIDE { | 21 base::TimeDelta delay) OVERRIDE { |
| 25 task.Run(); | 22 task.Run(); |
| 26 return true; | 23 return true; |
| 27 } | 24 } |
| 28 virtual bool RunsTasksOnCurrentThread() const OVERRIDE { return true; } | 25 virtual bool RunsTasksOnCurrentThread() const OVERRIDE { return true; } |
| 29 | 26 |
| 30 protected: | 27 protected: |
| 31 virtual ~FakeTaskRunner() {} | 28 virtual ~FakeTaskRunner() {} |
| 32 }; | 29 }; |
| 33 | 30 |
| 34 } // namespace | 31 } // namespace |
| 35 | 32 |
| 36 namespace chromeos { | 33 namespace chromeos { |
| 37 | 34 |
| 38 FakeUserManager::FakeUserManager() | 35 FakeUserManager::FakeUserManager() |
| 39 : ChromeUserManager(new FakeTaskRunner(), new FakeTaskRunner()), | 36 : ChromeUserManager(new FakeTaskRunner(), new FakeTaskRunner()), |
| 40 supervised_user_manager_(new FakeSupervisedUserManager), | 37 supervised_user_manager_(new FakeSupervisedUserManager), |
| 41 primary_user_(NULL), | 38 primary_user_(NULL), |
| 42 multi_profile_user_controller_(NULL) { | 39 multi_profile_user_controller_(NULL) { |
| 43 ProfileHelper::SetProfileToUserForTestingEnabled(true); | |
| 44 } | 40 } |
| 45 | 41 |
| 46 FakeUserManager::~FakeUserManager() { | 42 FakeUserManager::~FakeUserManager() { |
| 47 ProfileHelper::SetProfileToUserForTestingEnabled(false); | |
| 48 | |
| 49 // Can't use STLDeleteElements because of the private destructor of User. | 43 // Can't use STLDeleteElements because of the private destructor of User. |
| 50 for (user_manager::UserList::iterator it = user_list_.begin(); | 44 for (user_manager::UserList::iterator it = user_list_.begin(); |
| 51 it != user_list_.end(); | 45 it != user_list_.end(); |
| 52 it = user_list_.erase(it)) { | 46 it = user_list_.erase(it)) { |
| 53 delete *it; | 47 delete *it; |
| 54 } | 48 } |
| 55 } | 49 } |
| 56 | 50 |
| 57 const user_manager::User* FakeUserManager::AddUser(const std::string& email) { | 51 const user_manager::User* FakeUserManager::AddUser(const std::string& email) { |
| 58 user_manager::User* user = user_manager::User::CreateRegularUser(email); | 52 user_manager::User* user = user_manager::User::CreateRegularUser(email); |
| 59 user->set_username_hash(email + kUserIdHashSuffix); | 53 user->set_username_hash( |
| 54 ProfileHelper::Get()->GetUserIdHashByUserIdForTests(email)); |
| 60 user->SetStubImage(user_manager::UserImage( | 55 user->SetStubImage(user_manager::UserImage( |
| 61 *ResourceBundle::GetSharedInstance().GetImageSkiaNamed( | 56 *ResourceBundle::GetSharedInstance().GetImageSkiaNamed( |
| 62 IDR_PROFILE_PICTURE_LOADING)), | 57 IDR_PROFILE_PICTURE_LOADING)), |
| 63 user_manager::User::USER_IMAGE_PROFILE, | 58 user_manager::User::USER_IMAGE_PROFILE, |
| 64 false); | 59 false); |
| 65 user_list_.push_back(user); | 60 user_list_.push_back(user); |
| 66 ProfileHelper::Get()->SetProfileToUserMappingForTesting(user); | |
| 67 return user; | 61 return user; |
| 68 } | 62 } |
| 69 | 63 |
| 70 const user_manager::User* FakeUserManager::AddPublicAccountUser( | 64 const user_manager::User* FakeUserManager::AddPublicAccountUser( |
| 71 const std::string& email) { | 65 const std::string& email) { |
| 72 user_manager::User* user = user_manager::User::CreatePublicAccountUser(email); | 66 user_manager::User* user = user_manager::User::CreatePublicAccountUser(email); |
| 73 user->set_username_hash(email + kUserIdHashSuffix); | 67 user->set_username_hash( |
| 68 ProfileHelper::Get()->GetUserIdHashByUserIdForTests(email)); |
| 74 user->SetStubImage(user_manager::UserImage( | 69 user->SetStubImage(user_manager::UserImage( |
| 75 *ResourceBundle::GetSharedInstance().GetImageSkiaNamed( | 70 *ResourceBundle::GetSharedInstance().GetImageSkiaNamed( |
| 76 IDR_PROFILE_PICTURE_LOADING)), | 71 IDR_PROFILE_PICTURE_LOADING)), |
| 77 user_manager::User::USER_IMAGE_PROFILE, | 72 user_manager::User::USER_IMAGE_PROFILE, |
| 78 false); | 73 false); |
| 79 user_list_.push_back(user); | 74 user_list_.push_back(user); |
| 80 ProfileHelper::Get()->SetProfileToUserMappingForTesting(user); | |
| 81 return user; | 75 return user; |
| 82 } | 76 } |
| 83 | 77 |
| 84 void FakeUserManager::AddKioskAppUser(const std::string& kiosk_app_username) { | 78 void FakeUserManager::AddKioskAppUser(const std::string& kiosk_app_username) { |
| 85 user_manager::User* user = | 79 user_manager::User* user = |
| 86 user_manager::User::CreateKioskAppUser(kiosk_app_username); | 80 user_manager::User::CreateKioskAppUser(kiosk_app_username); |
| 87 user->set_username_hash(kiosk_app_username + kUserIdHashSuffix); | 81 user->set_username_hash( |
| 82 ProfileHelper::Get()->GetUserIdHashByUserIdForTests(kiosk_app_username)); |
| 88 user_list_.push_back(user); | 83 user_list_.push_back(user); |
| 89 ProfileHelper::Get()->SetProfileToUserMappingForTesting(user); | |
| 90 } | 84 } |
| 91 | 85 |
| 92 void FakeUserManager::RemoveUserFromList(const std::string& email) { | 86 void FakeUserManager::RemoveUserFromList(const std::string& email) { |
| 93 user_manager::UserList::iterator it = user_list_.begin(); | 87 user_manager::UserList::iterator it = user_list_.begin(); |
| 94 while (it != user_list_.end() && (*it)->email() != email) ++it; | 88 while (it != user_list_.end() && (*it)->email() != email) ++it; |
| 95 if (it != user_list_.end()) { | 89 if (it != user_list_.end()) { |
| 96 delete *it; | 90 delete *it; |
| 97 user_list_.erase(it); | 91 user_list_.erase(it); |
| 98 } | 92 } |
| 99 } | 93 } |
| 100 | 94 |
| 101 void FakeUserManager::LoginUser(const std::string& email) { | 95 void FakeUserManager::LoginUser(const std::string& email) { |
| 102 UserLoggedIn(email, email + kUserIdHashSuffix, false); | 96 UserLoggedIn( |
| 97 email, ProfileHelper::Get()->GetUserIdHashByUserIdForTests(email), false); |
| 103 } | 98 } |
| 104 | 99 |
| 105 const user_manager::UserList& FakeUserManager::GetUsers() const { | 100 const user_manager::UserList& FakeUserManager::GetUsers() const { |
| 106 return user_list_; | 101 return user_list_; |
| 107 } | 102 } |
| 108 | 103 |
| 109 user_manager::UserList FakeUserManager::GetUsersAdmittedForMultiProfile() | 104 user_manager::UserList FakeUserManager::GetUsersAdmittedForMultiProfile() |
| 110 const { | 105 const { |
| 111 user_manager::UserList result; | 106 user_manager::UserList result; |
| 112 for (user_manager::UserList::const_iterator it = user_list_.begin(); | 107 for (user_manager::UserList::const_iterator it = user_list_.begin(); |
| (...skipping 11 matching lines...) Expand all Loading... |
| 124 } | 119 } |
| 125 | 120 |
| 126 void FakeUserManager::UserLoggedIn(const std::string& email, | 121 void FakeUserManager::UserLoggedIn(const std::string& email, |
| 127 const std::string& username_hash, | 122 const std::string& username_hash, |
| 128 bool browser_restart) { | 123 bool browser_restart) { |
| 129 for (user_manager::UserList::const_iterator it = user_list_.begin(); | 124 for (user_manager::UserList::const_iterator it = user_list_.begin(); |
| 130 it != user_list_.end(); | 125 it != user_list_.end(); |
| 131 ++it) { | 126 ++it) { |
| 132 if ((*it)->username_hash() == username_hash) { | 127 if ((*it)->username_hash() == username_hash) { |
| 133 (*it)->set_is_logged_in(true); | 128 (*it)->set_is_logged_in(true); |
| 129 (*it)->set_profile_is_created(); |
| 134 logged_in_users_.push_back(*it); | 130 logged_in_users_.push_back(*it); |
| 135 | 131 |
| 136 if (!primary_user_) | 132 if (!primary_user_) |
| 137 primary_user_ = *it; | 133 primary_user_ = *it; |
| 138 break; | 134 break; |
| 139 } | 135 } |
| 140 } | 136 } |
| 141 } | 137 } |
| 142 | 138 |
| 143 user_manager::User* FakeUserManager::GetActiveUserInternal() const { | 139 user_manager::User* FakeUserManager::GetActiveUserInternal() const { |
| (...skipping 14 matching lines...) Expand all Loading... |
| 158 const user_manager::User* FakeUserManager::GetActiveUser() const { | 154 const user_manager::User* FakeUserManager::GetActiveUser() const { |
| 159 return GetActiveUserInternal(); | 155 return GetActiveUserInternal(); |
| 160 } | 156 } |
| 161 | 157 |
| 162 user_manager::User* FakeUserManager::GetActiveUser() { | 158 user_manager::User* FakeUserManager::GetActiveUser() { |
| 163 return GetActiveUserInternal(); | 159 return GetActiveUserInternal(); |
| 164 } | 160 } |
| 165 | 161 |
| 166 void FakeUserManager::SwitchActiveUser(const std::string& email) { | 162 void FakeUserManager::SwitchActiveUser(const std::string& email) { |
| 167 active_user_id_ = email; | 163 active_user_id_ = email; |
| 164 ProfileHelper::Get()->ActiveUserHashChanged( |
| 165 ProfileHelper::Get()->GetUserIdHashByUserIdForTests(email)); |
| 168 } | 166 } |
| 169 | 167 |
| 170 void FakeUserManager::SaveUserDisplayName( | 168 void FakeUserManager::SaveUserDisplayName( |
| 171 const std::string& username, | 169 const std::string& username, |
| 172 const base::string16& display_name) { | 170 const base::string16& display_name) { |
| 173 for (user_manager::UserList::iterator it = user_list_.begin(); | 171 for (user_manager::UserList::iterator it = user_list_.begin(); |
| 174 it != user_list_.end(); | 172 it != user_list_.end(); |
| 175 ++it) { | 173 ++it) { |
| 176 if ((*it)->email() == username) { | 174 if ((*it)->email() == username) { |
| 177 (*it)->set_display_name(display_name); | 175 (*it)->set_display_name(display_name); |
| (...skipping 166 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 344 bool FakeUserManager::IsKioskApp(const std::string& user_id) const { | 342 bool FakeUserManager::IsKioskApp(const std::string& user_id) const { |
| 345 return false; | 343 return false; |
| 346 } | 344 } |
| 347 | 345 |
| 348 bool FakeUserManager::IsPublicAccountMarkedForRemoval( | 346 bool FakeUserManager::IsPublicAccountMarkedForRemoval( |
| 349 const std::string& user_id) const { | 347 const std::string& user_id) const { |
| 350 return false; | 348 return false; |
| 351 } | 349 } |
| 352 | 350 |
| 353 } // namespace chromeos | 351 } // namespace chromeos |
| OLD | NEW |