Implement UMA log throttling for cellular connections

components/data_use_measurement/content/data_use_measurement_unittest.cc

 // Copyright 2015 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 "components/data_use_measurement/content/data_use_measurement.h"
 
 #include <string>
 
 #include "base/macros.h"
 #include "base/memory/scoped_ptr.h"
 #include "base/test/histogram_tester.h"
 #include "build/build_config.h"
 #include "content/public/browser/resource_request_info.h"
 #include "net/base/network_change_notifier.h"
 #include "net/base/request_priority.h"
 #include "net/socket/socket_test_util.h"
 #include "net/url_request/url_request.h"
 #include "net/url_request/url_request_test_util.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #include "url/gurl.h"
 
 #if defined(OS_ANDROID)
 #include "base/android/application_status_listener.h"
 #endif
 
 namespace data_use_measurement {
 
 class DataUseMeasurementTest : public testing::Test {
  public:
-  DataUseMeasurementTest() {
+  DataUseMeasurementTest()
+      : data_use_measurement_(
+            base::Bind(&DataUseMeasurementTest::FakeDataUseforwarder,
+                       base::Unretained(this))) {
     // During the test it is expected to not have cellular connection.
     DCHECK(!net::NetworkChangeNotifier::IsConnectionCellular(
         net::NetworkChangeNotifier::GetConnectionType()));
   }
 
-  // This function makes a user request and confirms that its effect is
-  // reflected in proper histograms.
-  void TestForAUserRequest(const std::string& target_dimension) {
+  // Sends a request and reports data use attaching either user data or service
+  // data based on |is_user_request|.
+  void SendRequest(bool is_user_request) {
     net::TestDelegate test_delegate;
     InitializeContext();
-    base::HistogramTester histogram_tester;
     net::MockRead reads[] = {net::MockRead("HTTP/1.1 200 OK\r\n"
                                            "Content-Length: 12\r\n\r\n"),
                              net::MockRead("Test Content")};
     net::StaticSocketDataProvider socket_data(reads, arraysize(reads), nullptr,
                                               0);
     socket_factory_->AddSocketDataProvider(&socket_data);
 
     scoped_ptr<net::URLRequest> request(context_->CreateRequest(
         GURL("http://foo.com"), net::DEFAULT_PRIORITY, &test_delegate));
-    request->SetUserData(
-        data_use_measurement::DataUseUserData::kUserDataKey,
-        new data_use_measurement::DataUseUserData(
-            data_use_measurement::DataUseUserData::SUGGESTIONS));
+    if (is_user_request) {
+      request->SetUserData(
+          data_use_measurement::DataUseUserData::kUserDataKey,
+          new data_use_measurement::DataUseUserData(
+              data_use_measurement::DataUseUserData::SUGGESTIONS));
+    } else {
+      content::ResourceRequestInfo::AllocateForTesting(
+          request.get(), content::RESOURCE_TYPE_MAIN_FRAME, nullptr, -2, -2, -2,
+          true, false, true, true, false);
+    }
+
     request->Start();
     loop_.RunUntilIdle();
 
     data_use_measurement_.ReportDataUseUMA(request.get());
+  }
+
+  // This function makes a user request and confirms that its effect is
+  // reflected in proper histograms.
+  void TestForAUserRequest(const std::string& target_dimension) {
+    base::HistogramTester histogram_tester;
+    SendRequest(true);
     histogram_tester.ExpectTotalCount("DataUse.TrafficSize.System.Downstream." +
                                           target_dimension + kConnectionType,
                                       1);
     histogram_tester.ExpectTotalCount("DataUse.TrafficSize.System.Upstream." +
                                           target_dimension + kConnectionType,
                                       1);
     // One upload and one download message, so total count should be 2.
     histogram_tester.ExpectTotalCount("DataUse.MessageSize.Suggestions", 2);
   }
 
   // This function makes a service request and confirms that its effect is
   // reflected in proper histograms.
   void TestForAServiceRequest(const std::string& target_dimension) {
-    net::TestDelegate test_delegate;
-    InitializeContext();
     base::HistogramTester histogram_tester;
-
-    net::MockRead reads[] = {net::MockRead("HTTP/1.1 200 OK\r\n"
-                                           "Content-Length: 12\r\n\r\n"),
-                             net::MockRead("Test Content")};
-    net::StaticSocketDataProvider socket_data(reads, arraysize(reads), nullptr,
-                                              0);
-    socket_factory_->AddSocketDataProvider(&socket_data);
-
-    scoped_ptr<net::URLRequest> request(context_->CreateRequest(
-        GURL("http://foo.com"), net::DEFAULT_PRIORITY, &test_delegate));
-    content::ResourceRequestInfo::AllocateForTesting(
-        request.get(), content::RESOURCE_TYPE_MAIN_FRAME, nullptr, -2, -2, -2,
-        true, false, true, true, false);
-    request->Start();
-    loop_.RunUntilIdle();
-
-    data_use_measurement_.ReportDataUseUMA(request.get());
+    SendRequest(false);
     histogram_tester.ExpectTotalCount("DataUse.TrafficSize.User.Downstream." +
                                           target_dimension + kConnectionType,
                                       1);
     histogram_tester.ExpectTotalCount("DataUse.TrafficSize.User.Upstream." +
                                           target_dimension + kConnectionType,
                                       1);
     histogram_tester.ExpectTotalCount(
         "DataUse.MessageSize.AllServices.Upstream." + target_dimension +
             kConnectionType,
         0);
     histogram_tester.ExpectTotalCount(
         "DataUse.MessageSize.AllServices.Downstream." + target_dimension +
             kConnectionType,
         0);
   }
 
   DataUseMeasurement* data_use_measurement() { return &data_use_measurement_; }
 
+  bool IsDataUseForwarderCalled() { return is_data_use_forwarder_called_; }
+
  private:
   void InitializeContext() {
     context_.reset(new net::TestURLRequestContext(true));
     socket_factory_.reset(new net::MockClientSocketFactory());
     context_->set_client_socket_factory(socket_factory_.get());
     context_->Init();
   }
 
+  void FakeDataUseforwarder(const std::string& service_name,
+                            int message_size,
+                            bool is_celllular) {
+    is_data_use_forwarder_called_ = true;
+  }
+
   base::MessageLoopForIO loop_;
   DataUseMeasurement data_use_measurement_;
   scoped_ptr<net::MockClientSocketFactory> socket_factory_;
   scoped_ptr<net::TestURLRequestContext> context_;
   const std::string kConnectionType = "NotCellular";
+  bool is_data_use_forwarder_called_ = false;
 
   DISALLOW_COPY_AND_ASSIGN(DataUseMeasurementTest);
 };
 
 // This test function tests recording of data use information in UMA histogram
 // when packet is originated from user or services when the app is in the
 // foreground or the OS is not Android.
 // TODO(amohammadkhan): Add tests for Cellular/non-cellular connection types
 // when support for testing is provided in its class.
 TEST_F(DataUseMeasurementTest, UserNotUserTest) {
@@ skipping 10 matching lines @@
 // when packet is originated from user or services when the app is in the
 // background and OS is Android.
 TEST_F(DataUseMeasurementTest, ApplicationStateTest) {
   data_use_measurement()->OnApplicationStateChangeForTesting(
       base::android::APPLICATION_STATE_HAS_STOPPED_ACTIVITIES);
   TestForAServiceRequest("Background.");
   TestForAUserRequest("Background.");
 }
 #endif
 
+TEST_F(DataUseMeasurementTest, DataUseForwarderIsCalled) {
+  EXPECT_FALSE(IsDataUseForwarderCalled());
+  SendRequest(true);
+  EXPECT_TRUE(IsDataUseForwarderCalled());
+}
+
 }  // namespace data_use_measurement