NQE: Use ResponseHeaders to get the HTTP status code

net/nqe/network_quality_estimator_test_util.h

 // 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.
 
 #ifndef NET_NQE_NETWORK_QUALITY_ESTIMATOR_TEST_UTIL_H_
 #define NET_NQE_NETWORK_QUALITY_ESTIMATOR_TEST_UTIL_H_
 
 #include <map>
 #include <memory>
 #include <string>
 #include <vector>
 
+#include "base/files/file_path.h"
 #include "base/macros.h"
 #include "base/optional.h"
 #include "base/time/time.h"
 #include "net/base/network_change_notifier.h"
 #include "net/nqe/effective_connection_type.h"
 #include "net/nqe/network_quality_estimator.h"
 #include "net/test/embedded_test_server/embedded_test_server.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #include "url/gurl.h"
 
@@ skipping 27 matching lines @@
 
   // Runs one URL request to completion.
   void RunOneRequest();
 
   // Overrides the current network type and id.
   // Notifies network quality estimator of a change in connection.
   void SimulateNetworkChange(
       NetworkChangeNotifier::ConnectionType new_connection_type,
       const std::string& network_id);
 
-  // Called by the embedded server when an HTTP request is received.
-  std::unique_ptr<test_server::HttpResponse> HandleRequest(
-      const test_server::HttpRequest& request);
-
   // Returns a GURL hosted at the embedded test server.
   const GURL GetEchoURL() const;
 
+  // Returns a GURL hosted at the embedded test server which contains redirect
+  // to another HTTPS URL.
+  const GURL GetRedirectURL() const;
+
   void set_effective_connection_type(EffectiveConnectionType type) {
     // Callers should not set effective connection type along with the
     // lower-layer metrics.
     DCHECK(!start_time_null_http_rtt_ && !recent_http_rtt_ &&
            !start_time_null_transport_rtt_ && !recent_transport_rtt_ &&
            !start_time_null_downlink_throughput_kbps_ &&
            !recent_downlink_throughput_kbps_);
     effective_connection_type_ = type;
   }
 
@@ skipping 96 matching lines @@
       const override;
 
   void set_rand_double(double rand_double) { rand_double_ = rand_double; }
 
   double RandDouble() const override;
 
   using NetworkQualityEstimator::SetTickClockForTesting;
   using NetworkQualityEstimator::OnConnectionTypeChanged;
 
  private:
+  class LocalHttpTestServer : public EmbeddedTestServer {
+   public:
+    explicit LocalHttpTestServer(const base::FilePath& document_root);
+  };
+
   // NetworkQualityEstimator implementation that returns the overridden
   // network
   // id (instead of invoking platform APIs).
   nqe::internal::NetworkID GetCurrentNetworkID() const override;
 
   // If set, GetEffectiveConnectionType() and GetRecentEffectiveConnectionType()
   // would return the set values, respectively.
   base::Optional<EffectiveConnectionType> effective_connection_type_;
   base::Optional<EffectiveConnectionType> recent_effective_connection_type_;
 
@@ skipping 17 matching lines @@
 
   // If set, GetRecentDownlinkThroughputKbps() would return one of the set
   // values. |start_time_null_downlink_throughput_kbps_| is returned if the
   // |start_time| is null. Otherwise, |recent_downlink_throughput_kbps_| is
   // returned.
   base::Optional<int32_t> start_time_null_downlink_throughput_kbps_;
   base::Optional<int32_t> recent_downlink_throughput_kbps_;
 
   double rand_double_;
 
-  EmbeddedTestServer embedded_test_server_;
+  LocalHttpTestServer embedded_test_server_;
 
   DISALLOW_COPY_AND_ASSIGN(TestNetworkQualityEstimator);
 };
 
 }  // namespace net
 
 #endif  // NET_NQE_NETWORK_QUALITY_ESTIMATOR_TEST_UTIL_H_