Refactored the ComponentCloudPolicyService.

chrome/browser/policy/cloud/component_cloud_policy_updater_unittest.cc

 // Copyright (c) 2013 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/cloud/component_cloud_policy_updater.h"
 
 #include <string>

[bartfab (slow), 2013/11/14 14:13:12]

Nit: Already included by header now.

[Joao da Silva, 2013/11/14 14:56:41]

Done.

 
 #include "base/callback.h"
 #include "base/compiler_specific.h"
 #include "base/files/scoped_temp_dir.h"
 #include "base/sequenced_task_runner.h"
 #include "base/sha1.h"
 #include "base/test/test_simple_task_runner.h"
 #include "base/values.h"
 #include "chrome/browser/policy/cloud/cloud_policy_constants.h"
 #include "chrome/browser/policy/cloud/component_cloud_policy_store.h"
@@ skipping 307 matching lines @@
   // Update the policy fetch response before the download has finished. The new
   // policy fetch response has no download URL.
   builder_.payload().Clear();
   updater_->UpdateExternalPolicy(CreateResponse());
   task_runner_->RunUntilIdle();
 
   // Verify that the download is no longer running.
   EXPECT_FALSE(fetcher_factory_.GetFetcherByID(0));
 }
 
+TEST_F(ComponentCloudPolicyUpdaterTest, CancelUpdate) {
+  // Submit a policy fetch response with a valid download URL.
+  updater_->UpdateExternalPolicy(CreateResponse());
+  task_runner_->RunUntilIdle();
+
+  // Verify that the download has been started.
+  EXPECT_TRUE(fetcher_factory_.GetFetcherByID(0));
+
+  // Now cancel that update before the download completes.
+  EXPECT_CALL(store_delegate_, OnComponentCloudPolicyStoreUpdated()).Times(0);
+  updater_->CancelUpdate(
+      PolicyNamespace(POLICY_DOMAIN_EXTENSIONS, kTestExtension));
+  task_runner_->RunUntilIdle();
+  Mock::VerifyAndClearExpectations(&store_delegate_);
+  EXPECT_FALSE(fetcher_factory_.GetFetcherByID(0));
+}
+
 }  // namespace policy