Index: chrome/browser/policy/mock_device_management_backend.h |
diff --git a/chrome/browser/policy/mock_device_management_backend.h b/chrome/browser/policy/mock_device_management_backend.h |
index a684e4fe19283da3b56593a270178b32038898c0..54c8c8fed8a0211e6486ba9aa160d4b2ac652191 100644 |
--- a/chrome/browser/policy/mock_device_management_backend.h |
+++ b/chrome/browser/policy/mock_device_management_backend.h |
@@ -9,6 +9,7 @@ |
#include <map> |
#include <string> |
+#include "base/time.h" |
#include "base/values.h" |
#include "chrome/browser/policy/device_management_backend.h" |
#include "chrome/browser/policy/proto/device_management_constants.h" |
@@ -46,6 +47,12 @@ class MockDeviceManagementBackend : public DeviceManagementBackend { |
const em::DevicePolicyRequest& request, |
DevicePolicyResponseDelegate* delegate)); |
+ MOCK_METHOD4(ProcessCloudPolicyRequest, void( |
+ const std::string& device_management_token, |
+ const std::string& device_id, |
+ const em::CloudPolicyRequest& request, |
+ DevicePolicyResponseDelegate* delegate)); |
+ |
private: |
DISALLOW_COPY_AND_ASSIGN(MockDeviceManagementBackend); |
}; |
@@ -73,6 +80,24 @@ ACTION_P2(MockDeviceManagementBackendSucceedBooleanPolicy, name, value) { |
arg3->HandlePolicyResponse(response); |
} |
+ACTION(MockDeviceManagementBackendSucceedSpdyCloudPolicy) { |
+ em::SignedCloudPolicyResponse signed_response; |
+ em::CloudPolicySettings* settings = signed_response.mutable_settings(); |
+ em::DisableSpdyProto* spdy_proto = settings->mutable_disablespdy(); |
+ spdy_proto->set_disablespdy(true); |
+ spdy_proto->mutable_policy_options()->set_mode(em::PolicyOptions::MANDATORY); |
+ signed_response.set_timestamp(base::Time::NowFromSystemTime().ToTimeT()); |
+ std::string serialized_signed_response; |
+ EXPECT_TRUE(signed_response.SerializeToString(&serialized_signed_response)); |
+ em::CloudPolicyResponse response; |
+ response.set_signed_response(serialized_signed_response); |
+ // TODO(jkummerow): Set proper certificate_chain and signature (when |
+ // implementing support for signature verification). |
+ response.set_signature("TODO"); |
+ response.add_certificate_chain("TODO"); |
+ arg3->HandleCloudPolicyResponse(response); |
+} |
+ |
ACTION_P(MockDeviceManagementBackendFailRegister, error) { |
arg3->OnError(error); |
} |