Tests for cloud policy UMA metrics.

chrome/browser/policy/mock_device_management_service.cc

 // Copyright (c) 2011 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_service.h"
 
 namespace policy {
 
 ProxyDeviceManagementBackend::ProxyDeviceManagementBackend(
     DeviceManagementBackend* backend)
@@ skipping 21 matching lines @@
     const std::string& device_management_token,
     const std::string& device_id,
     const em::DevicePolicyRequest& request,
     DevicePolicyResponseDelegate* delegate) {
   backend_->ProcessPolicyRequest(device_management_token, device_id,
                                  request, delegate);
 }
 
 MockDeviceManagementService::MockDeviceManagementService()
     : DeviceManagementService(""),
-      backend_(NULL) {}
+      backend_(NULL),
+      response_code_(-1) {}
 
 MockDeviceManagementService::~MockDeviceManagementService() {}
 
 DeviceManagementBackend* MockDeviceManagementService::CreateBackend() {
-  return new ProxyDeviceManagementBackend(backend_);
+  if (backend_ == NULL)
+    return DeviceManagementService::CreateBackend();
+  else
+    return new ProxyDeviceManagementBackend(backend_);
+}
+
+void MockDeviceManagementService::StartJob(DeviceManagementJob* job) {
+  if (backend_ == NULL)
+    job->HandleResponse(status_, response_code_, cookies_, data_);
+  else
+    DeviceManagementService::StartJob(job);
 }
 
 }  // namespace policy