chrome/browser/net/connection_tester_unittest.cc

   1 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "chrome/browser/net/connection_tester.h"
   6
   7 #include "base/prefs/testing_pref_service.h"
   8 #include "content/public/browser/cookie_store_factory.h"
   9 #include "content/public/test/test_browser_thread.h"
  10 #include "content/public/browser/cookie_store_factory.h"
  11 #include "net/cert/mock_cert_verifier.h"
  12 #include "net/dns/mock_host_resolver.h"
  13 #include "net/ftp/ftp_network_layer.h"
  14 #include "net/http/http_auth_handler_factory.h"
  15 #include "net/http/http_network_layer.h"
  16 #include "net/http/http_network_session.h"
  17 #include "net/http/http_server_properties_impl.h"
  18 #include "net/http/transport_security_state.h"
  19 #include "net/proxy/proxy_config_service_fixed.h"
  20 #include "net/proxy/proxy_service.h"
  21 #include "net/ssl/ssl_config_service_defaults.h"
  22 #include "net/test/spawned_test_server/spawned_test_server.h"
  23 #include "net/url_request/url_request_context.h"
  24 #include "net/url_request/url_request_test_util.h"
  25 #include "testing/gtest/include/gtest/gtest.h"
  26 #include "testing/platform_test.h"
  27
  28 using content::BrowserThread;
  29
  30 namespace {
  31
  32 // This is a testing delegate which simply counts how many times each of
  33 // the delegate's methods were invoked.
  34 class ConnectionTesterDelegate : public ConnectionTester::Delegate {
  35  public:
  36   ConnectionTesterDelegate()
  37      : start_connection_test_suite_count_(0),
  38        start_connection_test_experiment_count_(0),
  39        completed_connection_test_experiment_count_(0),
  40        completed_connection_test_suite_count_(0) {
  41   }
  42
  43   void OnStartConnectionTestSuite() override {
  44     start_connection_test_suite_count_++;
  45   }
  46
  47   void OnStartConnectionTestExperiment(
  48       const ConnectionTester::Experiment& experiment) override {
  49     start_connection_test_experiment_count_++;
  50   }
  51
  52   void OnCompletedConnectionTestExperiment(
  53       const ConnectionTester::Experiment& experiment,
  54       int result) override {
  55     completed_connection_test_experiment_count_++;
  56   }
  57
  58   void OnCompletedConnectionTestSuite() override {
  59     completed_connection_test_suite_count_++;
  60     base::MessageLoop::current()->Quit();
  61   }
  62
  63   int start_connection_test_suite_count() const {
  64     return start_connection_test_suite_count_;
  65   }
  66
  67   int start_connection_test_experiment_count() const {
  68     return start_connection_test_experiment_count_;
  69   }
  70
  71   int completed_connection_test_experiment_count() const {
  72     return completed_connection_test_experiment_count_;
  73   }
  74
  75   int completed_connection_test_suite_count() const {
  76     return completed_connection_test_suite_count_;
  77   }
  78
  79  private:
  80   int start_connection_test_suite_count_;
  81   int start_connection_test_experiment_count_;
  82   int completed_connection_test_experiment_count_;
  83   int completed_connection_test_suite_count_;
  84 };
  85
  86 // The test fixture is responsible for:
  87 //   - Making sure each test has an IO loop running
  88 //   - Catching any host resolve requests and mapping them to localhost
  89 //     (so the test doesn't use any external network dependencies).
  90 class ConnectionTesterTest : public PlatformTest {
  91  public:
  92   ConnectionTesterTest()
  93       : io_thread_(BrowserThread::IO, &message_loop_),
  94         test_server_(net::SpawnedTestServer::TYPE_HTTP,
  95                      net::SpawnedTestServer::kLocalhost,
  96                      // Nothing is read in this directory.
  97                      base::FilePath(FILE_PATH_LITERAL("chrome"))) {
  98   }
  99
 100  protected:
 101   // Destroy last the MessageLoop last to give a chance for objects like
 102   // ObserverListThreadSave to shut down properly.  For example,
 103   // SSLClientAuthCache calls RemoveObserver when destroyed, but if the
 104   // MessageLoop is already destroyed, then the RemoveObserver will be a
 105   // no-op, and the ObserverList will contain invalid entries.
 106   base::MessageLoopForIO message_loop_;
 107   content::TestBrowserThread io_thread_;
 108   net::SpawnedTestServer test_server_;
 109   net::TestURLRequestContext proxy_script_fetcher_context_;
 110   ConnectionTesterDelegate test_delegate_;
 111 };
 112
 113 TEST_F(ConnectionTesterTest, RunAllTests) {
 114   ASSERT_TRUE(test_server_.Start());
 115
 116   ConnectionTester tester(&test_delegate_,
 117                           &proxy_script_fetcher_context_,
 118                           NULL);
 119
 120   // Start the test suite on URL "echoall".
 121   // TODO(eroman): Is this URL right?
 122   tester.RunAllTests(test_server_.GetURL("echoall"));
 123
 124   // Wait for all the tests to complete.
 125   base::MessageLoop::current()->Run();
 126
 127   const int kNumExperiments =
 128       ConnectionTester::PROXY_EXPERIMENT_COUNT *
 129       ConnectionTester::HOST_RESOLVER_EXPERIMENT_COUNT;
 130
 131   EXPECT_EQ(1, test_delegate_.start_connection_test_suite_count());
 132   EXPECT_EQ(kNumExperiments,
 133             test_delegate_.start_connection_test_experiment_count());
 134   EXPECT_EQ(kNumExperiments,
 135             test_delegate_.completed_connection_test_experiment_count());
 136   EXPECT_EQ(1, test_delegate_.completed_connection_test_suite_count());
 137 }
 138
 139 TEST_F(ConnectionTesterTest, DeleteWhileInProgress) {
 140   ASSERT_TRUE(test_server_.Start());
 141
 142   scoped_ptr<ConnectionTester> tester(
 143       new ConnectionTester(&test_delegate_,
 144                            &proxy_script_fetcher_context_,
 145                            NULL));
 146
 147   // Start the test suite on URL "echoall".
 148   // TODO(eroman): Is this URL right?
 149   tester->RunAllTests(test_server_.GetURL("echoall"));
 150
 151   // Don't run the message loop at all.  Otherwise the experiment's request may
 152   // complete and post a task to run the next experiment before we quit the
 153   // message loop.
 154
 155   EXPECT_EQ(1, test_delegate_.start_connection_test_suite_count());
 156   EXPECT_EQ(1, test_delegate_.start_connection_test_experiment_count());
 157   EXPECT_EQ(0, test_delegate_.completed_connection_test_experiment_count());
 158   EXPECT_EQ(0, test_delegate_.completed_connection_test_suite_count());
 159
 160   // Delete the ConnectionTester while it is in progress.
 161   tester.reset();
 162
 163   // Drain the tasks on the message loop.
 164   //
 165   // Note that we cannot simply stop the message loop, since that will delete
 166   // any pending tasks instead of running them. This causes a problem with
 167   // net::ClientSocketPoolBaseHelper, since the "Group" holds a pointer
 168   // |backup_task| that it will try to deref during the destructor, but
 169   // depending on the order that pending tasks were deleted in, it might
 170   // already be invalid! See http://crbug.com/43291.
 171   base::MessageLoop::current()->PostTask(FROM_HERE,
 172                                          base::MessageLoop::QuitClosure());
 173   base::MessageLoop::current()->Run();
 174 }
 175
 176 }  // namespace