Store network quality samples so we can compute percentiles.

net/base/network_quality_estimator_unittest.cc

Index: net/base/network_quality_estimator_unittest.cc
diff --git a/net/base/network_quality_estimator_unittest.cc b/net/base/network_quality_estimator_unittest.cc
index 82e1bbaf3d23e58ce960121170ad5d917e83e6f9..a39324a28a2d6582edca266676fe910afa02400b 100644
--- a/net/base/network_quality_estimator_unittest.cc
+++ b/net/base/network_quality_estimator_unittest.cc
@@ -32,9 +32,9 @@ TEST(NetworkQualityEstimatorTest, TestPeakKbpsFastestRTTUpdates) {
   // Enable requests to local host to be used for network quality estimation.
   NetworkQualityEstimator estimator(true);
   {
-    NetworkQuality network_quality = estimator.GetEstimate();
-    EXPECT_EQ(network_quality.fastest_rtt_confidence, 0);
-    EXPECT_EQ(network_quality.peak_throughput_kbps_confidence, 0);
+    NetworkQuality network_quality = estimator.GetPeakEstimate();
+    EXPECT_EQ(network_quality.rtt(), base::TimeDelta());
+    EXPECT_EQ(network_quality.downstream_throughput_kbps(), 0);
   }
 
   TestDelegate d;
@@ -53,30 +53,28 @@ TEST(NetworkQualityEstimatorTest, TestPeakKbpsFastestRTTUpdates) {
 
   base::PlatformThread::Sleep(request_duration);
 
-  // With smaller transfer, |fastest_rtt| will be updated but not
-  // |peak_throughput_kbps|.
-  estimator.NotifyDataReceived(*(request.get()),
+  // With smaller transfer, RTT will be updated but not the downstream
+  // throughput.
+  estimator.NotifyDataReceived(*(request.get()), min_transfer_size_in_bytes - 1,
                                min_transfer_size_in_bytes - 1);
   {
-    NetworkQuality network_quality = estimator.GetEstimate();
-    EXPECT_GT(network_quality.fastest_rtt_confidence, 0);
-    EXPECT_EQ(network_quality.peak_throughput_kbps_confidence, 0);
+    NetworkQuality network_quality = estimator.GetPeakEstimate();
+    EXPECT_GT(network_quality.rtt(), base::TimeDelta());
+    EXPECT_EQ(network_quality.downstream_throughput_kbps(), 0);
   }
 
-  // With large transfer, both |fastest_rtt| and |peak_throughput_kbps| will be
-  // updated.
-  estimator.NotifyDataReceived(*(request.get()), min_transfer_size_in_bytes);
+  // With large transfer, both RTT and downlink throughput will be updated.
+  estimator.NotifyDataReceived(*(request.get()), min_transfer_size_in_bytes,
+                               min_transfer_size_in_bytes);
   {
-    NetworkQuality network_quality = estimator.GetEstimate();
-    EXPECT_GT(network_quality.fastest_rtt_confidence, 0);
-    EXPECT_GT(network_quality.peak_throughput_kbps_confidence, 0);
-    EXPECT_GE(network_quality.fastest_rtt, request_duration);
-    EXPECT_GT(network_quality.peak_throughput_kbps, uint32_t(0));
+    NetworkQuality network_quality = estimator.GetPeakEstimate();
+    EXPECT_GE(network_quality.rtt(), request_duration);
+    EXPECT_GT(network_quality.downstream_throughput_kbps(), 0);
     EXPECT_LE(
-        network_quality.peak_throughput_kbps,
+        network_quality.downstream_throughput_kbps(),
         min_transfer_size_in_bytes * 8.0 / request_duration.InMilliseconds());
   }
-  EXPECT_EQ(estimator.bytes_read_since_last_connection_change_, true);
+  EXPECT_EQ(estimator.have_received_bytes_since_last_connection_change_, true);
 
   // Check UMA histograms.
   base::HistogramTester histogram_tester;
@@ -88,13 +86,71 @@ TEST(NetworkQualityEstimatorTest, TestPeakKbpsFastestRTTUpdates) {
   histogram_tester.ExpectTotalCount("NQE.PeakKbps.Unknown", 1);
   histogram_tester.ExpectTotalCount("NQE.FastestRTT.Unknown", 1);
   {
-    NetworkQuality network_quality = estimator.GetEstimate();
+    NetworkQuality network_quality = estimator.GetPeakEstimate();
     EXPECT_EQ(estimator.current_connection_type_,
               NetworkChangeNotifier::ConnectionType::CONNECTION_WIFI);
-    EXPECT_EQ(network_quality.fastest_rtt_confidence, 0);
-    EXPECT_EQ(network_quality.peak_throughput_kbps_confidence, 0);
+    EXPECT_EQ(network_quality.rtt(), base::TimeDelta());
+    EXPECT_EQ(network_quality.downstream_throughput_kbps(), 0);
   }
 }
 #endif  // !defined(OS_IOS)
 
+// SpawnedTestServer not supported on iOS (see http://crbug.com/148666).
+#if !defined(OS_IOS)
+TEST(NetworkQualityEstimatorTest, StoreObservations) {
+  SpawnedTestServer test_server_(
+      SpawnedTestServer::TYPE_HTTP, SpawnedTestServer::kLocalhost,
+      base::FilePath(FILE_PATH_LITERAL("net/data/url_request_unittest")));
+  ASSERT_TRUE(test_server_.Start());
+
+  NetworkQualityEstimator estimator(true);
+  TestDelegate d;
+  TestURLRequestContext context(false);
+
+  uint64 min_transfer_size_in_bytes =
+      NetworkQualityEstimator::kMinTransferSizeInBytes;
+  base::TimeDelta request_duration = base::TimeDelta::FromMicroseconds(
+      NetworkQualityEstimator::kMinRequestDurationMicroseconds);
+
+  // Push more observations than the maximum buffer size.
+  for (int i = 1; i <= 1000; ++i) {
+    scoped_ptr<URLRequest> request(context.CreateRequest(
+        test_server_.GetURL("echo.html"), DEFAULT_PRIORITY, &d));
+    request->Start();
+
+    base::RunLoop().Run();
+
+    base::PlatformThread::Sleep(request_duration);
+
+    estimator.NotifyDataReceived(*(request.get()), min_transfer_size_in_bytes,
+                                 min_transfer_size_in_bytes);
+  }
+
+  // Number of observations should be < 1000 (Assuming buffer size is less than
+  // 1000).
+  EXPECT_GT(estimator.kbps_observations_.size(), 0U);
+  EXPECT_LT(estimator.kbps_observations_.size(), 1000U);
+}
+#endif  // !defined(OS_IOS)
+
+TEST(NetworkQualityEstimatorTest, TestAddObservation) {
+  NetworkQualityEstimator estimator(true);
+
+  EXPECT_EQ(0U, estimator.kbps_observations_.size());
+  estimator.kbps_observations_.AddObservation(0);
+  EXPECT_EQ(1U, estimator.kbps_observations_.size());
+
+  estimator.kbps_observations_.AddObservation(-1);
+  EXPECT_EQ(1U, estimator.kbps_observations_.size());
+
+  estimator.kbps_observations_.AddObservation(INT32_MAX);
+  EXPECT_EQ(2U, estimator.kbps_observations_.size());
+
+  estimator.kbps_observations_.AddObservation(INT32_MIN);
+  EXPECT_EQ(2U, estimator.kbps_observations_.size());
+
+  estimator.kbps_observations_.AddObservation(INT32_MAX + 1);
+  EXPECT_EQ(2U, estimator.kbps_observations_.size());
+}
+
 }  // namespace net