OLD | NEW |
| (Empty) |
1 // Copyright (c) 2011 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/browser_main.h" | |
6 | |
7 #include <string> | |
8 #include <vector> | |
9 #include "base/command_line.h" | |
10 #include "chrome/common/chrome_switches.h" | |
11 #include "chrome/test/testing_pref_service.h" | |
12 #include "content/common/main_function_params.h" | |
13 #include "content/common/sandbox_init_wrapper.h" | |
14 #include "net/socket/client_socket_pool_base.h" | |
15 #include "testing/gtest/include/gtest/gtest.h" | |
16 | |
17 class BrowserMainTest : public testing::Test { | |
18 public: | |
19 BrowserMainTest() : command_line_(CommandLine::NO_PROGRAM) {} | |
20 protected: | |
21 virtual void SetUp() { | |
22 sandbox_init_wrapper_.reset(new SandboxInitWrapper()); | |
23 } | |
24 | |
25 scoped_ptr<SandboxInitWrapper> sandbox_init_wrapper_; | |
26 TestingPrefService pref_service_; | |
27 CommandLine command_line_; | |
28 }; | |
29 | |
30 TEST_F(BrowserMainTest, WarmConnectionFieldTrial_WarmestSocket) { | |
31 command_line_.AppendSwitchASCII(switches::kSocketReusePolicy, "0"); | |
32 | |
33 scoped_ptr<MainFunctionParams> params( | |
34 new MainFunctionParams(command_line_, *sandbox_init_wrapper_, NULL)); | |
35 scoped_ptr<BrowserMainParts> bw(BrowserMainParts::CreateBrowserMainParts( | |
36 *params)); | |
37 | |
38 bw->WarmConnectionFieldTrial(); | |
39 | |
40 EXPECT_EQ(0, net::GetSocketReusePolicy()); | |
41 } | |
42 | |
43 TEST_F(BrowserMainTest, WarmConnectionFieldTrial_Random) { | |
44 scoped_ptr<MainFunctionParams> params( | |
45 new MainFunctionParams(command_line_, *sandbox_init_wrapper_, NULL)); | |
46 scoped_ptr<BrowserMainParts> bw(BrowserMainParts::CreateBrowserMainParts( | |
47 *params)); | |
48 | |
49 const int kNumRuns = 1000; | |
50 for (int i = 0; i < kNumRuns; i++) { | |
51 bw->WarmConnectionFieldTrial(); | |
52 int val = net::GetSocketReusePolicy(); | |
53 EXPECT_LE(val, 2); | |
54 EXPECT_GE(val, 0); | |
55 } | |
56 } | |
57 | |
58 TEST_F(BrowserMainTest, WarmConnectionFieldTrial_Invalid) { | |
59 command_line_.AppendSwitchASCII(switches::kSocketReusePolicy, "100"); | |
60 | |
61 scoped_ptr<MainFunctionParams> params( | |
62 new MainFunctionParams(command_line_, *sandbox_init_wrapper_, NULL)); | |
63 scoped_ptr<BrowserMainParts> bw(BrowserMainParts::CreateBrowserMainParts( | |
64 *params)); | |
65 | |
66 EXPECT_DEBUG_DEATH(bw->WarmConnectionFieldTrial(), | |
67 "Not a valid socket reuse policy group"); | |
68 } | |
OLD | NEW |