OLD | NEW |
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 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 | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "base/memory/ref_counted.h" | 5 #include "base/memory/ref_counted.h" |
6 #include "base/strings/stringprintf.h" | 6 #include "base/strings/stringprintf.h" |
7 #include "chrome/browser/extensions/extension_apitest.h" | 7 #include "chrome/browser/extensions/extension_apitest.h" |
8 #include "chrome/browser/extensions/extension_service.h" | 8 #include "chrome/browser/extensions/extension_service.h" |
9 #include "chrome/browser/extensions/extension_test_message_listener.h" | 9 #include "chrome/browser/extensions/extension_test_message_listener.h" |
10 #include "chrome/browser/ui/browser.h" | 10 #include "chrome/browser/ui/browser.h" |
11 #include "chrome/common/chrome_paths.h" | 11 #include "chrome/common/chrome_paths.h" |
12 #include "chrome/test/base/in_process_browser_test.h" | 12 #include "chrome/test/base/in_process_browser_test.h" |
13 #include "chrome/test/base/ui_test_utils.h" | 13 #include "chrome/test/base/ui_test_utils.h" |
14 #include "extensions/browser/api/dns/host_resolver_wrapper.h" | 14 #include "extensions/browser/api/dns/host_resolver_wrapper.h" |
15 #include "extensions/browser/api/dns/mock_host_resolver_creator.h" | 15 #include "extensions/browser/api/dns/mock_host_resolver_creator.h" |
16 #include "extensions/browser/api/socket/socket_api.h" | 16 #include "extensions/browser/api/socket/socket_api.h" |
| 17 #include "extensions/test/result_catcher.h" |
17 #include "net/dns/mock_host_resolver.h" | 18 #include "net/dns/mock_host_resolver.h" |
18 #include "net/test/spawned_test_server/spawned_test_server.h" | 19 #include "net/test/spawned_test_server/spawned_test_server.h" |
19 | 20 |
20 using extensions::Extension; | 21 using extensions::Extension; |
| 22 using extensions::ResultCatcher; |
21 | 23 |
22 namespace { | 24 namespace { |
23 | 25 |
24 const std::string kHostname = "127.0.0.1"; | 26 const std::string kHostname = "127.0.0.1"; |
25 const int kPort = 8888; | 27 const int kPort = 8888; |
26 | 28 |
27 class SocketApiTest : public ExtensionApiTest { | 29 class SocketApiTest : public ExtensionApiTest { |
28 public: | 30 public: |
29 SocketApiTest() : resolver_event_(true, false), | 31 SocketApiTest() : resolver_event_(true, false), |
30 resolver_creator_( | 32 resolver_creator_( |
(...skipping 31 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
62 EXPECT_TRUE(test_server->Start()); | 64 EXPECT_TRUE(test_server->Start()); |
63 | 65 |
64 net::HostPortPair host_port_pair = test_server->host_port_pair(); | 66 net::HostPortPair host_port_pair = test_server->host_port_pair(); |
65 int port = host_port_pair.port(); | 67 int port = host_port_pair.port(); |
66 ASSERT_GT(port, 0); | 68 ASSERT_GT(port, 0); |
67 | 69 |
68 // Test that sendTo() is properly resolving hostnames. | 70 // Test that sendTo() is properly resolving hostnames. |
69 host_port_pair.set_host("LOCALhost"); | 71 host_port_pair.set_host("LOCALhost"); |
70 | 72 |
71 ResultCatcher catcher; | 73 ResultCatcher catcher; |
72 catcher.RestrictToProfile(browser()->profile()); | 74 catcher.RestrictToBrowserContext(browser()->profile()); |
73 | 75 |
74 ExtensionTestMessageListener listener("info_please", true); | 76 ExtensionTestMessageListener listener("info_please", true); |
75 | 77 |
76 ASSERT_TRUE(LoadExtension(test_data_dir_.AppendASCII("socket/api"))); | 78 ASSERT_TRUE(LoadExtension(test_data_dir_.AppendASCII("socket/api"))); |
77 EXPECT_TRUE(listener.WaitUntilSatisfied()); | 79 EXPECT_TRUE(listener.WaitUntilSatisfied()); |
78 listener.Reply( | 80 listener.Reply( |
79 base::StringPrintf("udp:%s:%d", host_port_pair.host().c_str(), port)); | 81 base::StringPrintf("udp:%s:%d", host_port_pair.host().c_str(), port)); |
80 | 82 |
81 EXPECT_TRUE(catcher.GetNextResult()) << catcher.message(); | 83 EXPECT_TRUE(catcher.GetNextResult()) << catcher.message(); |
82 } | 84 } |
83 | 85 |
84 IN_PROC_BROWSER_TEST_F(SocketApiTest, SocketTCPExtension) { | 86 IN_PROC_BROWSER_TEST_F(SocketApiTest, SocketTCPExtension) { |
85 scoped_ptr<net::SpawnedTestServer> test_server( | 87 scoped_ptr<net::SpawnedTestServer> test_server( |
86 new net::SpawnedTestServer( | 88 new net::SpawnedTestServer( |
87 net::SpawnedTestServer::TYPE_TCP_ECHO, | 89 net::SpawnedTestServer::TYPE_TCP_ECHO, |
88 net::SpawnedTestServer::kLocalhost, | 90 net::SpawnedTestServer::kLocalhost, |
89 base::FilePath(FILE_PATH_LITERAL("net/data")))); | 91 base::FilePath(FILE_PATH_LITERAL("net/data")))); |
90 EXPECT_TRUE(test_server->Start()); | 92 EXPECT_TRUE(test_server->Start()); |
91 | 93 |
92 net::HostPortPair host_port_pair = test_server->host_port_pair(); | 94 net::HostPortPair host_port_pair = test_server->host_port_pair(); |
93 int port = host_port_pair.port(); | 95 int port = host_port_pair.port(); |
94 ASSERT_GT(port, 0); | 96 ASSERT_GT(port, 0); |
95 | 97 |
96 // Test that connect() is properly resolving hostnames. | 98 // Test that connect() is properly resolving hostnames. |
97 host_port_pair.set_host("lOcAlHoSt"); | 99 host_port_pair.set_host("lOcAlHoSt"); |
98 | 100 |
99 ResultCatcher catcher; | 101 ResultCatcher catcher; |
100 catcher.RestrictToProfile(browser()->profile()); | 102 catcher.RestrictToBrowserContext(browser()->profile()); |
101 | 103 |
102 ExtensionTestMessageListener listener("info_please", true); | 104 ExtensionTestMessageListener listener("info_please", true); |
103 | 105 |
104 ASSERT_TRUE(LoadExtension(test_data_dir_.AppendASCII("socket/api"))); | 106 ASSERT_TRUE(LoadExtension(test_data_dir_.AppendASCII("socket/api"))); |
105 EXPECT_TRUE(listener.WaitUntilSatisfied()); | 107 EXPECT_TRUE(listener.WaitUntilSatisfied()); |
106 listener.Reply( | 108 listener.Reply( |
107 base::StringPrintf("tcp:%s:%d", host_port_pair.host().c_str(), port)); | 109 base::StringPrintf("tcp:%s:%d", host_port_pair.host().c_str(), port)); |
108 | 110 |
109 EXPECT_TRUE(catcher.GetNextResult()) << catcher.message(); | 111 EXPECT_TRUE(catcher.GetNextResult()) << catcher.message(); |
110 } | 112 } |
111 | 113 |
112 IN_PROC_BROWSER_TEST_F(SocketApiTest, SocketTCPServerExtension) { | 114 IN_PROC_BROWSER_TEST_F(SocketApiTest, SocketTCPServerExtension) { |
113 ResultCatcher catcher; | 115 ResultCatcher catcher; |
114 catcher.RestrictToProfile(browser()->profile()); | 116 catcher.RestrictToBrowserContext(browser()->profile()); |
115 ExtensionTestMessageListener listener("info_please", true); | 117 ExtensionTestMessageListener listener("info_please", true); |
116 ASSERT_TRUE(LoadExtension(test_data_dir_.AppendASCII("socket/api"))); | 118 ASSERT_TRUE(LoadExtension(test_data_dir_.AppendASCII("socket/api"))); |
117 EXPECT_TRUE(listener.WaitUntilSatisfied()); | 119 EXPECT_TRUE(listener.WaitUntilSatisfied()); |
118 listener.Reply( | 120 listener.Reply( |
119 base::StringPrintf("tcp_server:%s:%d", kHostname.c_str(), kPort)); | 121 base::StringPrintf("tcp_server:%s:%d", kHostname.c_str(), kPort)); |
120 | 122 |
121 EXPECT_TRUE(catcher.GetNextResult()) << catcher.message(); | 123 EXPECT_TRUE(catcher.GetNextResult()) << catcher.message(); |
122 } | 124 } |
123 | 125 |
124 IN_PROC_BROWSER_TEST_F(SocketApiTest, SocketTCPServerUnbindOnUnload) { | 126 IN_PROC_BROWSER_TEST_F(SocketApiTest, SocketTCPServerUnbindOnUnload) { |
125 ResultCatcher catcher; | 127 ResultCatcher catcher; |
126 const Extension* extension = | 128 const Extension* extension = |
127 LoadExtension(test_data_dir_.AppendASCII("socket/unload")); | 129 LoadExtension(test_data_dir_.AppendASCII("socket/unload")); |
128 ASSERT_TRUE(extension); | 130 ASSERT_TRUE(extension); |
129 EXPECT_TRUE(catcher.GetNextResult()) << catcher.message(); | 131 EXPECT_TRUE(catcher.GetNextResult()) << catcher.message(); |
130 | 132 |
131 UnloadExtension(extension->id()); | 133 UnloadExtension(extension->id()); |
132 | 134 |
133 ASSERT_TRUE(LoadExtension(test_data_dir_.AppendASCII("socket/unload"))) | 135 ASSERT_TRUE(LoadExtension(test_data_dir_.AppendASCII("socket/unload"))) |
134 << message_; | 136 << message_; |
135 EXPECT_TRUE(catcher.GetNextResult()) << catcher.message(); | 137 EXPECT_TRUE(catcher.GetNextResult()) << catcher.message(); |
136 } | 138 } |
137 | 139 |
138 IN_PROC_BROWSER_TEST_F(SocketApiTest, SocketMulticast) { | 140 IN_PROC_BROWSER_TEST_F(SocketApiTest, SocketMulticast) { |
139 ResultCatcher catcher; | 141 ResultCatcher catcher; |
140 catcher.RestrictToProfile(browser()->profile()); | 142 catcher.RestrictToBrowserContext(browser()->profile()); |
141 ExtensionTestMessageListener listener("info_please", true); | 143 ExtensionTestMessageListener listener("info_please", true); |
142 ASSERT_TRUE(LoadExtension(test_data_dir_.AppendASCII("socket/api"))); | 144 ASSERT_TRUE(LoadExtension(test_data_dir_.AppendASCII("socket/api"))); |
143 EXPECT_TRUE(listener.WaitUntilSatisfied()); | 145 EXPECT_TRUE(listener.WaitUntilSatisfied()); |
144 listener.Reply( | 146 listener.Reply( |
145 base::StringPrintf("multicast:%s:%d", kHostname.c_str(), kPort)); | 147 base::StringPrintf("multicast:%s:%d", kHostname.c_str(), kPort)); |
146 | 148 |
147 EXPECT_TRUE(catcher.GetNextResult()) << catcher.message(); | 149 EXPECT_TRUE(catcher.GetNextResult()) << catcher.message(); |
148 } | 150 } |
OLD | NEW |