Implement exponential back-off mechanism.

chrome/service/cloud_print/cloud_print_url_fetcher_unittest.cc

 // Copyright (c) 2010 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 "base/command_line.h"
 #include "base/message_loop_proxy.h"
 #include "base/ref_counted.h"
 #include "base/thread.h"
 #include "base/waitable_event.h"
-#include "chrome/common/net/url_fetcher_protect.h"
 #include "chrome/common/net/url_request_context_getter.h"
 #include "chrome/service/service_process.h"
 #include "chrome/service/cloud_print/cloud_print_url_fetcher.h"
 #include "googleurl/src/gurl.h"
 #include "net/test/test_server.h"
+#include "net/url_request/url_request_status.h"
+#include "net/url_request/url_request_throttler_manager.h"
 #include "net/url_request/url_request_unittest.h"
-#include "net/url_request/url_request_status.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 using base::Time;
 using base::TimeDelta;
 
 namespace {
 
 const FilePath::CharType kDocRoot[] = FILE_PATH_LITERAL("chrome/test/data");
 
 int g_request_context_getter_instances = 0;
@@ skipping 34 matching lines @@
   virtual URLRequestContextGetter* GetRequestContextGetter() {
     return new TestURLRequestContextGetter(io_message_loop_proxy_.get());
   }
  private:
   scoped_refptr<base::MessageLoopProxy> io_message_loop_proxy_;
 };
 
 class CloudPrintURLFetcherTest : public testing::Test,
                                  public CloudPrintURLFetcher::Delegate {
  public:
-  CloudPrintURLFetcherTest() : fetcher_(NULL) { }
+  CloudPrintURLFetcherTest() : max_retries_(0), fetcher_(NULL) { }
 
   // Creates a URLFetcher, using the program's main thread to do IO.
-  virtual void CreateFetcher(const GURL& url, const std::string& retry_policy);
+  virtual void CreateFetcher(const GURL& url, int max_retries);
 
   // CloudPrintURLFetcher::Delegate
   virtual CloudPrintURLFetcher::ResponseAction HandleRawResponse(
       const URLFetcher* source,
       const GURL& url,
       const URLRequestStatus& status,
       int response_code,
       const ResponseCookies& cookies,
       const std::string& data);
 
@@ skipping 20 matching lines @@
     MessageLoop::current()->RunAllPending();
     EXPECT_EQ(0, g_request_context_getter_instances);
   }
 
   // URLFetcher is designed to run on the main UI thread, but in our tests
   // we assume that the current thread is the IO thread where the URLFetcher
   // dispatches its requests to.  When we wish to simulate being used from
   // a UI thread, we dispatch a worker thread to do so.
   MessageLoopForIO io_loop_;
   scoped_refptr<base::MessageLoopProxy> io_message_loop_proxy_;
-  std::string retry_policy_;
+  int max_retries_;
   Time start_time_;
   scoped_refptr<CloudPrintURLFetcher> fetcher_;
 };
 
 class CloudPrintURLFetcherBasicTest : public CloudPrintURLFetcherTest {
  public:
   CloudPrintURLFetcherBasicTest()
       : handle_raw_response_(false), handle_raw_data_(false) { }
   // CloudPrintURLFetcher::Delegate
   virtual CloudPrintURLFetcher::ResponseAction HandleRawResponse(
@@ skipping 54 matching lines @@
       const GURL& url,
       const std::string& data);
 
   virtual void OnRequestGiveUp();
 
  private:
   int response_count_;
 };
 
 
