1// Copyright (c) 2012 The Chromium Authors. All rights reserved.
2// Use of this source code is governed by a BSD-style license that can be
3// found in the LICENSE file.
4
5#ifndef CHROME_BROWSER_POLICY_CLOUD_MOCK_DEVICE_MANAGEMENT_SERVICE_H_
6#define CHROME_BROWSER_POLICY_CLOUD_MOCK_DEVICE_MANAGEMENT_SERVICE_H_
7
8#include <string>
9
10#include "base/basictypes.h"
11#include "chrome/browser/policy/cloud/device_management_service.h"
12#include "chrome/browser/policy/proto/cloud/device_management_backend.pb.h"
13#include "testing/gmock/include/gmock/gmock.h"
14
15namespace policy {
16
17class MockDeviceManagementJob {
18 public:
19  virtual ~MockDeviceManagementJob();
20  virtual void RetryJob() = 0;
21  virtual void SendResponse(
22      DeviceManagementStatus status,
23      const enterprise_management::DeviceManagementResponse& response) = 0;
24};
25
26class MockDeviceManagementService : public DeviceManagementService {
27 public:
28  MockDeviceManagementService();
29  virtual ~MockDeviceManagementService();
30
31  typedef DeviceManagementRequestJob* CreateJobFunction(
32      DeviceManagementRequestJob::JobType);
33
34  MOCK_METHOD1(CreateJob, CreateJobFunction);
35  MOCK_METHOD7(
36      StartJob,
37      void(const std::string& request_type,
38           const std::string& gaia_token,
39           const std::string& oauth_token,
40           const std::string& dm_token,
41           const std::string& user_affiliation,
42           const std::string& client_id,
43           const enterprise_management::DeviceManagementRequest& request));
44
45  // Creates a gmock action that will make the job succeed.
46  testing::Action<CreateJobFunction> SucceedJob(
47      const enterprise_management::DeviceManagementResponse& response);
48
49  // Creates a gmock action which will fail the job with the given error.
50  testing::Action<CreateJobFunction> FailJob(DeviceManagementStatus status);
51
52  // Creates a gmock action which will capture the job so the test code can
53  // delay job completion.
54  testing::Action<CreateJobFunction> CreateAsyncJob(
55      MockDeviceManagementJob** job);
56
57 private:
58  DISALLOW_COPY_AND_ASSIGN(MockDeviceManagementService);
59};
60
61}  // namespace policy
62
63#endif  // CHROME_BROWSER_POLICY_CLOUD_MOCK_DEVICE_MANAGEMENT_SERVICE_H_
64