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/message_loop/message_loop.h" | 5 #include "base/message_loop/message_loop.h" |
6 #include "base/test/test_simple_task_runner.h" | 6 #include "base/test/test_simple_task_runner.h" |
7 #include "content/browser/streams/stream.h" | 7 #include "content/browser/streams/stream.h" |
8 #include "content/browser/streams/stream_registry.h" | 8 #include "content/browser/streams/stream_registry.h" |
9 #include "content/browser/streams/stream_url_request_job.h" | 9 #include "content/browser/streams/stream_url_request_job.h" |
10 #include "content/browser/streams/stream_write_observer.h" | 10 #include "content/browser/streams/stream_write_observer.h" |
| 11 #include "net/base/request_priority.h" |
11 #include "net/http/http_response_headers.h" | 12 #include "net/http/http_response_headers.h" |
12 #include "net/url_request/url_request.h" | 13 #include "net/url_request/url_request.h" |
13 #include "net/url_request/url_request_context.h" | 14 #include "net/url_request/url_request_context.h" |
14 #include "net/url_request/url_request_job_factory_impl.h" | 15 #include "net/url_request/url_request_job_factory_impl.h" |
15 #include "net/url_request/url_request_test_util.h" | 16 #include "net/url_request/url_request_test_util.h" |
16 #include "testing/gtest/include/gtest/gtest.h" | 17 #include "testing/gtest/include/gtest/gtest.h" |
17 | 18 |
18 namespace content { | 19 namespace content { |
19 | 20 |
20 namespace { | 21 namespace { |
(...skipping 45 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
66 const std::string& expected_response) { | 67 const std::string& expected_response) { |
67 TestRequest("GET", url, net::HttpRequestHeaders(), 200, expected_response); | 68 TestRequest("GET", url, net::HttpRequestHeaders(), 200, expected_response); |
68 } | 69 } |
69 | 70 |
70 void TestRequest(const std::string& method, | 71 void TestRequest(const std::string& method, |
71 const GURL& url, | 72 const GURL& url, |
72 const net::HttpRequestHeaders& extra_headers, | 73 const net::HttpRequestHeaders& extra_headers, |
73 int expected_status_code, | 74 int expected_status_code, |
74 const std::string& expected_response) { | 75 const std::string& expected_response) { |
75 net::TestDelegate delegate; | 76 net::TestDelegate delegate; |
76 request_.reset(url_request_context_.CreateRequest(url, &delegate)); | 77 request_ = url_request_context_.CreateRequest( |
| 78 url, net::DEFAULT_PRIORITY, &delegate); |
77 request_->set_method(method); | 79 request_->set_method(method); |
78 if (!extra_headers.IsEmpty()) | 80 if (!extra_headers.IsEmpty()) |
79 request_->SetExtraRequestHeaders(extra_headers); | 81 request_->SetExtraRequestHeaders(extra_headers); |
80 request_->Start(); | 82 request_->Start(); |
81 | 83 |
82 base::MessageLoop::current()->RunUntilIdle(); | 84 base::MessageLoop::current()->RunUntilIdle(); |
83 | 85 |
84 // Verify response. | 86 // Verify response. |
85 EXPECT_TRUE(request_->status().is_success()); | 87 EXPECT_TRUE(request_->status().is_success()); |
86 ASSERT_TRUE(request_->response_headers()); | 88 ASSERT_TRUE(request_->response_headers()); |
(...skipping 36 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
123 scoped_refptr<net::StringIOBuffer> buffer( | 125 scoped_refptr<net::StringIOBuffer> buffer( |
124 new net::StringIOBuffer(large_data)); | 126 new net::StringIOBuffer(large_data)); |
125 | 127 |
126 stream->AddData(buffer, buffer->size()); | 128 stream->AddData(buffer, buffer->size()); |
127 stream->Finalize(); | 129 stream->Finalize(); |
128 TestSuccessRequest(kStreamURL, large_data); | 130 TestSuccessRequest(kStreamURL, large_data); |
129 } | 131 } |
130 | 132 |
131 TEST_F(StreamURLRequestJobTest, TestGetNonExistentStreamRequest) { | 133 TEST_F(StreamURLRequestJobTest, TestGetNonExistentStreamRequest) { |
132 net::TestDelegate delegate; | 134 net::TestDelegate delegate; |
133 request_.reset(url_request_context_.CreateRequest(kStreamURL, &delegate)); | 135 request_ = url_request_context_.CreateRequest( |
| 136 kStreamURL, net::DEFAULT_PRIORITY, &delegate); |
134 request_->set_method("GET"); | 137 request_->set_method("GET"); |
135 request_->Start(); | 138 request_->Start(); |
136 | 139 |
137 base::MessageLoop::current()->RunUntilIdle(); | 140 base::MessageLoop::current()->RunUntilIdle(); |
138 | 141 |
139 // Verify response. | 142 // Verify response. |
140 EXPECT_FALSE(request_->status().is_success()); | 143 EXPECT_FALSE(request_->status().is_success()); |
141 } | 144 } |
142 | 145 |
143 TEST_F(StreamURLRequestJobTest, TestRangeDataRequest) { | 146 TEST_F(StreamURLRequestJobTest, TestRangeDataRequest) { |
(...skipping 21 matching lines...) Expand all Loading... |
165 | 168 |
166 stream->AddData(buffer, buffer->size()); | 169 stream->AddData(buffer, buffer->size()); |
167 stream->Finalize(); | 170 stream->Finalize(); |
168 | 171 |
169 net::HttpRequestHeaders extra_headers; | 172 net::HttpRequestHeaders extra_headers; |
170 extra_headers.SetHeader(net::HttpRequestHeaders::kRange, "bytes=1-3"); | 173 extra_headers.SetHeader(net::HttpRequestHeaders::kRange, "bytes=1-3"); |
171 TestRequest("GET", kStreamURL, extra_headers, 405, std::string()); | 174 TestRequest("GET", kStreamURL, extra_headers, 405, std::string()); |
172 } | 175 } |
173 | 176 |
174 } // namespace content | 177 } // namespace content |
OLD | NEW |