-void CloudPrintURLFetcherTest::CreateFetcher(const GURL& url,
-                                             const std::string& retry_policy) {
+void CloudPrintURLFetcherTest::CreateFetcher(const GURL& url, int max_retries) {
   fetcher_ = new TestCloudPrintURLFetcher(io_message_loop_proxy());
-  retry_policy_ = retry_policy;
+  max_retries_ = max_retries;
   start_time_ = Time::Now();
-  fetcher_->StartGetRequest(url, this, "", retry_policy_);
+  fetcher_->StartGetRequest(url, this, "", max_retries_);
 }
 
 CloudPrintURLFetcher::ResponseAction
 CloudPrintURLFetcherTest::HandleRawResponse(
     const URLFetcher* source,
     const GURL& url,
     const URLRequestStatus& status,
     int response_code,
     const ResponseCookies& cookies,
     const std::string& data) {
@@ skipping 51 matching lines @@
   return CloudPrintURLFetcher::STOP_PROCESSING;
 }
 
 CloudPrintURLFetcher::ResponseAction
 CloudPrintURLFetcherOverloadTest::HandleRawData(const URLFetcher* source,
                                                 const GURL& url,
                                                 const std::string& data) {
   const TimeDelta one_second = TimeDelta::FromMilliseconds(1000);
   response_count_++;
   if (response_count_ < 20) {
-    fetcher_->StartGetRequest(url, this, "", retry_policy_);
+    fetcher_->StartGetRequest(url, this, "", max_retries_);
   } else {
     // We have already sent 20 requests continuously. And we expect that
-    // it takes more than 1 second due to the overload pretection settings.
+    // it takes more than 1 second due to the overload protection settings.
     EXPECT_TRUE(Time::Now() - start_time_ >= one_second);
     io_message_loop_proxy()->PostTask(FROM_HERE, new MessageLoop::QuitTask());
   }
   return CloudPrintURLFetcher::STOP_PROCESSING;
 }
 
 CloudPrintURLFetcher::ResponseAction
 CloudPrintURLFetcherRetryBackoffTest::HandleRawData(const URLFetcher* source,
                                                     const GURL& url,
                                                     const std::string& data) {
   response_count_++;
   // First attempt + 11 retries = 12 total responses.
   EXPECT_LE(response_count_, 12);
   return CloudPrintURLFetcher::RETRY_REQUEST;
 }
 
 void CloudPrintURLFetcherRetryBackoffTest::OnRequestGiveUp() {
   const TimeDelta one_second = TimeDelta::FromMilliseconds(1000);
   // It takes more than 1 second to finish all 11 requests.
   EXPECT_TRUE(Time::Now() - start_time_ >= one_second);
   io_message_loop_proxy()->PostTask(FROM_HERE, new MessageLoop::QuitTask());
 }
 
 TEST_F(CloudPrintURLFetcherBasicTest, HandleRawResponse) {
   net::TestServer test_server(net::TestServer::TYPE_HTTP, FilePath(kDocRoot));
   ASSERT_TRUE(test_server.Start());
   SetHandleRawResponse(true);
 
-  CreateFetcher(test_server.GetURL("echo"), "DummyRetryPolicy");
+  CreateFetcher(test_server.GetURL("echo"), 0);
   MessageLoop::current()->Run();
 }
 
 // http://code.google.com/p/chromium/issues/detail?id=62758
 TEST_F(CloudPrintURLFetcherBasicTest, FLAKY_HandleRawData) {
   net::TestServer test_server(net::TestServer::TYPE_HTTP, FilePath(kDocRoot));
   ASSERT_TRUE(test_server.Start());
 
   SetHandleRawData(true);
-  CreateFetcher(test_server.GetURL("echo"), "DummyRetryPolicy");
+  CreateFetcher(test_server.GetURL("echo"), 0);
   MessageLoop::current()->Run();
 }
 
 TEST_F(CloudPrintURLFetcherOverloadTest, Protect) {
   net::TestServer test_server(net::TestServer::TYPE_HTTP, FilePath(kDocRoot));
   ASSERT_TRUE(test_server.Start());
 
   GURL url(test_server.GetURL("defaultresponse"));
 
   // Registers an entry for test url. It only allows 3 requests to be sent
   // in 200 milliseconds.
-  std::string retry_policy = "OverloadTestPolicy";
-  URLFetcherProtectManager* manager = URLFetcherProtectManager::GetInstance();
-  URLFetcherProtectEntry* entry =
-      new URLFetcherProtectEntry(200, 3, 11, 1, 2.0, 0, 256);
-  manager->Register(retry_policy, entry);
+  scoped_refptr<net::URLRequestThrottlerEntry> entry(
+      new net::URLRequestThrottlerEntry(200, 3, 1, 0, 2.0, 0.0, 256));
+  net::URLRequestThrottlerManager::GetInstance()->OverrideEntryForTests(
+      url, entry);
 
-  CreateFetcher(url, retry_policy);
+  CreateFetcher(url, 11);
 
   MessageLoop::current()->Run();
+
+  net::URLRequestThrottlerManager::GetInstance()->EraseEntryForTests(url);
 }
 
 // http://code.google.com/p/chromium/issues/detail?id=62758
 TEST_F(CloudPrintURLFetcherRetryBackoffTest, FLAKY_GiveUp) {
   net::TestServer test_server(net::TestServer::TYPE_HTTP, FilePath(kDocRoot));
   ASSERT_TRUE(test_server.Start());
 
   GURL url(test_server.GetURL("defaultresponse"));
 
   // Registers an entry for test url. The backoff time is calculated by:
   //     new_backoff = 2.0 * old_backoff + 0
   // and maximum backoff time is 256 milliseconds.
   // Maximum retries allowed is set to 11.
-  std::string retry_policy = "BackoffTestPolicy";
-  URLFetcherProtectManager* manager = URLFetcherProtectManager::GetInstance();
-  URLFetcherProtectEntry* entry =
-      new URLFetcherProtectEntry(200, 3, 11, 1, 2.0, 0, 256);
-  manager->Register(retry_policy, entry);
+  scoped_refptr<net::URLRequestThrottlerEntry> entry(
+      new net::URLRequestThrottlerEntry(200, 3, 1, 0, 2.0, 0.0, 256));
+  net::URLRequestThrottlerManager::GetInstance()->OverrideEntryForTests(
+      url, entry);
 
-  CreateFetcher(url, retry_policy);
+  CreateFetcher(url, 11);
 
   MessageLoop::current()->Run();
+
+  net::URLRequestThrottlerManager::GetInstance()->EraseEntryForTests(url);
 }
 
 }  // namespace.