Index: chrome/browser/chromeos/policy/device_local_account_browsertest.cc |
diff --git a/chrome/browser/chromeos/policy/device_local_account_browsertest.cc b/chrome/browser/chromeos/policy/device_local_account_browsertest.cc |
index b7f2426823a833ef4efd9d6d4fbf9e43739cf728..dedbd23debaa9198ff9369554b82a5b9565888f6 100644 |
--- a/chrome/browser/chromeos/policy/device_local_account_browsertest.cc |
+++ b/chrome/browser/chromeos/policy/device_local_account_browsertest.cc |
@@ -137,10 +137,11 @@ class FakeSessionManagerClient : public chromeos::SessionManagerClient { |
MessageLoop::current()->PostTask(FROM_HERE, |
base::Bind(callback, device_policy_)); |
} |
- virtual void RetrieveUserPolicy( |
+ virtual void RetrievePolicyForUser( |
+ const std::string& username, |
const RetrievePolicyCallback& callback) OVERRIDE { |
- MessageLoop::current()->PostTask(FROM_HERE, |
- base::Bind(callback, user_policy_)); |
+ MessageLoop::current()->PostTask( |
+ FROM_HERE, base::Bind(callback, user_policies_[username])); |
} |
virtual void RetrieveDeviceLocalAccountPolicy( |
const std::string& account_id, |
@@ -154,9 +155,12 @@ class FakeSessionManagerClient : public chromeos::SessionManagerClient { |
device_policy_ = policy_blob; |
MessageLoop::current()->PostTask(FROM_HERE, base::Bind(callback, true)); |
} |
- virtual void StoreUserPolicy(const std::string& policy_blob, |
- const StorePolicyCallback& callback) OVERRIDE { |
- user_policy_ = policy_blob; |
+ virtual void StorePolicyForUser( |
+ const std::string& username, |
+ const std::string& policy_blob, |
+ const std::string& policy_key, |
+ const StorePolicyCallback& callback) OVERRIDE { |
+ user_policies_[username] = policy_blob; |
MessageLoop::current()->PostTask(FROM_HERE, base::Bind(callback, true)); |
} |
virtual void StoreDeviceLocalAccountPolicy( |
@@ -174,11 +178,14 @@ class FakeSessionManagerClient : public chromeos::SessionManagerClient { |
device_policy_ = policy_blob; |
} |
- const std::string& user_policy() const { |
- return user_policy_; |
+ const std::string& user_policy(const std::string& username) const { |
+ std::map<std::string, std::string>::const_iterator it = |
+ user_policies_.find(username); |
+ return it == user_policies_.end() ? EmptyString() : it->second; |
} |
- void set_user_policy(const std::string& policy_blob) { |
- user_policy_ = policy_blob; |
+ void set_user_policy(const std::string& username, |
+ const std::string& policy_blob) { |
+ user_policies_[username] = policy_blob; |
} |
const std::string& device_local_account_policy( |
@@ -195,7 +202,7 @@ class FakeSessionManagerClient : public chromeos::SessionManagerClient { |
private: |
std::string device_policy_; |
- std::string user_policy_; |
+ std::map<std::string, std::string> user_policies_; |
std::map<std::string, std::string> device_local_account_policy_; |
DISALLOW_COPY_AND_ASSIGN(FakeSessionManagerClient); |