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 |