| 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 "net/url_request/url_request_job.h" | |
| 6 | |
| 7 #include "base/memory/scoped_ptr.h" | |
| 8 #include "base/run_loop.h" | |
| 9 #include "net/base/request_priority.h" | |
| 10 #include "net/http/http_transaction_test_util.h" | |
| 11 #include "net/url_request/url_request.h" | |
| 12 #include "net/url_request/url_request_test_util.h" | |
| 13 #include "testing/gtest/include/gtest/gtest.h" | |
| 14 | |
| 15 namespace net { | |
| 16 | |
| 17 namespace { | |
| 18 | |
| 19 // This is a header that signals the end of the data. | |
| 20 const char kGzipData[] = "\x1f\x08b\x08\0\0\0\0\0\0\3\3\0\0\0\0\0\0\0\0"; | |
| 21 const char kGzipDataWithName[] = | |
| 22 "\x1f\x08b\x08\x08\0\0\0\0\0\0name\0\3\0\0\0\0\0\0\0\0"; | |
| 23 | |
| 24 void GZipServer(const HttpRequestInfo* request, | |
| 25 std::string* response_status, | |
| 26 std::string* response_headers, | |
| 27 std::string* response_data) { | |
| 28 response_data->assign(kGzipData, sizeof(kGzipData)); | |
| 29 } | |
| 30 | |
| 31 void BigGZipServer(const HttpRequestInfo* request, | |
| 32 std::string* response_status, | |
| 33 std::string* response_headers, | |
| 34 std::string* response_data) { | |
| 35 response_data->assign(kGzipDataWithName, sizeof(kGzipDataWithName)); | |
| 36 response_data->insert(10, 64 * 1024, 'a'); | |
| 37 } | |
| 38 | |
| 39 const MockTransaction kGZip_Transaction = { | |
| 40 "http://www.google.com/gzyp", | |
| 41 "GET", | |
| 42 base::Time(), | |
| 43 "", | |
| 44 LOAD_NORMAL, | |
| 45 "HTTP/1.1 200 OK", | |
| 46 "Cache-Control: max-age=10000\n" | |
| 47 "Content-Encoding: gzip\n" | |
| 48 "Content-Length: 30\n", // Intentionally wrong. | |
| 49 base::Time(), | |
| 50 "", | |
| 51 TEST_MODE_NORMAL, | |
| 52 &GZipServer, | |
| 53 0, | |
| 54 OK | |
| 55 }; | |
| 56 | |
| 57 const MockTransaction kRedirect_Transaction = { | |
| 58 "http://www.google.com/redirect", | |
| 59 "GET", | |
| 60 base::Time(), | |
| 61 "", | |
| 62 LOAD_NORMAL, | |
| 63 "HTTP/1.1 302 Found", | |
| 64 "Cache-Control: max-age=10000\n" | |
| 65 "Location: http://www.google.com/destination\n" | |
| 66 "Content-Length: 5\n", | |
| 67 base::Time(), | |
| 68 "hello", | |
| 69 TEST_MODE_NORMAL, | |
| 70 NULL, | |
| 71 0, | |
| 72 OK | |
| 73 }; | |
| 74 | |
| 75 } // namespace | |
| 76 | |
| 77 TEST(URLRequestJob, TransactionNotifiedWhenDone) { | |
| 78 MockNetworkLayer network_layer; | |
| 79 TestURLRequestContext context; | |
| 80 context.set_http_transaction_factory(&network_layer); | |
| 81 | |
| 82 TestDelegate d; | |
| 83 scoped_ptr<URLRequest> req(context.CreateRequest( | |
| 84 GURL(kGZip_Transaction.url), DEFAULT_PRIORITY, &d, NULL)); | |
| 85 AddMockTransaction(&kGZip_Transaction); | |
| 86 | |
| 87 req->set_method("GET"); | |
| 88 req->Start(); | |
| 89 | |
| 90 base::MessageLoop::current()->Run(); | |
| 91 | |
| 92 EXPECT_TRUE(network_layer.done_reading_called()); | |
| 93 | |
| 94 RemoveMockTransaction(&kGZip_Transaction); | |
| 95 } | |
| 96 | |
| 97 TEST(URLRequestJob, SyncTransactionNotifiedWhenDone) { | |
| 98 MockNetworkLayer network_layer; | |
| 99 TestURLRequestContext context; | |
| 100 context.set_http_transaction_factory(&network_layer); | |
| 101 | |
| 102 TestDelegate d; | |
| 103 scoped_ptr<URLRequest> req(context.CreateRequest( | |
| 104 GURL(kGZip_Transaction.url), DEFAULT_PRIORITY, &d, NULL)); | |
| 105 MockTransaction transaction(kGZip_Transaction); | |
| 106 transaction.test_mode = TEST_MODE_SYNC_ALL; | |
| 107 AddMockTransaction(&transaction); | |
| 108 | |
| 109 req->set_method("GET"); | |
| 110 req->Start(); | |
| 111 | |
| 112 base::RunLoop().Run(); | |
| 113 | |
| 114 EXPECT_TRUE(network_layer.done_reading_called()); | |
| 115 | |
| 116 RemoveMockTransaction(&transaction); | |
| 117 } | |
| 118 | |
| 119 // Tests processing a large gzip header one byte at a time. | |
| 120 TEST(URLRequestJob, SyncSlowTransaction) { | |
| 121 MockNetworkLayer network_layer; | |
| 122 TestURLRequestContext context; | |
| 123 context.set_http_transaction_factory(&network_layer); | |
| 124 | |
| 125 TestDelegate d; | |
| 126 scoped_ptr<URLRequest> req(context.CreateRequest( | |
| 127 GURL(kGZip_Transaction.url), DEFAULT_PRIORITY, &d, NULL)); | |
| 128 MockTransaction transaction(kGZip_Transaction); | |
| 129 transaction.test_mode = TEST_MODE_SYNC_ALL | TEST_MODE_SLOW_READ; | |
| 130 transaction.handler = &BigGZipServer; | |
| 131 AddMockTransaction(&transaction); | |
| 132 | |
| 133 req->set_method("GET"); | |
| 134 req->Start(); | |
| 135 | |
| 136 base::RunLoop().Run(); | |
| 137 | |
| 138 EXPECT_TRUE(network_layer.done_reading_called()); | |
| 139 | |
| 140 RemoveMockTransaction(&transaction); | |
| 141 } | |
| 142 | |
| 143 TEST(URLRequestJob, RedirectTransactionNotifiedWhenDone) { | |
| 144 MockNetworkLayer network_layer; | |
| 145 TestURLRequestContext context; | |
| 146 context.set_http_transaction_factory(&network_layer); | |
| 147 | |
| 148 TestDelegate d; | |
| 149 scoped_ptr<URLRequest> req(context.CreateRequest( | |
| 150 GURL(kRedirect_Transaction.url), DEFAULT_PRIORITY, &d, NULL)); | |
| 151 AddMockTransaction(&kRedirect_Transaction); | |
| 152 | |
| 153 req->set_method("GET"); | |
| 154 req->Start(); | |
| 155 | |
| 156 base::RunLoop().Run(); | |
| 157 | |
| 158 EXPECT_TRUE(network_layer.done_reading_called()); | |
| 159 | |
| 160 RemoveMockTransaction(&kRedirect_Transaction); | |
| 161 } | |
| 162 | |
| 163 TEST(URLRequestJob, TransactionNotCachedWhenNetworkDelegateRedirects) { | |
| 164 MockNetworkLayer network_layer; | |
| 165 TestNetworkDelegate network_delegate; | |
| 166 network_delegate.set_redirect_on_headers_received_url(GURL("http://foo")); | |
| 167 TestURLRequestContext context; | |
| 168 context.set_http_transaction_factory(&network_layer); | |
| 169 context.set_network_delegate(&network_delegate); | |
| 170 | |
| 171 TestDelegate d; | |
| 172 scoped_ptr<URLRequest> req(context.CreateRequest( | |
| 173 GURL(kGZip_Transaction.url), DEFAULT_PRIORITY, &d, NULL)); | |
| 174 AddMockTransaction(&kGZip_Transaction); | |
| 175 | |
| 176 req->set_method("GET"); | |
| 177 req->Start(); | |
| 178 | |
| 179 base::RunLoop().Run(); | |
| 180 | |
| 181 EXPECT_TRUE(network_layer.stop_caching_called()); | |
| 182 | |
| 183 RemoveMockTransaction(&kGZip_Transaction); | |
| 184 } | |
| 185 | |
| 186 } // namespace net | |
| OLD | NEW |