Add UMA stats to track whitelist download latency.

components/autofill/browser/autocheckout/whitelist_manager_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 "base/command_line.h"
 #include "base/memory/scoped_ptr.h"
 #include "base/message_loop.h"
 #include "chrome/test/base/testing_profile.h"
 #include "components/autofill/browser/autocheckout/whitelist_manager.h"
+#include "components/autofill/browser/autofill_metrics.h"
 #include "components/autofill/common/autofill_switches.h"
 #include "content/public/test/test_browser_thread.h"
 #include "googleurl/src/gurl.h"
 #include "net/base/net_errors.h"
 #include "net/http/http_status_code.h"
 #include "net/url_request/test_url_fetcher_factory.h"
 #include "net/url_request/url_fetcher_delegate.h"
 #include "net/url_request/url_request_status.h"
+#include "testing/gmock/include/gmock/gmock.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 namespace {
 
 const size_t kTestDownloadInterval = 3;  // 3 seconds
 
 const char kDownloadWhitelistResponse[] =
     "https://www.merchant1.com/checkout/\n"
     "https://cart.merchant2.com/";
 
 }  // namespace
 
 namespace autofill {
 namespace autocheckout {
 
 class WhitelistManagerTest;
 
+class MockAutofillMetrics : public autofill::AutofillMetrics {
+ public:
+  MockAutofillMetrics() {}
+  MOCK_CONST_METHOD2(LogAutocheckoutWhitelistDownloadDuration,
+      void(const base::TimeDelta& duration,
+           autofill::AutofillMetrics::WhitelistDownloadStatus));
+ private:
+  DISALLOW_COPY_AND_ASSIGN(MockAutofillMetrics);
+};
+
 class TestWhitelistManager : public WhitelistManager {
  public:
   TestWhitelistManager()
       : WhitelistManager(),
         did_start_download_timer_(false) {}
 
   virtual void ScheduleDownload(size_t interval_seconds) OVERRIDE {
     did_start_download_timer_ = false;
     return WhitelistManager::ScheduleDownload(interval_seconds);
   }
@@ skipping 12 matching lines @@
   }
 
   void StopDownloadTimer() {
     WhitelistManager::StopDownloadTimer();
   }
 
   const std::vector<std::string>& url_prefixes() const {
     return WhitelistManager::url_prefixes();
   }
 
+  virtual const AutofillMetrics& GetMetricLogger() const OVERRIDE {
+      return mock_metrics_logger_;
+  }
+
  private:
   bool did_start_download_timer_;
 
+  MockAutofillMetrics mock_metrics_logger_;
+
   DISALLOW_COPY_AND_ASSIGN(TestWhitelistManager);
 };
 
 class WhitelistManagerTest : public testing::Test {
  public:
   WhitelistManagerTest() : io_thread_(content::BrowserThread::IO) {}
 
   virtual void SetUp() {
     io_thread_.StartIOThread();
     profile_.CreateRequestContext();
@@ skipping 10 matching lines @@
       whitelist_manager_.reset(new TestWhitelistManager());
     }
   }
 
   void DownloadWhitelist(int response_code, const std::string& response) {
     // Create and register factory.
     net::TestURLFetcherFactory factory;
 
     CreateWhitelistManager();
 
+    EXPECT_CALL(
+        static_cast<const MockAutofillMetrics&>(
+            whitelist_manager_->GetMetricLogger()),
+        LogAutocheckoutWhitelistDownloadDuration(
+            testing::_,
+            response_code == net::HTTP_OK ?
+                autofill::AutofillMetrics::WHITELIST_DOWNLOAD_SUCCEEDED :
+                autofill::AutofillMetrics::WHITELIST_DOWNLOAD_FAILED))
+        .Times(1);

[Ilya Sherman, 2013/04/17 00:18:01]

nit: This is pretty hard to read.  Could you break this into multiple statements
by introducing some intermediate variables?

[benquan, 2013/04/17 01:51:30]

Done.

+
     whitelist_manager_->TriggerDownload();
     net::TestURLFetcher* fetcher = factory.GetFetcherByID(0);
     ASSERT_TRUE(fetcher);
     fetcher->set_response_code(response_code);
     fetcher->SetResponseString(response);
     fetcher->delegate()->OnURLFetchComplete(fetcher);
   }
 
   void ResetBackOff() {
     whitelist_manager_->StopDownloadTimer();
@@ skipping 147 matching lines @@
                 GURL("https://cart.merchant2.com/ShippingInfo?a=b&c=d")));
   // Bypass other urls.
   EXPECT_EQ(std::string("https://bypass.me/"),
             whitelist_manager_->GetMatchedURLPrefix(
                 GURL("https://bypass.me/")));
 }
 
 }  // namespace autocheckout
 }  // namespace autofill