Consolidate TestURLFetcherFactory::SetFakeResponse

chrome/browser/chromeos/policy/cloud_external_data_manager_base_unittest.cc

 // Copyright 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/chromeos/policy/cloud_external_data_manager_base.h"
 
 #include <map>
 #include <string>
 
 #include "base/bind.h"
@@ skipping 84 matching lines @@
     net::URLFetcher::RequestType request_type,
     net::URLFetcherDelegate* delegate) {
   net::URLFetcher* fetcher = net::FakeURLFetcherFactory::CreateURLFetcher(
       id, url, request_type, delegate);
   EXPECT_TRUE(fetcher);
   return fetcher;
 }
 
 }  // namespace
 
-class CouldExternalDataManagerBaseTest : public testing::Test {
+class CloudExternalDataManagerBaseTest : public testing::Test {
  protected:
-  CouldExternalDataManagerBaseTest();
+  CloudExternalDataManagerBaseTest();
 
   virtual void SetUp() OVERRIDE;
   virtual void TearDown() OVERRIDE;
 
   void SetUpExternalDataManager();
 
   scoped_ptr<base::DictionaryValue> ConstructMetadata(const std::string& url,
                                                       const std::string& hash);
   void SetExternalDataReference(const std::string& policy,
                                 scoped_ptr<base::DictionaryValue> metadata);
 
   ExternalDataFetcher::FetchCallback ConstructFetchCallback(int id);
   void ResetCallbackData();
 
   void OnFetchDone(int id, scoped_ptr<std::string> data);
 
   void FetchAll();
 
+  void SetFakeResponse(const std::string& url,
+                       const std::string& repsonse_data,
+                       bool success);
+
   base::MessageLoop message_loop_;
   base::ScopedTempDir temp_dir_;
   scoped_ptr<ResourceCache> resource_cache_;
   MockCloudPolicyStore cloud_policy_store_;
   scoped_refptr<net::TestURLRequestContextGetter> request_content_getter_;
   FakeURLFetcherFactory fetcher_factory_;
 
   scoped_ptr<CloudExternalDataManagerBase> external_data_manager_;
 
   std::map<int, std::string*> callback_data_;
 
-  DISALLOW_COPY_AND_ASSIGN(CouldExternalDataManagerBaseTest);
+  DISALLOW_COPY_AND_ASSIGN(CloudExternalDataManagerBaseTest);
 };
 
-CouldExternalDataManagerBaseTest::CouldExternalDataManagerBaseTest() {
+CloudExternalDataManagerBaseTest::CloudExternalDataManagerBaseTest() {
 }
 
-void CouldExternalDataManagerBaseTest::SetUp() {
+void CloudExternalDataManagerBaseTest::SetUp() {
   ASSERT_TRUE(temp_dir_.CreateUniqueTempDir());
   resource_cache_.reset(new ResourceCache(temp_dir_.path(),
                                           message_loop_.message_loop_proxy()));
   SetUpExternalDataManager();
 
   // Set |kStringPolicy| to a string value.
   cloud_policy_store_.policy_map_.Set(kStringPolicy,
                                       POLICY_LEVEL_MANDATORY,
                                       POLICY_SCOPE_USER,
                                       new base::StringValue(std::string()),
                                       NULL);
   // Make |k10BytePolicy| reference 10 bytes of external data.
   SetExternalDataReference(
       k10BytePolicy,
       ConstructMetadata(k10BytePolicyURL, base::SHA1HashString(k10ByteData)));
   // Make |k20BytePolicy| reference 20 bytes of external data.
   SetExternalDataReference(
       k20BytePolicy,
       ConstructMetadata(k20BytePolicyURL, base::SHA1HashString(k20ByteData)));
   cloud_policy_store_.NotifyStoreLoaded();
 
   request_content_getter_ = new net::TestURLRequestContextGetter(
       base::MessageLoopProxy::current());
 }
 
-void CouldExternalDataManagerBaseTest::TearDown() {
+void CloudExternalDataManagerBaseTest::TearDown() {
   external_data_manager_.reset();
   base::RunLoop().RunUntilIdle();
   ResetCallbackData();
 }
 
-void CouldExternalDataManagerBaseTest::SetUpExternalDataManager() {
+void CloudExternalDataManagerBaseTest::SetUpExternalDataManager() {
   external_data_manager_.reset(new CloudExternalDataManagerBase(
       &kPolicyDefinitionList,
       message_loop_.message_loop_proxy(),
       message_loop_.message_loop_proxy()));
   external_data_manager_->SetExternalDataStore(make_scoped_ptr(
       new CloudExternalDataStore(kCacheKey,
                                  message_loop_.message_loop_proxy(),
                                  resource_cache_.get())));
   external_data_manager_->SetPolicyStore(&cloud_policy_store_);
 }
 
 scoped_ptr<base::DictionaryValue>
-    CouldExternalDataManagerBaseTest::ConstructMetadata(
+    CloudExternalDataManagerBaseTest::ConstructMetadata(
         const std::string& url,
         const std::string& hash) {
   scoped_ptr<base::DictionaryValue> metadata(new base::DictionaryValue);
   metadata->SetStringWithoutPathExpansion("url", url);
   metadata->SetStringWithoutPathExpansion("hash", base::HexEncode(hash.c_str(),
                                                                   hash.size()));
   return metadata.Pass();
 }
 
-void CouldExternalDataManagerBaseTest::SetExternalDataReference(
+void CloudExternalDataManagerBaseTest::SetExternalDataReference(
     const std::string& policy,
     scoped_ptr<base::DictionaryValue> metadata) {
   cloud_policy_store_.policy_map_.Set(
       policy,
       POLICY_LEVEL_MANDATORY,
       POLICY_SCOPE_USER,
       metadata.release(),
       new ExternalDataFetcher(
           external_data_manager_->weak_factory_.GetWeakPtr(), policy));
 }
 
 ExternalDataFetcher::FetchCallback
-CouldExternalDataManagerBaseTest::ConstructFetchCallback(int id) {
-  return base::Bind(&CouldExternalDataManagerBaseTest::OnFetchDone,
+CloudExternalDataManagerBaseTest::ConstructFetchCallback(int id) {
+  return base::Bind(&CloudExternalDataManagerBaseTest::OnFetchDone,
                     base::Unretained(this),
                     id);
 }
 
-void CouldExternalDataManagerBaseTest::ResetCallbackData() {
+void CloudExternalDataManagerBaseTest::ResetCallbackData() {
   STLDeleteValues(&callback_data_);
 }
 
-void CouldExternalDataManagerBaseTest::OnFetchDone(
+void CloudExternalDataManagerBaseTest::OnFetchDone(
     int id,
     scoped_ptr<std::string> data) {
   delete callback_data_[id];
   callback_data_[id] = data.release();
 }
 
-void CouldExternalDataManagerBaseTest::FetchAll() {
+void CloudExternalDataManagerBaseTest::FetchAll() {
   external_data_manager_->FetchAll();
 }
 
+void CloudExternalDataManagerBaseTest::SetFakeResponse(
+    const std::string& url,
+    const std::string& response_data,
+    bool success) {
+  fetcher_factory_.SetFakeResponse(GURL(url), response_data, success);
+}
+
 // Verifies that when no valid external data reference has been set for a
 // policy, the attempt to retrieve the external data fails immediately.
-TEST_F(CouldExternalDataManagerBaseTest, FailToFetchInvalid) {
+TEST_F(CloudExternalDataManagerBaseTest, FailToFetchInvalid) {
   external_data_manager_->Connect(request_content_getter_);
 
   // Attempt to retrieve external data for |kStringPolicy|, which is a string
   // policy that does not reference any external data.
   external_data_manager_->Fetch(kStringPolicy, ConstructFetchCallback(0));
   base::RunLoop().RunUntilIdle();
   EXPECT_EQ(1u, callback_data_.size());
   EXPECT_TRUE(callback_data_.find(0) != callback_data_.end());
   EXPECT_FALSE(callback_data_[0]);
   ResetCallbackData();
@@ skipping 18 matching lines @@
   base::RunLoop().RunUntilIdle();
   EXPECT_EQ(1u, callback_data_.size());
   EXPECT_TRUE(callback_data_.find(2) != callback_data_.end());
   EXPECT_FALSE(callback_data_[2]);
   ResetCallbackData();
 }
 
 // Verifies that external data referenced by a policy is downloaded and cached
 // when first requested. Subsequent requests are served from the cache without
 // further download attempts.
-TEST_F(CouldExternalDataManagerBaseTest, DownloadAndCache) {
+TEST_F(CloudExternalDataManagerBaseTest, DownloadAndCache) {
   // Serve valid external data for |k10BytePolicy|.
-  fetcher_factory_.SetFakeResponse(k10BytePolicyURL, k10ByteData, true);
+  SetFakeResponse(k10BytePolicyURL, k10ByteData, true);
   external_data_manager_->Connect(request_content_getter_);
 
   // Retrieve external data for |k10BytePolicy|. Verify that a download happens
   // and the callback is invoked with the downloaded data.
   external_data_manager_->Fetch(k10BytePolicy, ConstructFetchCallback(0));
   base::RunLoop().RunUntilIdle();
   EXPECT_EQ(1u, callback_data_.size());
   ASSERT_TRUE(callback_data_[0]);
   EXPECT_EQ(k10ByteData, *callback_data_[0]);
   ResetCallbackData();
@@ skipping 32 matching lines @@
   EXPECT_TRUE(CloudExternalDataStore(kCacheKey,
                                      message_loop_.message_loop_proxy(),
                                      resource_cache_.get()).Load(
       k10BytePolicy, base::SHA1HashString(k10ByteData), 10, &data));
   EXPECT_EQ(k10ByteData, data);
 }
 
 // Verifies that a request to download and cache all external data referenced by
 // policies is carried out correctly. Subsequent requests for the data are
 // served from the cache without further download attempts.
-TEST_F(CouldExternalDataManagerBaseTest, DownloadAndCacheAll) {
+TEST_F(CloudExternalDataManagerBaseTest, DownloadAndCacheAll) {
   // Serve valid external data for |k10BytePolicy| and |k20BytePolicy|.
-  fetcher_factory_.SetFakeResponse(k10BytePolicyURL, k10ByteData, true);
-  fetcher_factory_.SetFakeResponse(k20BytePolicyURL, k20ByteData, true);
+  SetFakeResponse(k10BytePolicyURL, k10ByteData, true);
+  SetFakeResponse(k20BytePolicyURL, k20ByteData, true);
   external_data_manager_->Connect(request_content_getter_);
 
   // Request that external data referenced by all policies be downloaded.
   FetchAll();
   base::RunLoop().RunUntilIdle();
 
   // Stop serving external data for |k10BytePolicy| and |k20BytePolicy|.
   fetcher_factory_.ClearFakeResponses();
 
   // Retrieve external data for |k10BytePolicy| and |k20BytePolicy|. Verify that
@@ skipping 37 matching lines @@
                          &data));
   EXPECT_EQ(k10ByteData, data);
   EXPECT_TRUE(cache.Load(k20BytePolicy, base::SHA1HashString(k20ByteData), 20,
                          &data));
   EXPECT_EQ(k20ByteData, data);
 }
 
 // Verifies that when the external data referenced by a policy is not present in
 // the cache and downloads are not allowed, a request to retrieve the data is
 // enqueued and carried out when downloads become possible.
-TEST_F(CouldExternalDataManagerBaseTest, DownloadAfterConnect) {
+TEST_F(CloudExternalDataManagerBaseTest, DownloadAfterConnect) {
   // Attempt to retrieve external data for |k10BytePolicy|. Verify that the
   // callback is not invoked as the request remains pending.
   external_data_manager_->Fetch(k10BytePolicy, ConstructFetchCallback(0));
   base::RunLoop().RunUntilIdle();
   EXPECT_TRUE(callback_data_.empty());
   ResetCallbackData();
 
   // Serve valid external data for |k10BytePolicy| and allow the
   // external_data_manager_ to perform downloads.
-  fetcher_factory_.SetFakeResponse(k10BytePolicyURL, k10ByteData, true);
+  SetFakeResponse(k10BytePolicyURL, k10ByteData, true);
   external_data_manager_->Connect(request_content_getter_);
 
   // Verify that a download happens and the callback is invoked with the
   // downloaded data.
   base::RunLoop().RunUntilIdle();
   EXPECT_EQ(1u, callback_data_.size());
   ASSERT_TRUE(callback_data_[0]);
   EXPECT_EQ(k10ByteData, *callback_data_[0]);
   ResetCallbackData();
 }
 
 // Verifies that when the external data referenced by a policy is not present in
 // the cache and cannot be downloaded at this time, a request to retrieve the
 // data is enqueued to be retried later.
-TEST_F(CouldExternalDataManagerBaseTest, DownloadError) {
+TEST_F(CloudExternalDataManagerBaseTest, DownloadError) {
   // Make attempts to download the external data for |k20BytePolicy| fail with
   // an error.
-  fetcher_factory_.SetFakeResponse(k20BytePolicyURL, std::string(), false);
+  SetFakeResponse(k20BytePolicyURL, std::string(), false);
   external_data_manager_->Connect(request_content_getter_);
 
   // Attempt to retrieve external data for |k20BytePolicy|. Verify that the
   // callback is not invoked as the download attempt fails and the request
   // remains pending.
   external_data_manager_->Fetch(k20BytePolicy, ConstructFetchCallback(0));
   base::RunLoop().RunUntilIdle();
   EXPECT_TRUE(callback_data_.empty());
   ResetCallbackData();
 
@@ skipping 14 matching lines @@
 
   // Modify the external data reference for |k20BytePolicy|, allowing the
   // download to be retried immediately.
   SetExternalDataReference(
       k20BytePolicy,
       ConstructMetadata(k20BytePolicyURL, base::SHA1HashString(k20ByteData)));
   cloud_policy_store_.NotifyStoreLoaded();
 
   // Serve external data for |k20BytePolicy| that does not match the hash
   // specified in its current external data reference.
-  fetcher_factory_.SetFakeResponse(k20BytePolicyURL, k10ByteData, true);
+  SetFakeResponse(k20BytePolicyURL, k10ByteData, true);
 
   // Attempt to retrieve external data for |k20BytePolicy| again. Verify that
   // no callback is invoked still as the downloaded succeeds but returns data
   // that does not match the external data reference.
   external_data_manager_->Fetch(k20BytePolicy, ConstructFetchCallback(2));
   base::RunLoop().RunUntilIdle();
   EXPECT_TRUE(callback_data_.empty());
   ResetCallbackData();
 
   // Modify the external data reference for |k20BytePolicy|, allowing the
@@ skipping 17 matching lines @@
   ASSERT_TRUE(callback_data_[2]);
   EXPECT_EQ(k10ByteData, *callback_data_[2]);
   ASSERT_TRUE(callback_data_[3]);
   EXPECT_EQ(k10ByteData, *callback_data_[3]);
   ResetCallbackData();
 }
 
 // Verifies that when the external data referenced by a policy is present in the
 // cache, a request to retrieve it is served from the cache without any download
 // attempts.
-TEST_F(CouldExternalDataManagerBaseTest, LoadFromCache) {
+TEST_F(CloudExternalDataManagerBaseTest, LoadFromCache) {
   // Store valid external data for |k10BytePolicy| in the cache.
   external_data_manager_.reset();
   base::RunLoop().RunUntilIdle();
   EXPECT_TRUE(CloudExternalDataStore(kCacheKey,
                                      message_loop_.message_loop_proxy(),
                                      resource_cache_.get()).Store(
       k10BytePolicy, base::SHA1HashString(k10ByteData), k10ByteData));
 
   // Instantiate an external_data_manager_ that uses the primed cache.
   SetUpExternalDataManager();
   external_data_manager_->Connect(request_content_getter_);
 
   // Retrieve external data for |k10BytePolicy|. Verify that no download is
   // attempted but the callback is still invoked with the expected data, served
   // from the cache.
   external_data_manager_->Fetch(k10BytePolicy, ConstructFetchCallback(0));
   base::RunLoop().RunUntilIdle();
   EXPECT_EQ(1u, callback_data_.size());
   ASSERT_TRUE(callback_data_[0]);
   EXPECT_EQ(k10ByteData, *callback_data_[0]);
   ResetCallbackData();
 }
 
 // Verifies that cache entries which do not correspond to the external data
 // referenced by any policy are pruned on startup.
-TEST_F(CouldExternalDataManagerBaseTest, PruneCacheOnStartup) {
+TEST_F(CloudExternalDataManagerBaseTest, PruneCacheOnStartup) {
   external_data_manager_.reset();
   base::RunLoop().RunUntilIdle();
   scoped_ptr<CloudExternalDataStore>
       cache(new CloudExternalDataStore(kCacheKey,
                                        message_loop_.message_loop_proxy(),
                                        resource_cache_.get()));
   // Store valid external data for |k10BytePolicy| in the cache.
   EXPECT_TRUE(cache->Store(k10BytePolicy,
                            base::SHA1HashString(k10ByteData),
                            k10ByteData));
@@ skipping 26 matching lines @@
   // Verify that the external data for |k20BytePolicy| and |kUnknownPolicy| has
   // been pruned from the cache.
   EXPECT_FALSE(cache->Load(k20BytePolicy, base::SHA1HashString(k10ByteData),
                            20, &data));
   EXPECT_FALSE(cache->Load(kUnknownPolicy, base::SHA1HashString(k10ByteData),
                            20, &data));
 }
 
 // Verifies that when the external data referenced by a policy is present in the
 // cache and the reference changes, the old data is pruned from the cache.
-TEST_F(CouldExternalDataManagerBaseTest, PruneCacheOnChange) {
+TEST_F(CloudExternalDataManagerBaseTest, PruneCacheOnChange) {
   // Store valid external data for |k20BytePolicy| in the cache.
   external_data_manager_.reset();
   base::RunLoop().RunUntilIdle();
   scoped_ptr<CloudExternalDataStore>
       cache(new CloudExternalDataStore(kCacheKey,
                                        message_loop_.message_loop_proxy(),
                                        resource_cache_.get()));
   EXPECT_TRUE(cache->Store(k20BytePolicy,
                            base::SHA1HashString(k20ByteData),
                            k20ByteData));
@@ skipping 15 matching lines @@
   base::RunLoop().RunUntilIdle();
   cache.reset(new CloudExternalDataStore(kCacheKey,
                                          message_loop_.message_loop_proxy(),
                                          resource_cache_.get()));
   std::string data;
   EXPECT_FALSE(cache->Load(k20BytePolicy, base::SHA1HashString(k20ByteData), 20,
                            &data));
 }
 
 // Verifies that corrupt cache entries are detected and deleted when accessed.
-TEST_F(CouldExternalDataManagerBaseTest, CacheCorruption) {
+TEST_F(CloudExternalDataManagerBaseTest, CacheCorruption) {
   external_data_manager_.reset();
   base::RunLoop().RunUntilIdle();
   scoped_ptr<CloudExternalDataStore>
       cache(new CloudExternalDataStore(kCacheKey,
                                        message_loop_.message_loop_proxy(),
                                        resource_cache_.get()));
   // Store external data for |k10BytePolicy| that exceeds the maximal external
   // data size allowed for that policy.
   EXPECT_TRUE(cache->Store(k10BytePolicy,
                            base::SHA1HashString(k20ByteData),
                            k20ByteData));
   // Store external data for |k20BytePolicy| that is corrupted and does not
   // match the expected hash.
   EXPECT_TRUE(cache->Store(k20BytePolicy,
                            base::SHA1HashString(k20ByteData),
                            k10ByteData));
   cache.reset();
 
   SetUpExternalDataManager();
   // Serve external data for |k10BytePolicy| that exceeds the maximal external
   // data size allowed for that policy.
-  fetcher_factory_.SetFakeResponse(k10BytePolicyURL, k20ByteData, true);
+  SetFakeResponse(k10BytePolicyURL, k20ByteData, true);
   external_data_manager_->Connect(request_content_getter_);
 
   // Modify the external data reference for |k10BytePolicy| to match the
   // external data being served.
   SetExternalDataReference(
       k10BytePolicy,
       ConstructMetadata(k10BytePolicyURL, base::SHA1HashString(k20ByteData)));
   cloud_policy_store_.NotifyStoreLoaded();
 
   // Retrieve external data for |k10BytePolicy|. Verify that the callback is
   // not invoked as the cached and downloaded external data exceed the maximal
   // size allowed for this policy and the request remains pending.
   external_data_manager_->Fetch(k10BytePolicy, ConstructFetchCallback(0));
   base::RunLoop().RunUntilIdle();
   EXPECT_TRUE(callback_data_.empty());
   ResetCallbackData();
 
   // Serve valid external data for |k20BytePolicy|.
-  fetcher_factory_.SetFakeResponse(k20BytePolicyURL, k20ByteData, true);
+  SetFakeResponse(k20BytePolicyURL, k20ByteData, true);
 
   // Retrieve external data for |k20BytePolicy|. Verify that the callback is
   // invoked with the valid downloaded data, not the invalid data in the cache.
   external_data_manager_->Fetch(k20BytePolicy, ConstructFetchCallback(1));
   base::RunLoop().RunUntilIdle();
   EXPECT_EQ(1u, callback_data_.size());
   ASSERT_TRUE(callback_data_[1]);
   EXPECT_EQ(k20ByteData, *callback_data_[1]);
   ResetCallbackData();
 
@@ skipping 17 matching lines @@
   // Verify that the invalid external data for |k20BytePolicy| has been replaced
   // with the downloaded valid data in the cache.
   EXPECT_TRUE(cache->Load(k20BytePolicy, base::SHA1HashString(k20ByteData), 20,
                           &data));
   EXPECT_EQ(k20ByteData, data);
 }
 
 // Verifies that when the external data reference for a policy changes while a
 // download of the external data for that policy is pending, the download is
 // immediately retried using the new reference.
-TEST_F(CouldExternalDataManagerBaseTest, PolicyChangeWhileDownloadPending) {
+TEST_F(CloudExternalDataManagerBaseTest, PolicyChangeWhileDownloadPending) {
   // Make attempts to download the external data for |k10BytePolicy| and
   // |k20BytePolicy| fail with an error.
-  fetcher_factory_.SetFakeResponse(k10BytePolicyURL, std::string(), false);
-  fetcher_factory_.SetFakeResponse(k20BytePolicyURL, std::string(), false);
+  SetFakeResponse(k10BytePolicyURL, std::string(), false);
+  SetFakeResponse(k20BytePolicyURL, std::string(), false);
   external_data_manager_->Connect(request_content_getter_);
 
   // Attempt to retrieve external data for |k10BytePolicy| and |k20BytePolicy|.
   // Verify that no callbacks are invoked as the download attempts fail and the
   // requests remain pending.
   external_data_manager_->Fetch(k10BytePolicy, ConstructFetchCallback(0));
   external_data_manager_->Fetch(k20BytePolicy, ConstructFetchCallback(1));
   base::RunLoop().RunUntilIdle();
   EXPECT_TRUE(callback_data_.empty());
   ResetCallbackData();
 
   // Modify the external data reference for |k10BytePolicy| to be invalid.
   // Verify that the callback is invoked as the policy no longer has a valid
   // external data reference.
   cloud_policy_store_.policy_map_.Erase(k10BytePolicy);
   cloud_policy_store_.NotifyStoreLoaded();
   base::RunLoop().RunUntilIdle();
   EXPECT_EQ(1u, callback_data_.size());
   EXPECT_TRUE(callback_data_.find(0) != callback_data_.end());
   EXPECT_FALSE(callback_data_[0]);
   ResetCallbackData();
 
   // Serve valid external data for |k20BytePolicy|.
   fetcher_factory_.ClearFakeResponses();
-  fetcher_factory_.SetFakeResponse(k20BytePolicyURL, k10ByteData, true);
+  SetFakeResponse(k20BytePolicyURL, k10ByteData, true);
 
   // Modify the external data reference for |k20BytePolicy| to match the
   // external data now being served. Verify that the callback is invoked with
   // the downloaded data.
   SetExternalDataReference(
       k20BytePolicy,
       ConstructMetadata(k20BytePolicyURL, base::SHA1HashString(k10ByteData)));
   cloud_policy_store_.NotifyStoreLoaded();
   base::RunLoop().RunUntilIdle();
   EXPECT_EQ(1u, callback_data_.size());
   ASSERT_TRUE(callback_data_[1]);
   EXPECT_EQ(k10ByteData, *callback_data_[1]);
   ResetCallbackData();
 }
 
 }  // namespace policy