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/affiliation.h" | 13 #include "chrome/browser/chromeos/login/users/affiliation.h" |
14 #include "chrome/browser/chromeos/login/users/avatar/mock_user_image_manager.h" | 14 #include "chrome/browser/chromeos/login/users/avatar/mock_user_image_manager.h" |
15 #include "chrome/browser/chromeos/login/users/chrome_user_manager.h" | 15 #include "chrome/browser/chromeos/login/users/chrome_user_manager.h" |
16 #include "components/signin/core/account_id/account_id.h" | |
16 #include "components/user_manager/user.h" | 17 #include "components/user_manager/user.h" |
17 #include "components/user_manager/user_image/user_image.h" | 18 #include "components/user_manager/user_image/user_image.h" |
18 #include "testing/gmock/include/gmock/gmock.h" | 19 #include "testing/gmock/include/gmock/gmock.h" |
19 | 20 |
20 namespace chromeos { | 21 namespace chromeos { |
21 | 22 |
22 class FakeSupervisedUserManager; | 23 class FakeSupervisedUserManager; |
23 | 24 |
24 class MockUserManager : public ChromeUserManager { | 25 class MockUserManager : public ChromeUserManager { |
25 public: | 26 public: |
26 MockUserManager(); | 27 MockUserManager(); |
27 virtual ~MockUserManager(); | 28 virtual ~MockUserManager(); |
28 | 29 |
29 MOCK_METHOD0(Shutdown, void(void)); | 30 MOCK_METHOD0(Shutdown, void(void)); |
30 MOCK_CONST_METHOD0(GetUsersAllowedForMultiProfile, | 31 MOCK_CONST_METHOD0(GetUsersAllowedForMultiProfile, |
31 user_manager::UserList(void)); | 32 user_manager::UserList(void)); |
32 MOCK_CONST_METHOD0(GetUsersAllowedForSupervisedUsersCreation, | 33 MOCK_CONST_METHOD0(GetUsersAllowedForSupervisedUsersCreation, |
33 user_manager::UserList(void)); | 34 user_manager::UserList(void)); |
34 MOCK_CONST_METHOD0(GetLoggedInUsers, const user_manager::UserList&(void)); | 35 MOCK_CONST_METHOD0(GetLoggedInUsers, const user_manager::UserList&(void)); |
35 MOCK_CONST_METHOD0(GetLRULoggedInUsers, const user_manager::UserList&(void)); | 36 MOCK_CONST_METHOD0(GetLRULoggedInUsers, const user_manager::UserList&(void)); |
36 MOCK_METHOD3(UserLoggedIn, void( | 37 MOCK_METHOD3(UserLoggedIn, void(const AccountId&, const std::string&, bool)); |
37 const std::string&, const std::string&, bool)); | 38 MOCK_METHOD1(SwitchActiveUser, void(const AccountId& account_id)); |
38 MOCK_METHOD1(SwitchActiveUser, void(const std::string& email)); | |
39 MOCK_METHOD0(SessionStarted, void(void)); | 39 MOCK_METHOD0(SessionStarted, void(void)); |
40 MOCK_METHOD2(RemoveUser, | 40 MOCK_METHOD2(RemoveUser, |
41 void(const std::string&, user_manager::RemoveUserDelegate*)); | 41 void(const AccountId&, user_manager::RemoveUserDelegate*)); |
42 MOCK_METHOD1(RemoveUserFromList, void(const std::string&)); | 42 MOCK_METHOD1(RemoveUserFromList, void(const AccountId&)); |
43 MOCK_CONST_METHOD1(IsKnownUser, bool(const std::string&)); | 43 MOCK_CONST_METHOD1(IsKnownUser, bool(const AccountId&)); |
44 MOCK_CONST_METHOD1(FindUser, const user_manager::User*(const std::string&)); | 44 MOCK_CONST_METHOD1(FindUser, const user_manager::User*(const AccountId&)); |
45 MOCK_METHOD1(FindUserAndModify, user_manager::User*(const std::string&)); | 45 MOCK_METHOD1(FindUserAndModify, user_manager::User*(const AccountId&)); |
46 MOCK_METHOD2(SaveUserOAuthStatus, | 46 MOCK_METHOD2(SaveUserOAuthStatus, |
47 void(const std::string&, user_manager::User::OAuthTokenStatus)); | 47 void(const AccountId&, user_manager::User::OAuthTokenStatus)); |
48 MOCK_METHOD2(SaveForceOnlineSignin, void(const std::string&, bool)); | 48 MOCK_METHOD2(SaveForceOnlineSignin, void(const AccountId&, bool)); |
49 MOCK_METHOD2(SaveUserDisplayName, void(const std::string&, | 49 MOCK_METHOD2(SaveUserDisplayName, |
50 const base::string16&)); | 50 void(const AccountId&, const base::string16&)); |
51 MOCK_METHOD2(UpdateUserAccountData, | 51 MOCK_METHOD2(UpdateUserAccountData, |
52 void(const std::string&, const UserAccountData&)); | 52 void(const AccountId&, const UserAccountData&)); |
53 MOCK_CONST_METHOD1(GetUserDisplayName, base::string16(const std::string&)); | 53 MOCK_CONST_METHOD1(GetUserDisplayName, base::string16(const AccountId&)); |
54 MOCK_METHOD2(SaveUserDisplayEmail, void(const std::string&, | 54 MOCK_METHOD2(SaveUserDisplayEmail, |
55 const std::string&)); | 55 void(const AccountId&, const std::string&)); |
56 MOCK_CONST_METHOD1(GetUserDisplayEmail, std::string(const std::string&)); | 56 MOCK_CONST_METHOD1(GetUserDisplayEmail, std::string(const AccountId&)); |
57 MOCK_CONST_METHOD0(IsCurrentUserOwner, bool(void)); | 57 MOCK_CONST_METHOD0(IsCurrentUserOwner, bool(void)); |
58 MOCK_CONST_METHOD0(IsCurrentUserNew, bool(void)); | 58 MOCK_CONST_METHOD0(IsCurrentUserNew, bool(void)); |
59 MOCK_CONST_METHOD0(IsCurrentUserNonCryptohomeDataEphemeral, bool(void)); | 59 MOCK_CONST_METHOD0(IsCurrentUserNonCryptohomeDataEphemeral, bool(void)); |
60 MOCK_CONST_METHOD0(CanCurrentUserLock, bool(void)); | 60 MOCK_CONST_METHOD0(CanCurrentUserLock, bool(void)); |
61 MOCK_CONST_METHOD0(IsUserLoggedIn, bool(void)); | 61 MOCK_CONST_METHOD0(IsUserLoggedIn, bool(void)); |
62 MOCK_CONST_METHOD0(IsLoggedInAsUserWithGaiaAccount, bool(void)); | 62 MOCK_CONST_METHOD0(IsLoggedInAsUserWithGaiaAccount, bool(void)); |
63 MOCK_CONST_METHOD0(IsLoggedInAsPublicAccount, bool(void)); | 63 MOCK_CONST_METHOD0(IsLoggedInAsPublicAccount, bool(void)); |
64 MOCK_CONST_METHOD0(IsLoggedInAsGuest, bool(void)); | 64 MOCK_CONST_METHOD0(IsLoggedInAsGuest, bool(void)); |
65 MOCK_CONST_METHOD0(IsLoggedInAsSupervisedUser, bool(void)); | 65 MOCK_CONST_METHOD0(IsLoggedInAsSupervisedUser, bool(void)); |
66 MOCK_CONST_METHOD0(IsLoggedInAsKioskApp, bool(void)); | 66 MOCK_CONST_METHOD0(IsLoggedInAsKioskApp, bool(void)); |
67 MOCK_CONST_METHOD0(IsLoggedInAsStub, bool(void)); | 67 MOCK_CONST_METHOD0(IsLoggedInAsStub, bool(void)); |
68 MOCK_CONST_METHOD0(IsSessionStarted, bool(void)); | 68 MOCK_CONST_METHOD0(IsSessionStarted, bool(void)); |
69 MOCK_CONST_METHOD1(IsUserNonCryptohomeDataEphemeral, | 69 MOCK_CONST_METHOD1(IsUserNonCryptohomeDataEphemeral, bool(const AccountId&)); |
70 bool(const std::string&)); | |
71 MOCK_METHOD1(AddObserver, void(UserManager::Observer*)); | 70 MOCK_METHOD1(AddObserver, void(UserManager::Observer*)); |
72 MOCK_METHOD1(RemoveObserver, void(UserManager::Observer*)); | 71 MOCK_METHOD1(RemoveObserver, void(UserManager::Observer*)); |
73 MOCK_METHOD1(AddSessionStateObserver, | 72 MOCK_METHOD1(AddSessionStateObserver, |
74 void(UserManager::UserSessionStateObserver*)); | 73 void(UserManager::UserSessionStateObserver*)); |
75 MOCK_METHOD1(RemoveSessionStateObserver, | 74 MOCK_METHOD1(RemoveSessionStateObserver, |
76 void(UserManager::UserSessionStateObserver*)); | 75 void(UserManager::UserSessionStateObserver*)); |
77 MOCK_METHOD0(NotifyLocalStateChanged, void(void)); | 76 MOCK_METHOD0(NotifyLocalStateChanged, void(void)); |
78 MOCK_CONST_METHOD0(AreSupervisedUsersAllowed, bool(void)); | 77 MOCK_CONST_METHOD0(AreSupervisedUsersAllowed, bool(void)); |
79 | 78 |
80 // UserManagerBase overrides: | 79 // UserManagerBase overrides: |
81 MOCK_CONST_METHOD0(AreEphemeralUsersEnabled, bool(void)); | 80 MOCK_CONST_METHOD0(AreEphemeralUsersEnabled, bool(void)); |
82 MOCK_CONST_METHOD0(GetApplicationLocale, const std::string&(void)); | 81 MOCK_CONST_METHOD0(GetApplicationLocale, const std::string&(void)); |
83 MOCK_CONST_METHOD0(GetLocalState, PrefService*(void)); | 82 MOCK_CONST_METHOD0(GetLocalState, PrefService*(void)); |
84 MOCK_CONST_METHOD2(HandleUserOAuthTokenStatusChange, | 83 MOCK_CONST_METHOD2(HandleUserOAuthTokenStatusChange, |
85 void(const std::string&, | 84 void(const AccountId&, |
86 user_manager::User::OAuthTokenStatus status)); | 85 user_manager::User::OAuthTokenStatus status)); |
87 MOCK_CONST_METHOD0(IsEnterpriseManaged, bool(void)); | 86 MOCK_CONST_METHOD0(IsEnterpriseManaged, bool(void)); |
88 MOCK_METHOD1(LoadPublicAccounts, void(std::set<std::string>*)); | 87 MOCK_METHOD1(LoadPublicAccounts, void(std::set<AccountId>*)); |
89 MOCK_METHOD0(PerformPreUserListLoadingActions, void(void)); | 88 MOCK_METHOD0(PerformPreUserListLoadingActions, void(void)); |
90 MOCK_METHOD0(PerformPostUserListLoadingActions, void(void)); | 89 MOCK_METHOD0(PerformPostUserListLoadingActions, void(void)); |
91 MOCK_METHOD1(PerformPostUserLoggedInActions, void(bool)); | 90 MOCK_METHOD1(PerformPostUserLoggedInActions, void(bool)); |
92 MOCK_CONST_METHOD1(IsDemoApp, bool(const std::string&)); | 91 MOCK_CONST_METHOD1(IsDemoApp, bool(const AccountId&)); |
93 MOCK_CONST_METHOD1(IsKioskApp, bool(const std::string&)); | 92 MOCK_CONST_METHOD1(IsKioskApp, bool(const AccountId&)); |
94 MOCK_CONST_METHOD1(IsPublicAccountMarkedForRemoval, bool(const std::string&)); | 93 MOCK_CONST_METHOD1(IsPublicAccountMarkedForRemoval, bool(const AccountId&)); |
95 MOCK_METHOD0(DemoAccountLoggedIn, void(void)); | 94 MOCK_METHOD0(DemoAccountLoggedIn, void(void)); |
96 MOCK_METHOD1(KioskAppLoggedIn, void(const std::string&)); | 95 MOCK_METHOD1(KioskAppLoggedIn, void(const AccountId&)); |
97 MOCK_METHOD1(PublicAccountUserLoggedIn, void(user_manager::User*)); | 96 MOCK_METHOD1(PublicAccountUserLoggedIn, void(user_manager::User*)); |
98 MOCK_METHOD1(SupervisedUserLoggedIn, void(const std::string&)); | 97 MOCK_METHOD1(SupervisedUserLoggedIn, void(const AccountId&)); |
99 MOCK_METHOD1(OnUserRemoved, void(const std::string&)); | 98 MOCK_METHOD1(OnUserRemoved, void(const AccountId&)); |
100 MOCK_METHOD2(SetUserAffiliation, | 99 MOCK_METHOD2(SetUserAffiliation, |
101 void(const std::string& user_id, | 100 void(const std::string& user_id, |
achuithb
2015/10/28 23:11:45
Why not AccountId conversion here?
Alexander Alekseev
2015/10/29 02:00:41
Because this is actually a part of ChromeUserManag
| |
102 const chromeos::AffiliationIDSet& user_affiliation_ids)); | 101 const chromeos::AffiliationIDSet& user_affiliation_ids)); |
103 | 102 |
104 // You can't mock these functions easily because nobody can create | 103 // You can't mock these functions easily because nobody can create |
105 // User objects but the ChromeUserManager and us. | 104 // User objects but the ChromeUserManager and us. |
106 const user_manager::UserList& GetUsers() const override; | 105 const user_manager::UserList& GetUsers() const override; |
107 const user_manager::User* GetLoggedInUser() const override; | 106 const user_manager::User* GetLoggedInUser() const override; |
108 user_manager::UserList GetUnlockUsers() const override; | 107 user_manager::UserList GetUnlockUsers() const override; |
109 const std::string& GetOwnerEmail() const override; | 108 const AccountId& GetOwnerAccountId() const override; |
110 user_manager::User* GetLoggedInUser() override; | 109 user_manager::User* GetLoggedInUser() override; |
111 const user_manager::User* GetActiveUser() const override; | 110 const user_manager::User* GetActiveUser() const override; |
112 user_manager::User* GetActiveUser() override; | 111 user_manager::User* GetActiveUser() override; |
113 const user_manager::User* GetPrimaryUser() const override; | 112 const user_manager::User* GetPrimaryUser() const override; |
114 | 113 |
115 // ChromeUserManager overrides: | 114 // ChromeUserManager overrides: |
116 BootstrapManager* GetBootstrapManager() override; | 115 BootstrapManager* GetBootstrapManager() override; |
117 MultiProfileUserController* GetMultiProfileUserController() override; | 116 MultiProfileUserController* GetMultiProfileUserController() override; |
118 UserImageManager* GetUserImageManager(const std::string& user_id) override; | 117 UserImageManager* GetUserImageManager(const AccountId& account_id) override; |
119 SupervisedUserManager* GetSupervisedUserManager() override; | 118 SupervisedUserManager* GetSupervisedUserManager() override; |
120 MOCK_METHOD2(SetUserFlow, void(const std::string&, UserFlow*)); | 119 MOCK_METHOD2(SetUserFlow, void(const AccountId&, UserFlow*)); |
121 MOCK_METHOD1(ResetUserFlow, void(const std::string&)); | 120 MOCK_METHOD1(ResetUserFlow, void(const AccountId&)); |
122 UserFlow* GetCurrentUserFlow() const override; | 121 UserFlow* GetCurrentUserFlow() const override; |
123 UserFlow* GetUserFlow(const std::string&) const override; | 122 UserFlow* GetUserFlow(const AccountId&) const override; |
124 | 123 |
125 bool ShouldReportUser(const std::string& user_id) const override; | 124 bool ShouldReportUser(const std::string& user_id) const override; |
achuithb
2015/10/28 23:11:45
Why not AccountId conversion here?
Alexander Alekseev
2015/10/29 02:00:41
Because this is a part of ChromeUserManager API, n
| |
126 | 125 |
127 // Sets a new User instance. Users previously created by this MockUserManager | 126 // Sets a new User instance. Users previously created by this MockUserManager |
128 // become invalid. | 127 // become invalid. |
129 void SetActiveUser(const std::string& email); | 128 void SetActiveUser(const AccountId& account_id); |
130 | 129 |
131 // Creates a new public session user. Users previously created by this | 130 // Creates a new public session user. Users previously created by this |
132 // MockUserManager become invalid. | 131 // MockUserManager become invalid. |
133 user_manager::User* CreatePublicAccountUser(const std::string& email); | 132 user_manager::User* CreatePublicAccountUser(const AccountId& account_id); |
134 | 133 |
135 // Creates a new kiosk app user. Users previously created by this | 134 // Creates a new kiosk app user. Users previously created by this |
136 // MockUserManager become invalid. | 135 // MockUserManager become invalid. |
137 user_manager::User* CreateKioskAppUser(const std::string& user_id); | 136 user_manager::User* CreateKioskAppUser(const AccountId& account_id); |
138 | 137 |
139 // Adds a new User instance to the back of the user list. Users previously | 138 // Adds a new User instance to the back of the user list. Users previously |
140 // created by this MockUserManager remain valid. The added User is not | 139 // created by this MockUserManager remain valid. The added User is not |
141 // affiliated with the domain, that owns the device. | 140 // affiliated with the domain, that owns the device. |
142 void AddUser(const std::string& email); | 141 void AddUser(const AccountId& account_id); |
143 | 142 |
144 // The same as AddUser, but allows specifying affiliation with the domain, | 143 // The same as AddUser, but allows specifying affiliation with the domain, |
145 // that owns the device. | 144 // that owns the device. |
146 void AddUserWithAffiliation(const std::string& email, bool is_affiliated); | 145 void AddUserWithAffiliation(const AccountId& account_id, bool is_affiliated); |
147 | 146 |
148 // Clears the user list and the active user. Users previously created by this | 147 // Clears the user list and the active user. Users previously created by this |
149 // MockUserManager become invalid. | 148 // MockUserManager become invalid. |
150 void ClearUserList(); | 149 void ClearUserList(); |
151 | 150 |
152 scoped_ptr<UserFlow> user_flow_; | 151 scoped_ptr<UserFlow> user_flow_; |
153 scoped_ptr<MockUserImageManager> user_image_manager_; | 152 scoped_ptr<MockUserImageManager> user_image_manager_; |
154 scoped_ptr<FakeSupervisedUserManager> supervised_user_manager_; | 153 scoped_ptr<FakeSupervisedUserManager> supervised_user_manager_; |
155 user_manager::UserList user_list_; | 154 user_manager::UserList user_list_; |
155 // TODO (alemate): remove temporary_owner_account_id_ as soon as | |
156 // User::GetAccountId will | |
157 // return constant reference. crbug.com/546863 | |
158 mutable AccountId temporary_owner_account_id_ = EmptyAccountId(); | |
156 }; | 159 }; |
157 | 160 |
158 } // namespace chromeos | 161 } // namespace chromeos |
159 | 162 |
160 #endif // CHROME_BROWSER_CHROMEOS_LOGIN_USERS_MOCK_USER_MANAGER_H_ | 163 #endif // CHROME_BROWSER_CHROMEOS_LOGIN_USERS_MOCK_USER_MANAGER_H_ |
OLD | NEW |