From ffe2d339d60f9b7d1b3a59f523643366e6fb4dcd Mon Sep 17 00:00:00 2001
From: "danno@chromium.org"
 <danno@chromium.org@0039d316-1c4b-4281-b951-d872f2087c98>
Date: Tue, 9 Nov 2010 13:08:29 +0000
Subject: Implement device management policy provider

BUG=62462
TEST=DeviceManagementPolicyProviderTest*

Review URL: http://codereview.chromium.org/4382001

git-svn-id: svn://svn.chromium.org/chrome/trunk/src@65520 0039d316-1c4b-4281-b951-d872f2087c98
---
 .../policy/mock_device_management_backend.h        | 49 ++++++++++++++++++----
 1 file changed, 40 insertions(+), 9 deletions(-)

(limited to '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 290e615..e48e214 100644
--- a/chrome/browser/policy/mock_device_management_backend.h
+++ b/chrome/browser/policy/mock_device_management_backend.h
@@ -6,9 +6,13 @@
 #define CHROME_BROWSER_POLICY_MOCK_DEVICE_MANAGEMENT_BACKEND_H_
 #pragma once
 
+#include <map>
 #include <string>
 
+#include "base/values.h"
 #include "chrome/browser/policy/device_management_backend.h"
+#include "testing/gtest/include/gtest/gtest.h"
+#include "testing/gmock/include/gmock/gmock.h"
 
 namespace policy {
 
@@ -21,29 +25,56 @@ class MockDeviceManagementBackend
     : public DeviceManagementBackend {
  public:
   MockDeviceManagementBackend();
-  virtual ~MockDeviceManagementBackend() {}
-
-  void SetFailure(bool failure) { failure_ = failure; }
+  virtual ~MockDeviceManagementBackend();
 
   // DeviceManagementBackend method overrides:
-  virtual void ProcessRegisterRequest(
+  MOCK_METHOD4(ProcessRegisterRequest, void(
       const std::string& auth_token,
       const std::string& device_id,
       const em::DeviceRegisterRequest& request,
-      DeviceRegisterResponseDelegate* delegate);
+      DeviceRegisterResponseDelegate* delegate));
 
-  virtual void ProcessUnregisterRequest(
+  MOCK_METHOD3(ProcessUnregisterRequest, void(
       const std::string& device_management_token,
       const em::DeviceUnregisterRequest& request,
-      DeviceUnregisterResponseDelegate* delegate);
+      DeviceUnregisterResponseDelegate* delegate));
+
+  MOCK_METHOD3(ProcessPolicyRequest, void(
+      const std::string& device_management_token,
+      const em::DevicePolicyRequest& request,
+      DevicePolicyResponseDelegate* delegate));
+
+  void AllShouldSucceed();
+  void AllShouldFail();
+
+  void SimulateSuccessfulRegisterRequest(
+      const std::string& auth_token,
+      const std::string& device_id,
+      const em::DeviceRegisterRequest& request,
+      DeviceRegisterResponseDelegate* delegate);
 
-  virtual void ProcessPolicyRequest(
+  void SimulateSuccessfulPolicyRequest(
       const std::string& device_management_token,
       const em::DevicePolicyRequest& request,
       DevicePolicyResponseDelegate* delegate);
 
+  void SimulateFailedRegisterRequest(
+      const std::string& auth_token,
+      const std::string& device_id,
+      const em::DeviceRegisterRequest& request,
+      DeviceRegisterResponseDelegate* delegate);
+
+  void SimulateFailedPolicyRequest(
+      const std::string& device_management_token,
+      const em::DevicePolicyRequest& request,
+      DevicePolicyResponseDelegate* delegate);
+
+  void AddBooleanPolicy(const char* policy_name, bool value);
+
  private:
-  bool failure_;
+  em::DevicePolicyResponse policy_response_;
+  em::DevicePolicySetting* policy_setting_;
+
   DISALLOW_COPY_AND_ASSIGN(MockDeviceManagementBackend);
 };
 
-- 
cgit v1.1