Refresh policies from DM server periodically

chrome/browser/policy/mock_device_management_backend.cc

 // Copyright (c) 2010 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "chrome/browser/policy/mock_device_management_backend.h"
 
 #include "base/basictypes.h"
 #include "base/logging.h"
 #include "base/stl_util-inl.h"
 
@@ skipping 10 matching lines @@
 using ::testing::Invoke;
 
 using em::DeviceRegisterRequest;
 using em::DeviceUnregisterRequest;
 using em::DevicePolicyRequest;
 using em::DeviceRegisterResponse;
 using em::DeviceUnregisterResponse;
 using em::DevicePolicyResponse;
 
 MockDeviceManagementBackend::MockDeviceManagementBackend()
-    : DeviceManagementBackend() {
+    : DeviceManagementBackend(),
+      policy_remaining_fail_count_(0),
+      register_remaining_fail_count_(0),
+      policy_remaining_success_count_(0) {
   policy_setting_ = policy_response_.add_setting();
   policy_setting_->set_policy_key("chrome-policy");
   policy_setting_->set_watermark("fresh");
 }
 
 MockDeviceManagementBackend::~MockDeviceManagementBackend() {
 }
 
 void MockDeviceManagementBackend::AllShouldSucceed() {
   ON_CALL(*this, ProcessRegisterRequest(_, _, _, _)).
@@ skipping 17 matching lines @@
           &MockDeviceManagementBackend::SimulateFailedPolicyRequest));
 }
 
 void MockDeviceManagementBackend::UnmanagedDevice() {
   ON_CALL(*this, ProcessRegisterRequest(_, _, _, _)).
       WillByDefault(Invoke(
           this,
           &MockDeviceManagementBackend::SimulateUnmanagedRegisterRequest));
 }
 
+void MockDeviceManagementBackend::RegisterFailsOncePolicyFailsTwice() {
+  register_remaining_fail_count_ = 1;
+  policy_remaining_fail_count_ = 2;
+  AllShouldFail();
+}
+
+void MockDeviceManagementBackend::AllWorksFirstPolicyFailsLater() {
+  policy_remaining_success_count_ = 3;
+  AllShouldSucceed();
+}
+
 void MockDeviceManagementBackend::AddBooleanPolicy(const char* policy_name,
                                                    bool value) {
   em::GenericSetting* policy_value = policy_setting_->mutable_policy_value();
   em::GenericNamedValue* named_value = policy_value->add_named_value();
   named_value->set_name(policy_name);
   named_value->mutable_value()->set_value_type(
       em::GenericValue::VALUE_TYPE_BOOL);
   named_value->mutable_value()->set_bool_value(value);
 }
 
 void MockDeviceManagementBackend::SimulateSuccessfulRegisterRequest(
     const std::string& auth_token,
     const std::string& device_id,
     const em::DeviceRegisterRequest& request,
     DeviceRegisterResponseDelegate* delegate) {
   DeviceRegisterResponse response;
   std::string token(kFakeDeviceManagementToken);
   token += next_token_suffix_++;
   response.set_device_management_token(token);
   delegate->HandleRegisterResponse(response);
 }
 
 void MockDeviceManagementBackend::SimulateSuccessfulPolicyRequest(
     const std::string& device_management_token,
     const std::string& device_id,
     const em::DevicePolicyRequest& request,
     DevicePolicyResponseDelegate* delegate) {
+  if (policy_remaining_success_count_) {
+    policy_remaining_success_count_--;
+    if (!policy_remaining_success_count_) {
+      ON_CALL(*this, ProcessPolicyRequest(_, _, _, _)).
+          WillByDefault(Invoke(
+              this,
+              &MockDeviceManagementBackend::SimulateFailedPolicyRequest));
+    }
+  }
   delegate->HandlePolicyResponse(policy_response_);
 }
 
 void MockDeviceManagementBackend::SimulateFailedRegisterRequest(
     const std::string& auth_token,
     const std::string& device_id,
     const em::DeviceRegisterRequest& request,
     DeviceRegisterResponseDelegate* delegate) {
+  if (register_remaining_fail_count_) {
+    register_remaining_fail_count_--;
+    if (!register_remaining_fail_count_) {
+      ON_CALL(*this, ProcessRegisterRequest(_, _, _, _)).WillByDefault(Invoke(
+          this,
+          &MockDeviceManagementBackend::SimulateSuccessfulRegisterRequest));
+    }
+  }
   delegate->OnError(kErrorRequestFailed);
 }
 
 void MockDeviceManagementBackend::SimulateFailedPolicyRequest(
     const std::string& device_management_token,
     const std::string& device_id,
     const em::DevicePolicyRequest& request,
     DevicePolicyResponseDelegate* delegate) {
+  if (policy_remaining_fail_count_) {
+    policy_remaining_fail_count_--;
+    if (!policy_remaining_fail_count_) {
+      ON_CALL(*this, ProcessPolicyRequest(_, _, _, _)).
+          WillByDefault(Invoke(
+              this,
+              &MockDeviceManagementBackend::SimulateSuccessfulPolicyRequest));
+    }
+  }
   delegate->OnError(kErrorRequestFailed);
 }
 
 void MockDeviceManagementBackend::SimulateUnmanagedRegisterRequest(
     const std::string& auth_token,
     const std::string& device_id,
     const em::DeviceRegisterRequest& request,
     DeviceRegisterResponseDelegate* delegate) {
   delegate->OnError(kErrorServiceManagementNotSupported);
 }
 
 }  // namespace