OLD | NEW |
1 // Copyright (c) 2010 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2010 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 "net/base/cert_verifier.h" | 5 #include "net/base/cert_verifier.h" |
6 #include "net/base/mock_host_resolver.h" | 6 #include "net/base/mock_host_resolver.h" |
7 #include "net/base/net_log.h" | 7 #include "net/base/net_log.h" |
8 #include "net/base/ssl_config_service_defaults.h" | 8 #include "net/base/ssl_config_service_defaults.h" |
9 #include "net/http/http_network_layer.h" | 9 #include "net/http/http_network_layer.h" |
| 10 #include "net/http/http_network_session.h" |
10 #include "net/http/http_transaction_unittest.h" | 11 #include "net/http/http_transaction_unittest.h" |
11 #include "net/proxy/proxy_service.h" | 12 #include "net/proxy/proxy_service.h" |
12 #include "net/socket/socket_test_util.h" | 13 #include "net/socket/socket_test_util.h" |
| 14 #include "net/spdy/spdy_session_pool.h" |
13 #include "testing/gtest/include/gtest/gtest.h" | 15 #include "testing/gtest/include/gtest/gtest.h" |
14 #include "testing/platform_test.h" | 16 #include "testing/platform_test.h" |
15 | 17 |
16 namespace net { | 18 namespace net { |
17 | 19 |
18 namespace { | 20 namespace { |
19 | 21 |
20 class HttpNetworkLayerTest : public PlatformTest { | 22 class HttpNetworkLayerTest : public PlatformTest { |
21 }; | 23 }; |
22 | 24 |
23 TEST_F(HttpNetworkLayerTest, CreateAndDestroy) { | 25 TEST_F(HttpNetworkLayerTest, CreateAndDestroy) { |
| 26 MockClientSocketFactory mock_socket_factory; |
24 MockHostResolver host_resolver; | 27 MockHostResolver host_resolver; |
25 net::CertVerifier cert_verifier; | 28 CertVerifier cert_verifier; |
26 net::HttpNetworkLayer factory( | 29 scoped_refptr<HttpNetworkSession> network_session( |
27 NULL, | 30 new HttpNetworkSession( |
28 &host_resolver, | 31 &host_resolver, |
29 &cert_verifier, | 32 &cert_verifier, |
30 NULL /* dnsrr_resolver */, | 33 NULL /* dnsrr_resolver */, |
31 NULL /* dns_cert_checker */, | 34 NULL /* dns_cert_checker */, |
32 NULL /* ssl_host_info_factory */, | 35 NULL /* ssl_host_info_factory */, |
33 net::ProxyService::CreateDirect(), | 36 ProxyService::CreateDirect(), |
34 new net::SSLConfigServiceDefaults, | 37 &mock_socket_factory, |
35 NULL, | 38 new SSLConfigServiceDefaults, |
36 NULL, | 39 new SpdySessionPool(NULL), |
37 NULL); | 40 NULL, |
| 41 NULL, |
| 42 NULL)); |
38 | 43 |
39 scoped_ptr<net::HttpTransaction> trans; | 44 HttpNetworkLayer factory(network_session); |
| 45 |
| 46 scoped_ptr<HttpTransaction> trans; |
40 int rv = factory.CreateTransaction(&trans); | 47 int rv = factory.CreateTransaction(&trans); |
41 EXPECT_EQ(net::OK, rv); | 48 EXPECT_EQ(OK, rv); |
42 EXPECT_TRUE(trans.get() != NULL); | 49 EXPECT_TRUE(trans.get() != NULL); |
43 } | 50 } |
44 | 51 |
45 TEST_F(HttpNetworkLayerTest, Suspend) { | 52 TEST_F(HttpNetworkLayerTest, Suspend) { |
| 53 MockClientSocketFactory mock_socket_factory; |
46 MockHostResolver host_resolver; | 54 MockHostResolver host_resolver; |
47 net::CertVerifier cert_verifier; | 55 CertVerifier cert_verifier; |
48 net::HttpNetworkLayer factory( | 56 scoped_refptr<HttpNetworkSession> network_session( |
49 NULL, | 57 new HttpNetworkSession( |
50 &host_resolver, | 58 &host_resolver, |
51 &cert_verifier, | 59 &cert_verifier, |
52 NULL /* dnsrr_resolver */, | 60 NULL /* dnsrr_resolver */, |
53 NULL /* dns_cert_checker */, | 61 NULL /* dns_cert_checker */, |
54 NULL /* ssl_host_info_factory */, | 62 NULL /* ssl_host_info_factory */, |
55 net::ProxyService::CreateDirect(), | 63 ProxyService::CreateDirect(), |
56 new net::SSLConfigServiceDefaults, | 64 &mock_socket_factory, |
57 NULL, | 65 new SSLConfigServiceDefaults, |
58 NULL, | 66 new SpdySessionPool(NULL), |
59 NULL); | 67 NULL, |
| 68 NULL, |
| 69 NULL)); |
| 70 HttpNetworkLayer factory(network_session); |
60 | 71 |
61 scoped_ptr<net::HttpTransaction> trans; | 72 scoped_ptr<HttpTransaction> trans; |
62 int rv = factory.CreateTransaction(&trans); | 73 int rv = factory.CreateTransaction(&trans); |
63 EXPECT_EQ(net::OK, rv); | 74 EXPECT_EQ(OK, rv); |
64 | 75 |
65 trans.reset(); | 76 trans.reset(); |
66 | 77 |
67 factory.Suspend(true); | 78 factory.Suspend(true); |
68 | 79 |
69 rv = factory.CreateTransaction(&trans); | 80 rv = factory.CreateTransaction(&trans); |
70 EXPECT_EQ(net::ERR_NETWORK_IO_SUSPENDED, rv); | 81 EXPECT_EQ(ERR_NETWORK_IO_SUSPENDED, rv); |
71 | 82 |
72 ASSERT_TRUE(trans == NULL); | 83 ASSERT_TRUE(trans == NULL); |
73 | 84 |
74 factory.Suspend(false); | 85 factory.Suspend(false); |
75 | 86 |
76 rv = factory.CreateTransaction(&trans); | 87 rv = factory.CreateTransaction(&trans); |
77 EXPECT_EQ(net::OK, rv); | 88 EXPECT_EQ(OK, rv); |
78 } | 89 } |
79 | 90 |
80 TEST_F(HttpNetworkLayerTest, GET) { | 91 TEST_F(HttpNetworkLayerTest, GET) { |
81 net::MockClientSocketFactory mock_socket_factory; | 92 MockClientSocketFactory mock_socket_factory; |
82 net::MockRead data_reads[] = { | 93 MockRead data_reads[] = { |
83 net::MockRead("HTTP/1.0 200 OK\r\n\r\n"), | 94 MockRead("HTTP/1.0 200 OK\r\n\r\n"), |
84 net::MockRead("hello world"), | 95 MockRead("hello world"), |
85 net::MockRead(false, net::OK), | 96 MockRead(false, OK), |
86 }; | 97 }; |
87 net::MockWrite data_writes[] = { | 98 MockWrite data_writes[] = { |
88 net::MockWrite("GET / HTTP/1.1\r\n" | 99 MockWrite("GET / HTTP/1.1\r\n" |
89 "Host: www.google.com\r\n" | 100 "Host: www.google.com\r\n" |
90 "Connection: keep-alive\r\n" | 101 "Connection: keep-alive\r\n" |
91 "User-Agent: Foo/1.0\r\n\r\n"), | 102 "User-Agent: Foo/1.0\r\n\r\n"), |
92 }; | 103 }; |
93 net::StaticSocketDataProvider data(data_reads, arraysize(data_reads), | 104 StaticSocketDataProvider data(data_reads, arraysize(data_reads), |
94 data_writes, arraysize(data_writes)); | 105 data_writes, arraysize(data_writes)); |
95 mock_socket_factory.AddSocketDataProvider(&data); | 106 mock_socket_factory.AddSocketDataProvider(&data); |
96 | 107 |
97 MockHostResolver host_resolver; | 108 MockHostResolver host_resolver; |
98 net::CertVerifier cert_verifier; | 109 CertVerifier cert_verifier; |
99 net::HttpNetworkLayer factory( | 110 scoped_refptr<HttpNetworkSession> network_session( |
100 &mock_socket_factory, | 111 new HttpNetworkSession( |
101 &host_resolver, | 112 &host_resolver, |
102 &cert_verifier, | 113 &cert_verifier, |
103 NULL /* dnsrr_resolver */, | 114 NULL /* dnsrr_resolver */, |
104 NULL /* dns_cert_checker */, | 115 NULL /* dns_cert_checker */, |
105 NULL /* ssl_host_info_factory */, | 116 NULL /* ssl_host_info_factory */, |
106 net::ProxyService::CreateDirect(), | 117 ProxyService::CreateDirect(), |
107 new net::SSLConfigServiceDefaults, | 118 &mock_socket_factory, |
108 NULL, | 119 new SSLConfigServiceDefaults, |
109 NULL, | 120 new SpdySessionPool(NULL), |
110 NULL); | 121 NULL, |
| 122 NULL, |
| 123 NULL)); |
| 124 HttpNetworkLayer factory(network_session); |
111 | 125 |
112 TestCompletionCallback callback; | 126 TestCompletionCallback callback; |
113 | 127 |
114 scoped_ptr<net::HttpTransaction> trans; | 128 scoped_ptr<HttpTransaction> trans; |
115 int rv = factory.CreateTransaction(&trans); | 129 int rv = factory.CreateTransaction(&trans); |
116 EXPECT_EQ(net::OK, rv); | 130 EXPECT_EQ(OK, rv); |
117 | 131 |
118 net::HttpRequestInfo request_info; | 132 HttpRequestInfo request_info; |
119 request_info.url = GURL("http://www.google.com/"); | 133 request_info.url = GURL("http://www.google.com/"); |
120 request_info.method = "GET"; | 134 request_info.method = "GET"; |
121 request_info.extra_headers.SetHeader(net::HttpRequestHeaders::kUserAgent, | 135 request_info.extra_headers.SetHeader(HttpRequestHeaders::kUserAgent, |
122 "Foo/1.0"); | 136 "Foo/1.0"); |
123 request_info.load_flags = net::LOAD_NORMAL; | 137 request_info.load_flags = LOAD_NORMAL; |
124 | 138 |
125 rv = trans->Start(&request_info, &callback, net::BoundNetLog()); | 139 rv = trans->Start(&request_info, &callback, BoundNetLog()); |
126 if (rv == net::ERR_IO_PENDING) | 140 if (rv == ERR_IO_PENDING) |
127 rv = callback.WaitForResult(); | 141 rv = callback.WaitForResult(); |
128 ASSERT_EQ(net::OK, rv); | 142 ASSERT_EQ(OK, rv); |
129 | 143 |
130 std::string contents; | 144 std::string contents; |
131 rv = ReadTransaction(trans.get(), &contents); | 145 rv = ReadTransaction(trans.get(), &contents); |
132 EXPECT_EQ(net::OK, rv); | 146 EXPECT_EQ(OK, rv); |
133 EXPECT_EQ("hello world", contents); | 147 EXPECT_EQ("hello world", contents); |
134 } | 148 } |
135 | 149 |
136 } // namespace | 150 } // namespace |
137 | 151 |
138 } // namespace net | 152 } // namespace net |
OLD | NEW |