chrome/browser/chromeos/login/users/mock_user_manager.cc

   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
   3 // found in the LICENSE file.
   4
   5 #include "chrome/browser/chromeos/login/users/mock_user_manager.h"
   6
   7 #include "base/task_runner.h"
   8 #include "chrome/browser/chromeos/login/users/fake_supervised_user_manager.h"
   9 #include "chrome/browser/chromeos/profiles/profile_helper.h"
  10
  11 namespace {
  12
  13 class FakeTaskRunner : public base::TaskRunner {
  14  public:
  15   bool PostDelayedTask(const tracked_objects::Location& from_here,
  16                        const base::Closure& task,
  17                        base::TimeDelta delay) override {
  18     task.Run();
  19     return true;
  20   }
  21   bool RunsTasksOnCurrentThread() const override { return true; }
  22
  23  protected:
  24   ~FakeTaskRunner() override {}
  25 };
  26
  27 }  // namespace
  28
  29 namespace chromeos {
  30
  31 MockUserManager::MockUserManager()
  32     : ChromeUserManager(new FakeTaskRunner(), new FakeTaskRunner()),
  33       user_flow_(new DefaultUserFlow()),
  34       supervised_user_manager_(new FakeSupervisedUserManager()) {
  35   ProfileHelper::SetProfileToUserForTestingEnabled(true);
  36 }
  37
  38 MockUserManager::~MockUserManager() {
  39   ProfileHelper::SetProfileToUserForTestingEnabled(false);
  40   ClearUserList();
  41 }
  42
  43 const user_manager::UserList& MockUserManager::GetUsers() const {
  44   return user_list_;
  45 }
  46
  47 const user_manager::User* MockUserManager::GetLoggedInUser() const {
  48   return user_list_.empty() ? NULL : user_list_.front();
  49 }
  50
  51 user_manager::User* MockUserManager::GetLoggedInUser() {
  52   return user_list_.empty() ? NULL : user_list_.front();
  53 }
  54
  55 user_manager::UserList MockUserManager::GetUnlockUsers() const {
  56   return user_list_;
  57 }
  58
  59 const std::string& MockUserManager::GetOwnerEmail() const {
  60   return GetLoggedInUser()->email();
  61 }
  62
  63 const user_manager::User* MockUserManager::GetActiveUser() const {
  64   return GetLoggedInUser();
  65 }
  66
  67 user_manager::User* MockUserManager::GetActiveUser() {
  68   return GetLoggedInUser();
  69 }
  70
  71 const user_manager::User* MockUserManager::GetPrimaryUser() const {
  72   return GetLoggedInUser();
  73 }
  74
  75 BootstrapManager* MockUserManager::GetBootstrapManager() {
  76   return NULL;
  77 }
  78
  79 MultiProfileUserController* MockUserManager::GetMultiProfileUserController() {
  80   return NULL;
  81 }
  82
  83 UserImageManager* MockUserManager::GetUserImageManager(
  84     const std::string& user_id) {
  85   return NULL;
  86 }
  87
  88 SupervisedUserManager* MockUserManager::GetSupervisedUserManager() {
  89   return supervised_user_manager_.get();
  90 }
  91
  92 // Creates a new User instance.
  93 void MockUserManager::SetActiveUser(const std::string& email) {
  94   ClearUserList();
  95   AddUser(email);
  96 }
  97
  98 UserFlow* MockUserManager::GetCurrentUserFlow() const {
  99   return user_flow_.get();
 100 }
 101
 102 UserFlow* MockUserManager::GetUserFlow(const std::string&) const {
 103   return user_flow_.get();
 104 }
 105
 106 user_manager::User* MockUserManager::CreatePublicAccountUser(
 107     const std::string& email) {
 108   ClearUserList();
 109   user_manager::User* user = user_manager::User::CreatePublicAccountUser(email);
 110   user_list_.push_back(user);
 111   ProfileHelper::Get()->SetProfileToUserMappingForTesting(user);
 112   return user_list_.back();
 113 }
 114
 115 user_manager::User* MockUserManager::CreateKioskAppUser(
 116     const std::string& email) {
 117   ClearUserList();
 118   user_list_.push_back(user_manager::User::CreateKioskAppUser(email));
 119   ProfileHelper::Get()->SetProfileToUserMappingForTesting(user_list_.back());
 120   return user_list_.back();
 121 }
 122
 123 void MockUserManager::AddUser(const std::string& email) {
 124   AddUserWithAffiliation(email, false);
 125 }
 126
 127 void MockUserManager::AddUserWithAffiliation(const std::string& email,
 128                                              bool is_affiliated) {
 129   user_manager::User* user = user_manager::User::CreateRegularUser(email);
 130   user->set_affiliation(is_affiliated);
 131   user_list_.push_back(user);
 132   ProfileHelper::Get()->SetProfileToUserMappingForTesting(user);
 133 }
 134
 135 void MockUserManager::ClearUserList() {
 136   // Can't use STLDeleteElements because of the protected destructor of User.
 137   user_manager::UserList::iterator user;
 138   for (user = user_list_.begin(); user != user_list_.end(); ++user)
 139     delete *user;
 140   user_list_.clear();
 141 }
 142
 143 bool MockUserManager::ShouldReportUser(const std::string& user_id) const {
 144   for (const auto& user : user_list_) {
 145     if (user->GetUserID() == user_id) {
 146       return user->is_affiliated();
 147     }
 148   }
 149   NOTREACHED();
 150   return false;
 151 }
 152
 153 }  // namespace chromeos