| 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 #ifndef CHROME_BROWSER_CHROMEOS_LOGIN_USERS_MOCK_USER_MANAGER_H_ | 5 #ifndef CHROME_BROWSER_CHROMEOS_LOGIN_USERS_MOCK_USER_MANAGER_H_ |
| 6 #define CHROME_BROWSER_CHROMEOS_LOGIN_USERS_MOCK_USER_MANAGER_H_ | 6 #define CHROME_BROWSER_CHROMEOS_LOGIN_USERS_MOCK_USER_MANAGER_H_ |
| 7 | 7 |
| 8 #include <string> | 8 #include <string> |
| 9 | 9 |
| 10 #include "base/files/file_path.h" | 10 #include "base/files/file_path.h" |
| 11 #include "base/memory/weak_ptr.h" | 11 #include "base/memory/weak_ptr.h" |
| 12 #include "chrome/browser/chromeos/login/user_flow.h" | 12 #include "chrome/browser/chromeos/login/user_flow.h" |
| 13 #include "chrome/browser/chromeos/login/users/avatar/mock_user_image_manager.h" | 13 #include "chrome/browser/chromeos/login/users/avatar/mock_user_image_manager.h" |
| 14 #include "chrome/browser/chromeos/login/users/user_manager.h" | 14 #include "chrome/browser/chromeos/login/users/chrome_user_manager.h" |
| 15 #include "components/user_manager/user.h" | 15 #include "components/user_manager/user.h" |
| 16 #include "components/user_manager/user_image/user_image.h" | 16 #include "components/user_manager/user_image/user_image.h" |
| 17 #include "testing/gmock/include/gmock/gmock.h" | 17 #include "testing/gmock/include/gmock/gmock.h" |
| 18 | 18 |
| 19 namespace chromeos { | 19 namespace chromeos { |
| 20 | 20 |
| 21 class FakeSupervisedUserManager; | 21 class FakeSupervisedUserManager; |
| 22 | 22 |
| 23 class MockUserManager : public UserManager { | 23 class MockUserManager : public ChromeUserManager { |
| 24 public: | 24 public: |
| 25 MockUserManager(); | 25 MockUserManager(); |
| 26 virtual ~MockUserManager(); | 26 virtual ~MockUserManager(); |
| 27 | 27 |
| 28 MOCK_METHOD0(Shutdown, void(void)); | 28 MOCK_METHOD0(Shutdown, void(void)); |
| 29 MOCK_CONST_METHOD0(GetUsersAdmittedForMultiProfile, | 29 MOCK_CONST_METHOD0(GetUsersAdmittedForMultiProfile, |
| 30 user_manager::UserList(void)); | 30 user_manager::UserList(void)); |
| 31 MOCK_CONST_METHOD0(GetLoggedInUsers, const user_manager::UserList&(void)); | 31 MOCK_CONST_METHOD0(GetLoggedInUsers, const user_manager::UserList&(void)); |
| 32 MOCK_CONST_METHOD0(GetLRULoggedInUsers, const user_manager::UserList&(void)); | 32 MOCK_CONST_METHOD0(GetLRULoggedInUsers, const user_manager::UserList&(void)); |
| 33 MOCK_METHOD3(UserLoggedIn, void( | 33 MOCK_METHOD3(UserLoggedIn, void( |
| 34 const std::string&, const std::string&, bool)); | 34 const std::string&, const std::string&, bool)); |
| 35 MOCK_METHOD1(SwitchActiveUser, void(const std::string& email)); | 35 MOCK_METHOD1(SwitchActiveUser, void(const std::string& email)); |
| 36 MOCK_METHOD0(SessionStarted, void(void)); | 36 MOCK_METHOD0(SessionStarted, void(void)); |
| 37 MOCK_METHOD2(RemoveUser, void(const std::string&, RemoveUserDelegate*)); | 37 MOCK_METHOD2(RemoveUser, |
| 38 void(const std::string&, user_manager::RemoveUserDelegate*)); |
| 38 MOCK_METHOD1(RemoveUserFromList, void(const std::string&)); | 39 MOCK_METHOD1(RemoveUserFromList, void(const std::string&)); |
| 39 MOCK_CONST_METHOD1(IsKnownUser, bool(const std::string&)); | 40 MOCK_CONST_METHOD1(IsKnownUser, bool(const std::string&)); |
| 40 MOCK_CONST_METHOD1(FindUser, const user_manager::User*(const std::string&)); | 41 MOCK_CONST_METHOD1(FindUser, const user_manager::User*(const std::string&)); |
| 41 MOCK_METHOD1(FindUserAndModify, user_manager::User*(const std::string&)); | 42 MOCK_METHOD1(FindUserAndModify, user_manager::User*(const std::string&)); |
| 42 MOCK_METHOD2(SaveUserOAuthStatus, | 43 MOCK_METHOD2(SaveUserOAuthStatus, |
| 43 void(const std::string&, user_manager::User::OAuthTokenStatus)); | 44 void(const std::string&, user_manager::User::OAuthTokenStatus)); |
| 44 MOCK_METHOD2(SaveForceOnlineSignin, void(const std::string&, bool)); | 45 MOCK_METHOD2(SaveForceOnlineSignin, void(const std::string&, bool)); |
| 45 MOCK_METHOD2(SaveUserDisplayName, void(const std::string&, | 46 MOCK_METHOD2(SaveUserDisplayName, void(const std::string&, |
| 46 const base::string16&)); | 47 const base::string16&)); |
| 47 MOCK_METHOD2(UpdateUserAccountData, | 48 MOCK_METHOD2(UpdateUserAccountData, |
| (...skipping 17 matching lines...) Expand all Loading... |
| 65 MOCK_CONST_METHOD0(IsSessionStarted, bool(void)); | 66 MOCK_CONST_METHOD0(IsSessionStarted, bool(void)); |
| 66 MOCK_CONST_METHOD1(IsUserNonCryptohomeDataEphemeral, | 67 MOCK_CONST_METHOD1(IsUserNonCryptohomeDataEphemeral, |
| 67 bool(const std::string&)); | 68 bool(const std::string&)); |
| 68 MOCK_METHOD1(AddObserver, void(UserManager::Observer*)); | 69 MOCK_METHOD1(AddObserver, void(UserManager::Observer*)); |
| 69 MOCK_METHOD1(RemoveObserver, void(UserManager::Observer*)); | 70 MOCK_METHOD1(RemoveObserver, void(UserManager::Observer*)); |
| 70 MOCK_METHOD1(AddSessionStateObserver, | 71 MOCK_METHOD1(AddSessionStateObserver, |
| 71 void(UserManager::UserSessionStateObserver*)); | 72 void(UserManager::UserSessionStateObserver*)); |
| 72 MOCK_METHOD1(RemoveSessionStateObserver, | 73 MOCK_METHOD1(RemoveSessionStateObserver, |
| 73 void(UserManager::UserSessionStateObserver*)); | 74 void(UserManager::UserSessionStateObserver*)); |
| 74 MOCK_METHOD0(NotifyLocalStateChanged, void(void)); | 75 MOCK_METHOD0(NotifyLocalStateChanged, void(void)); |
| 75 MOCK_METHOD2(SetUserFlow, void(const std::string&, UserFlow*)); | 76 MOCK_CONST_METHOD0(AreSupervisedUsersAllowed, bool(void)); |
| 76 MOCK_METHOD1(ResetUserFlow, void(const std::string&)); | |
| 77 | 77 |
| 78 MOCK_CONST_METHOD0(AreSupervisedUsersAllowed, bool(void)); | 78 // UserManagerBase overrides: |
| 79 MOCK_CONST_METHOD0(AreEphemeralUsersEnabled, bool(void)); |
| 80 MOCK_CONST_METHOD0(GetApplicationLocale, const std::string&(void)); |
| 81 MOCK_CONST_METHOD0(GetLocalState, PrefService*(void)); |
| 82 MOCK_CONST_METHOD2(HandleUserOAuthTokenStatusChange, |
| 83 void(const std::string&, |
| 84 user_manager::User::OAuthTokenStatus status)); |
| 85 MOCK_CONST_METHOD0(IsEnterpriseManaged, bool(void)); |
| 86 MOCK_METHOD1(LoadPublicAccounts, void(std::set<std::string>*)); |
| 87 MOCK_METHOD0(PerformPreUserListLoadingActions, void(void)); |
| 88 MOCK_METHOD0(PerformPostUserListLoadingActions, void(void)); |
| 89 MOCK_METHOD1(PerformPostUserLoggedInActions, void(bool)); |
| 90 MOCK_CONST_METHOD1(IsDemoApp, bool(const std::string&)); |
| 91 MOCK_CONST_METHOD1(IsKioskApp, bool(const std::string&)); |
| 92 MOCK_CONST_METHOD1(IsPublicAccountMarkedForRemoval, bool(const std::string&)); |
| 93 MOCK_METHOD0(DemoAccountLoggedIn, void(void)); |
| 94 MOCK_METHOD1(KioskAppLoggedIn, void(const std::string&)); |
| 95 MOCK_METHOD1(PublicAccountUserLoggedIn, void(user_manager::User*)); |
| 96 MOCK_METHOD0(RetailModeUserLoggedIn, void(void)); |
| 97 MOCK_METHOD1(SupervisedUserLoggedIn, void(const std::string&)); |
| 79 | 98 |
| 80 // You can't mock these functions easily because nobody can create | 99 // You can't mock these functions easily because nobody can create |
| 81 // User objects but the ChromeUserManager and us. | 100 // User objects but the ChromeUserManager and us. |
| 82 virtual const user_manager::UserList& GetUsers() const OVERRIDE; | 101 virtual const user_manager::UserList& GetUsers() const OVERRIDE; |
| 83 virtual const user_manager::User* GetLoggedInUser() const OVERRIDE; | 102 virtual const user_manager::User* GetLoggedInUser() const OVERRIDE; |
| 84 virtual user_manager::UserList GetUnlockUsers() const OVERRIDE; | 103 virtual user_manager::UserList GetUnlockUsers() const OVERRIDE; |
| 85 virtual const std::string& GetOwnerEmail() const OVERRIDE; | 104 virtual const std::string& GetOwnerEmail() const OVERRIDE; |
| 86 virtual user_manager::User* GetLoggedInUser() OVERRIDE; | 105 virtual user_manager::User* GetLoggedInUser() OVERRIDE; |
| 87 virtual const user_manager::User* GetActiveUser() const OVERRIDE; | 106 virtual const user_manager::User* GetActiveUser() const OVERRIDE; |
| 88 virtual user_manager::User* GetActiveUser() OVERRIDE; | 107 virtual user_manager::User* GetActiveUser() OVERRIDE; |
| 89 virtual const user_manager::User* GetPrimaryUser() const OVERRIDE; | 108 virtual const user_manager::User* GetPrimaryUser() const OVERRIDE; |
| 90 | 109 |
| 110 // ChromeUserManager overrides: |
| 91 virtual MultiProfileUserController* GetMultiProfileUserController() OVERRIDE; | 111 virtual MultiProfileUserController* GetMultiProfileUserController() OVERRIDE; |
| 92 virtual UserImageManager* GetUserImageManager( | 112 virtual UserImageManager* GetUserImageManager( |
| 93 const std::string& user_id) OVERRIDE; | 113 const std::string& user_id) OVERRIDE; |
| 94 virtual SupervisedUserManager* GetSupervisedUserManager() OVERRIDE; | 114 virtual SupervisedUserManager* GetSupervisedUserManager() OVERRIDE; |
| 95 | 115 MOCK_METHOD2(SetUserFlow, void(const std::string&, UserFlow*)); |
| 116 MOCK_METHOD1(ResetUserFlow, void(const std::string&)); |
| 96 virtual UserFlow* GetCurrentUserFlow() const OVERRIDE; | 117 virtual UserFlow* GetCurrentUserFlow() const OVERRIDE; |
| 97 virtual UserFlow* GetUserFlow(const std::string&) const OVERRIDE; | 118 virtual UserFlow* GetUserFlow(const std::string&) const OVERRIDE; |
| 98 | 119 |
| 99 // Sets a new User instance. Users previously created by this MockUserManager | 120 // Sets a new User instance. Users previously created by this MockUserManager |
| 100 // become invalid. | 121 // become invalid. |
| 101 void SetActiveUser(const std::string& email); | 122 void SetActiveUser(const std::string& email); |
| 102 | 123 |
| 103 // Creates a new public session user. Users previously created by this | 124 // Creates a new public session user. Users previously created by this |
| 104 // MockUserManager become invalid. | 125 // MockUserManager become invalid. |
| 105 user_manager::User* CreatePublicAccountUser(const std::string& email); | 126 user_manager::User* CreatePublicAccountUser(const std::string& email); |
| 106 | 127 |
| 107 // Adds a new User instance to the back of the user list. Users previously | 128 // Adds a new User instance to the back of the user list. Users previously |
| 108 // created by this MockUserManager remain valid. | 129 // created by this MockUserManager remain valid. |
| 109 void AddUser(const std::string& email); | 130 void AddUser(const std::string& email); |
| 110 | 131 |
| 111 // Clears the user list and the active user. Users previously created by this | 132 // Clears the user list and the active user. Users previously created by this |
| 112 // MockUserManager become invalid. | 133 // MockUserManager become invalid. |
| 113 void ClearUserList(); | 134 void ClearUserList(); |
| 114 | 135 |
| 115 scoped_ptr<UserFlow> user_flow_; | 136 scoped_ptr<UserFlow> user_flow_; |
| 116 scoped_ptr<MockUserImageManager> user_image_manager_; | 137 scoped_ptr<MockUserImageManager> user_image_manager_; |
| 117 scoped_ptr<FakeSupervisedUserManager> supervised_user_manager_; | 138 scoped_ptr<FakeSupervisedUserManager> supervised_user_manager_; |
| 118 user_manager::UserList user_list_; | 139 user_manager::UserList user_list_; |
| 119 }; | 140 }; |
| 120 | 141 |
| 121 } // namespace chromeos | 142 } // namespace chromeos |
| 122 | 143 |
| 123 #endif // CHROME_BROWSER_CHROMEOS_LOGIN_USERS_MOCK_USER_MANAGER_H_ | 144 #endif // CHROME_BROWSER_CHROMEOS_LOGIN_USERS_MOCK_USER_MANAGER_H_ |
| OLD | NEW |