Index: chrome/browser/chromeos/login/users/mock_user_manager.h |
diff --git a/chrome/browser/chromeos/login/users/mock_user_manager.h b/chrome/browser/chromeos/login/users/mock_user_manager.h |
index c7545bc6b6e7dcfbf8a7c085e3fb82fc20e23868..2b4af299dff2acef87ff84e327a67d5b08e20509 100644 |
--- a/chrome/browser/chromeos/login/users/mock_user_manager.h |
+++ b/chrome/browser/chromeos/login/users/mock_user_manager.h |
@@ -10,6 +10,7 @@ |
#include "base/files/file_path.h" |
#include "base/memory/weak_ptr.h" |
#include "chrome/browser/chromeos/login/user_flow.h" |
+#include "chrome/browser/chromeos/login/users/affiliation.h" |
#include "chrome/browser/chromeos/login/users/avatar/mock_user_image_manager.h" |
#include "chrome/browser/chromeos/login/users/chrome_user_manager.h" |
#include "components/user_manager/user.h" |
@@ -95,6 +96,9 @@ class MockUserManager : public ChromeUserManager { |
MOCK_METHOD1(KioskAppLoggedIn, void(const std::string&)); |
MOCK_METHOD1(PublicAccountUserLoggedIn, void(user_manager::User*)); |
MOCK_METHOD1(SupervisedUserLoggedIn, void(const std::string&)); |
+ MOCK_METHOD2(SetUserAffiliation, |
+ void(const std::string& user_id, |
+ const chromeos::AffiliationIDSet& user_affiliation_ids)); |
// You can't mock these functions easily because nobody can create |
// User objects but the ChromeUserManager and us. |