summaryrefslogtreecommitdiffstats
path: root/chrome/browser/policy/mock_device_management_service.h
diff options
context:
space:
mode:
Diffstat (limited to 'chrome/browser/policy/mock_device_management_service.h')
-rw-r--r--chrome/browser/policy/mock_device_management_service.h15
1 files changed, 15 insertions, 0 deletions
diff --git a/chrome/browser/policy/mock_device_management_service.h b/chrome/browser/policy/mock_device_management_service.h
index 51593bf..c8c5828 100644
--- a/chrome/browser/policy/mock_device_management_service.h
+++ b/chrome/browser/policy/mock_device_management_service.h
@@ -10,10 +10,19 @@
#include "base/basictypes.h"
#include "chrome/browser/policy/device_management_service.h"
+#include "chrome/browser/policy/proto/device_management_backend.pb.h"
#include "testing/gmock/include/gmock/gmock.h"
namespace policy {
+class MockDeviceManagementJob {
+ public:
+ virtual ~MockDeviceManagementJob();
+ virtual void SendResponse(
+ DeviceManagementStatus status,
+ const enterprise_management::DeviceManagementResponse& response) = 0;
+};
+
class MockDeviceManagementService : public DeviceManagementService {
public:
MockDeviceManagementService();
@@ -36,9 +45,15 @@ class MockDeviceManagementService : public DeviceManagementService {
// Creates a gmock action that will make the job succeed.
testing::Action<CreateJobFunction> SucceedJob(
const enterprise_management::DeviceManagementResponse& response);
+
// Creates a gmock action which will fail the job with the given error.
testing::Action<CreateJobFunction> FailJob(DeviceManagementStatus status);
+ // Creates a gmock action which will capture the job so the test code can
+ // delay job completion.
+ testing::Action<CreateJobFunction> CreateAsyncJob(
+ MockDeviceManagementJob** job);
+
private:
DISALLOW_COPY_AND_ASSIGN(MockDeviceManagementService);
};