| 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);
|
|
|