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 virtual void OnStartConnectionTestSuite() OVERRIDE { 44 start_connection_test_suite_count_++; 45 } 46 47 virtual void OnStartConnectionTestExperiment( 48 const ConnectionTester::Experiment& experiment) OVERRIDE { 49 start_connection_test_experiment_count_++; 50 } 51 52 virtual void OnCompletedConnectionTestExperiment( 53 const ConnectionTester::Experiment& experiment, 54 int result) OVERRIDE { 55 completed_connection_test_experiment_count_++; 56 } 57 58 virtual 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 177