| OLD | NEW |
| (Empty) |
| 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 <string> | |
| 6 #include <vector> | |
| 7 #include "base/command_line.h" | |
| 8 #include "base/memory/scoped_ptr.h" | |
| 9 #include "base/memory/scoped_vector.h" | |
| 10 #include "base/prefs/testing_pref_service.h" | |
| 11 #include "chrome/browser/chrome_browser_main.h" | |
| 12 #include "chrome/browser/chrome_content_browser_client.h" | |
| 13 #include "chrome/common/chrome_switches.h" | |
| 14 #include "content/public/browser/content_browser_client.h" | |
| 15 #include "content/public/common/main_function_params.h" | |
| 16 #include "net/socket/client_socket_pool_base.h" | |
| 17 #include "testing/gtest/include/gtest/gtest.h" | |
| 18 | |
| 19 class BrowserMainTest : public testing::Test { | |
| 20 public: | |
| 21 BrowserMainTest() | |
| 22 : command_line_(CommandLine::NO_PROGRAM) { | |
| 23 ChromeBrowserMainParts::disable_enforcing_cookie_policies_for_tests_ = true; | |
| 24 } | |
| 25 | |
| 26 protected: | |
| 27 CommandLine command_line_; | |
| 28 }; | |
| 29 | |
| 30 TEST_F(BrowserMainTest, WarmConnectionFieldTrial_WarmestSocket) { | |
| 31 command_line_.AppendSwitchASCII(switches::kSocketReusePolicy, "0"); | |
| 32 | |
| 33 scoped_ptr<content::MainFunctionParams> params( | |
| 34 new content::MainFunctionParams(command_line_)); | |
| 35 scoped_ptr<content::BrowserMainParts> browser_main_parts( | |
| 36 content::GetContentClient()->browser()->CreateBrowserMainParts(*params)); | |
| 37 ChromeBrowserMainParts* chrome_main_parts = | |
| 38 static_cast<ChromeBrowserMainParts*>(browser_main_parts.get()); | |
| 39 EXPECT_TRUE(chrome_main_parts); | |
| 40 if (chrome_main_parts) { | |
| 41 base::FieldTrialList field_trial_list_(NULL); | |
| 42 chrome_main_parts->browser_field_trials_.WarmConnectionFieldTrial(); | |
| 43 EXPECT_EQ(0, net::GetSocketReusePolicy()); | |
| 44 } | |
| 45 } | |
| 46 | |
| 47 TEST_F(BrowserMainTest, WarmConnectionFieldTrial_Random) { | |
| 48 scoped_ptr<content::MainFunctionParams> params( | |
| 49 new content::MainFunctionParams(command_line_)); | |
| 50 scoped_ptr<content::BrowserMainParts> browser_main_parts( | |
| 51 content::GetContentClient()->browser()->CreateBrowserMainParts(*params)); | |
| 52 ChromeBrowserMainParts* chrome_main_parts = | |
| 53 static_cast<ChromeBrowserMainParts*>(browser_main_parts.get()); | |
| 54 EXPECT_TRUE(chrome_main_parts); | |
| 55 if (chrome_main_parts) { | |
| 56 const int kNumRuns = 1000; | |
| 57 for (int i = 0; i < kNumRuns; i++) { | |
| 58 base::FieldTrialList field_trial_list_(NULL); | |
| 59 chrome_main_parts->browser_field_trials_.WarmConnectionFieldTrial(); | |
| 60 int val = net::GetSocketReusePolicy(); | |
| 61 EXPECT_LE(val, 2); | |
| 62 EXPECT_GE(val, 0); | |
| 63 } | |
| 64 } | |
| 65 } | |
| 66 | |
| 67 #if GTEST_HAS_DEATH_TEST | |
| 68 TEST_F(BrowserMainTest, WarmConnectionFieldTrial_Invalid) { | |
| 69 command_line_.AppendSwitchASCII(switches::kSocketReusePolicy, "100"); | |
| 70 | |
| 71 scoped_ptr<content::MainFunctionParams> params( | |
| 72 new content::MainFunctionParams(command_line_)); | |
| 73 // This test ends up launching a new process, and that doesn't initialize the | |
| 74 // ContentClient interfaces. | |
| 75 scoped_ptr<content::BrowserMainParts> browser_main_parts; | |
| 76 if (content::GetContentClient()) { | |
| 77 browser_main_parts.reset(content::GetContentClient()->browser()-> | |
| 78 CreateBrowserMainParts(*params)); | |
| 79 } else { | |
| 80 chrome::ChromeContentBrowserClient client; | |
| 81 browser_main_parts.reset(client.CreateBrowserMainParts(*params)); | |
| 82 } | |
| 83 ChromeBrowserMainParts* parts = | |
| 84 static_cast<ChromeBrowserMainParts*>(browser_main_parts.get()); | |
| 85 EXPECT_TRUE(parts); | |
| 86 if (parts) { | |
| 87 #if defined(NDEBUG) && defined(DCHECK_ALWAYS_ON) | |
| 88 EXPECT_DEATH(parts->browser_field_trials_.WarmConnectionFieldTrial(), | |
| 89 "Not a valid socket reuse policy group"); | |
| 90 #else | |
| 91 EXPECT_DEBUG_DEATH(parts->browser_field_trials_.WarmConnectionFieldTrial(), | |
| 92 "Not a valid socket reuse policy group"); | |
| 93 #endif | |
| 94 } | |
| 95 } | |
| 96 #endif | |
| OLD | NEW |