NQE: Exclude network observations from private networks

net/nqe/throughput_analyzer_unittest.cc

 // Copyright 2016 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 "net/nqe/throughput_analyzer.h"
 
 #include <stdint.h>
 
 #include <deque>
 #include <map>
 #include <memory>
 #include <string>
+#include <utility>
+#include <vector>
 
 #include "base/bind.h"
 #include "base/bind_helpers.h"
 #include "base/logging.h"
 #include "base/macros.h"
+#include "base/memory/ptr_util.h"
 #include "base/run_loop.h"
 #include "base/single_thread_task_runner.h"
 #include "base/strings/string_number_conversions.h"
 #include "base/threading/thread_task_runner_handle.h"
-#include "net/base/url_util.h"
+#include "net/dns/mock_host_resolver.h"
+#include "net/log/test_net_log.h"
 #include "net/nqe/network_quality_estimator_params.h"
+#include "net/nqe/network_quality_estimator_util.h"
 #include "net/traffic_annotation/network_traffic_annotation_test_helper.h"
 #include "net/url_request/url_request.h"
 #include "net/url_request/url_request_test_util.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 namespace net {
 
 namespace nqe {
 
 namespace {
 
 class TestThroughputAnalyzer : public internal::ThroughputAnalyzer {
  public:
   explicit TestThroughputAnalyzer(NetworkQualityEstimatorParams* params)
       : internal::ThroughputAnalyzer(
             params,
             base::ThreadTaskRunnerHandle::Get(),
             base::Bind(
                 &TestThroughputAnalyzer::OnNewThroughputObservationAvailable,
                 base::Unretained(this)),
             false,
-            false),
+            false,
+            base::MakeUnique<BoundTestNetLog>()->bound()),
         throughput_observations_received_(0),
         bits_received_(0) {}
 
   ~TestThroughputAnalyzer() override {}
 
   int32_t throughput_observations_received() const {
     return throughput_observations_received_;
   }
 
   void OnNewThroughputObservationAvailable(int32_t downstream_kbps) {
     throughput_observations_received_++;
   }
 
   int64_t GetBitsReceived() const override { return bits_received_; }
 
   void IncrementBitsReceived(int64_t additional_bits_received) {
     bits_received_ += additional_bits_received;
   }
 
+  // Uses a mock resolver to force example.com to resolve to a public IP
+  // address.
+  void AddIPAddressResolution(TestURLRequestContext* context) {
+    scoped_refptr<net::RuleBasedHostResolverProc> rules(
+        new net::RuleBasedHostResolverProc(nullptr));
+    // example1.com resolves to a public IP address.
+    rules->AddRule("example.com", "27.0.0.3");
+    mock_host_resolver_.set_rules(rules.get());
+    context->set_host_resolver(&mock_host_resolver_);
+  }
+
   using internal::ThroughputAnalyzer::disable_throughput_measurements;
 
  private:
   int throughput_observations_received_;
 
   int64_t bits_received_;
 
+  MockCachingHostResolver mock_host_resolver_;
+
   DISALLOW_COPY_AND_ASSIGN(TestThroughputAnalyzer);
 };
 
 TEST(ThroughputAnalyzerTest, MaximumRequests) {
   const struct {
     bool use_local_requests;
   } tests[] = {{
                    false,
                },
                {
                    true,
                }};
 
   for (const auto& test : tests) {
     std::map<std::string, std::string> variation_params;
     NetworkQualityEstimatorParams params(variation_params);
     TestThroughputAnalyzer throughput_analyzer(&params);
 
     TestDelegate test_delegate;
     TestURLRequestContext context;
+    throughput_analyzer.AddIPAddressResolution(&context);
 
     ASSERT_FALSE(throughput_analyzer.disable_throughput_measurements());
     std::deque<std::unique_ptr<URLRequest>> requests;
 
     // Start more requests than the maximum number of requests that can be held
     // in the memory.
     const std::string url = test.use_local_requests
                                 ? "http://127.0.0.1/test.html"
                                 : "http://example.com/test.html";
+
+    EXPECT_EQ(test.use_local_requests,
+              nqe::internal::IsPrivateHost(
+                  context.host_resolver(),
+                  HostPortPair(GURL(url).host(), GURL(url).EffectiveIntPort()),
+                  base::MakeUnique<BoundTestNetLog>()->bound()));
     for (size_t i = 0; i < 1000; ++i) {
       std::unique_ptr<URLRequest> request(
           context.CreateRequest(GURL(url), DEFAULT_PRIORITY, &test_delegate,
                                 TRAFFIC_ANNOTATION_FOR_TESTS));
-      ASSERT_EQ(test.use_local_requests, IsLocalhost(request->url().host()));
-
       throughput_analyzer.NotifyStartTransaction(*(request.get()));
       requests.push_back(std::move(request));
     }
     // Too many local requests should cause the |throughput_analyzer| to disable
     // throughput measurements.
-    EXPECT_EQ(test.use_local_requests,
-              throughput_analyzer.disable_throughput_measurements());
+    EXPECT_NE(test.use_local_requests,
+              throughput_analyzer.IsCurrentlyTrackingThroughput());
   }
 }
 
 // Tests if the throughput observation is taken correctly when local and network
 // requests overlap.
 TEST(ThroughputAnalyzerTest, TestThroughputWithMultipleRequestsOverlap) {
   static const struct {
     bool start_local_request;
     bool local_request_completes_first;
     bool expect_throughput_observation;
@@ skipping 10 matching lines @@
   };
 
   for (const auto& test : tests) {
     // Localhost requests are not allowed for estimation purposes.
     std::map<std::string, std::string> variation_params;
     NetworkQualityEstimatorParams params(variation_params);
     TestThroughputAnalyzer throughput_analyzer(&params);
 
     TestDelegate test_delegate;
     TestURLRequestContext context;
+    throughput_analyzer.AddIPAddressResolution(&context);
 
     std::unique_ptr<URLRequest> request_local;
 
     std::unique_ptr<URLRequest> request_not_local(context.CreateRequest(
         GURL("http://example.com/echo.html"), DEFAULT_PRIORITY, &test_delegate,
         TRAFFIC_ANNOTATION_FOR_TESTS));
     request_not_local->Start();
 
     if (test.start_local_request) {
-      request_local = context.CreateRequest(GURL("http://localhost/echo.html"),
+      request_local = context.CreateRequest(GURL("http://127.0.0.1/echo.html"),
                                             DEFAULT_PRIORITY, &test_delegate,
                                             TRAFFIC_ANNOTATION_FOR_TESTS);
       request_local->Start();
     }
 
     base::RunLoop().Run();
 
     EXPECT_EQ(0, throughput_analyzer.throughput_observations_received());
 
     // If |test.start_local_request| is true, then |request_local| starts
@@ skipping 58 matching lines @@
 
   for (const auto& test : tests) {
     // Localhost requests are not allowed for estimation purposes.
     std::map<std::string, std::string> variation_params;
     variation_params["throughput_min_requests_in_flight"] =
         base::IntToString(test.throughput_min_requests_in_flight);
     NetworkQualityEstimatorParams params(variation_params);
     TestThroughputAnalyzer throughput_analyzer(&params);
     TestDelegate test_delegate;
     TestURLRequestContext context;
+    throughput_analyzer.AddIPAddressResolution(&context);
 
     EXPECT_EQ(0, throughput_analyzer.throughput_observations_received());
 
     std::vector<std::unique_ptr<URLRequest>> requests_in_flight;
 
     for (size_t i = 0; i < test.number_requests_in_flight; ++i) {
       std::unique_ptr<URLRequest> request_network_1 = context.CreateRequest(
           GURL("http://example.com/echo.html"), DEFAULT_PRIORITY,
           &test_delegate, TRAFFIC_ANNOTATION_FOR_TESTS);
       requests_in_flight.push_back(std::move(request_network_1));
@@ skipping 32 matching lines @@
 // Tests if the throughput observation is taken correctly when the start and end
 // of network requests overlap, and the minimum number of in flight requests
 // when taking an observation is more than 1.
 TEST(ThroughputAnalyzerTest, TestThroughputWithMultipleNetworkRequests) {
   std::map<std::string, std::string> variation_params;
   variation_params["throughput_min_requests_in_flight"] = "3";
   NetworkQualityEstimatorParams params(variation_params);
   TestThroughputAnalyzer throughput_analyzer(&params);
   TestDelegate test_delegate;
   TestURLRequestContext context;
+  throughput_analyzer.AddIPAddressResolution(&context);
 
   EXPECT_EQ(0, throughput_analyzer.throughput_observations_received());
 
   std::unique_ptr<URLRequest> request_1 = context.CreateRequest(
       GURL("http://example.com/echo.html"), DEFAULT_PRIORITY, &test_delegate,
       TRAFFIC_ANNOTATION_FOR_TESTS);
   std::unique_ptr<URLRequest> request_2 = context.CreateRequest(
       GURL("http://example.com/echo.html"), DEFAULT_PRIORITY, &test_delegate,
       TRAFFIC_ANNOTATION_FOR_TESTS);
   std::unique_ptr<URLRequest> request_3 = context.CreateRequest(
@@ skipping 42 matching lines @@
   throughput_analyzer.NotifyRequestCompleted(*(request_3.get()));
   throughput_analyzer.NotifyRequestCompleted(*(request_4.get()));
   EXPECT_EQ(1, throughput_analyzer.throughput_observations_received());
 }
 
 }  // namespace
 
 }  // namespace nqe
 
 }  // namespace net