| 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" |
| (...skipping 81 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 92 base::MessageLoop message_loop_; | 92 base::MessageLoop message_loop_; |
| 93 scoped_ptr<StreamRegistry> registry_; | 93 scoped_ptr<StreamRegistry> registry_; |
| 94 | 94 |
| 95 net::URLRequestContext url_request_context_; | 95 net::URLRequestContext url_request_context_; |
| 96 net::URLRequestJobFactoryImpl url_request_job_factory_; | 96 net::URLRequestJobFactoryImpl url_request_job_factory_; |
| 97 scoped_ptr<net::URLRequest> request_; | 97 scoped_ptr<net::URLRequest> request_; |
| 98 }; | 98 }; |
| 99 | 99 |
| 100 TEST_F(StreamURLRequestJobTest, TestGetSimpleDataRequest) { | 100 TEST_F(StreamURLRequestJobTest, TestGetSimpleDataRequest) { |
| 101 scoped_refptr<Stream> stream( | 101 scoped_refptr<Stream> stream( |
| 102 new Stream(registry_.get(), NULL, GURL(), kStreamURL)); | 102 new Stream(registry_.get(), NULL, kStreamURL)); |
| 103 | 103 |
| 104 scoped_refptr<net::StringIOBuffer> buffer( | 104 scoped_refptr<net::StringIOBuffer> buffer( |
| 105 new net::StringIOBuffer(kTestData1)); | 105 new net::StringIOBuffer(kTestData1)); |
| 106 | 106 |
| 107 stream->AddData(buffer, buffer->size()); | 107 stream->AddData(buffer, buffer->size()); |
| 108 stream->Finalize(); | 108 stream->Finalize(); |
| 109 | 109 |
| 110 TestSuccessRequest(kStreamURL, kTestData1); | 110 TestSuccessRequest(kStreamURL, kTestData1); |
| 111 } | 111 } |
| 112 | 112 |
| 113 TEST_F(StreamURLRequestJobTest, TestGetLargeStreamRequest) { | 113 TEST_F(StreamURLRequestJobTest, TestGetLargeStreamRequest) { |
| 114 scoped_refptr<Stream> stream( | 114 scoped_refptr<Stream> stream( |
| 115 new Stream(registry_.get(), NULL, GURL(), kStreamURL)); | 115 new Stream(registry_.get(), NULL, kStreamURL)); |
| 116 | 116 |
| 117 std::string large_data; | 117 std::string large_data; |
| 118 large_data.reserve(kBufferSize * 5); | 118 large_data.reserve(kBufferSize * 5); |
| 119 for (int i = 0; i < kBufferSize * 5; ++i) | 119 for (int i = 0; i < kBufferSize * 5; ++i) |
| 120 large_data.append(1, static_cast<char>(i % 256)); | 120 large_data.append(1, static_cast<char>(i % 256)); |
| 121 | 121 |
| 122 scoped_refptr<net::StringIOBuffer> buffer( | 122 scoped_refptr<net::StringIOBuffer> buffer( |
| 123 new net::StringIOBuffer(large_data)); | 123 new net::StringIOBuffer(large_data)); |
| 124 | 124 |
| 125 stream->AddData(buffer, buffer->size()); | 125 stream->AddData(buffer, buffer->size()); |
| 126 stream->Finalize(); | 126 stream->Finalize(); |
| 127 TestSuccessRequest(kStreamURL, large_data); | 127 TestSuccessRequest(kStreamURL, large_data); |
| 128 } | 128 } |
| 129 | 129 |
| 130 TEST_F(StreamURLRequestJobTest, TestGetNonExistentStreamRequest) { | 130 TEST_F(StreamURLRequestJobTest, TestGetNonExistentStreamRequest) { |
| 131 net::TestDelegate delegate; | 131 net::TestDelegate delegate; |
| 132 request_.reset(url_request_context_.CreateRequest(kStreamURL, &delegate)); | 132 request_.reset(url_request_context_.CreateRequest(kStreamURL, &delegate)); |
| 133 request_->set_method("GET"); | 133 request_->set_method("GET"); |
| 134 request_->Start(); | 134 request_->Start(); |
| 135 | 135 |
| 136 base::MessageLoop::current()->RunUntilIdle(); | 136 base::MessageLoop::current()->RunUntilIdle(); |
| 137 | 137 |
| 138 // Verify response. | 138 // Verify response. |
| 139 EXPECT_FALSE(request_->status().is_success()); | 139 EXPECT_FALSE(request_->status().is_success()); |
| 140 } | 140 } |
| 141 | 141 |
| 142 TEST_F(StreamURLRequestJobTest, TestRangeDataRequest) { | 142 TEST_F(StreamURLRequestJobTest, TestRangeDataRequest) { |
| 143 scoped_refptr<Stream> stream( | 143 scoped_refptr<Stream> stream( |
| 144 new Stream(registry_.get(), NULL, GURL(), kStreamURL)); | 144 new Stream(registry_.get(), NULL, kStreamURL)); |
| 145 | 145 |
| 146 scoped_refptr<net::StringIOBuffer> buffer( | 146 scoped_refptr<net::StringIOBuffer> buffer( |
| 147 new net::StringIOBuffer(kTestData2)); | 147 new net::StringIOBuffer(kTestData2)); |
| 148 | 148 |
| 149 stream->AddData(buffer, buffer->size()); | 149 stream->AddData(buffer, buffer->size()); |
| 150 stream->Finalize(); | 150 stream->Finalize(); |
| 151 | 151 |
| 152 net::HttpRequestHeaders extra_headers; | 152 net::HttpRequestHeaders extra_headers; |
| 153 extra_headers.SetHeader(net::HttpRequestHeaders::kRange, "bytes=0-3"); | 153 extra_headers.SetHeader(net::HttpRequestHeaders::kRange, "bytes=0-3"); |
| 154 TestRequest("GET", kStreamURL, extra_headers, | 154 TestRequest("GET", kStreamURL, extra_headers, |
| 155 200, std::string(kTestData2, 4)); | 155 200, std::string(kTestData2, 4)); |
| 156 } | 156 } |
| 157 | 157 |
| 158 TEST_F(StreamURLRequestJobTest, TestInvalidRangeDataRequest) { | 158 TEST_F(StreamURLRequestJobTest, TestInvalidRangeDataRequest) { |
| 159 scoped_refptr<Stream> stream( | 159 scoped_refptr<Stream> stream( |
| 160 new Stream(registry_.get(), NULL, GURL(), kStreamURL)); | 160 new Stream(registry_.get(), NULL, kStreamURL)); |
| 161 | 161 |
| 162 scoped_refptr<net::StringIOBuffer> buffer( | 162 scoped_refptr<net::StringIOBuffer> buffer( |
| 163 new net::StringIOBuffer(kTestData2)); | 163 new net::StringIOBuffer(kTestData2)); |
| 164 | 164 |
| 165 stream->AddData(buffer, buffer->size()); | 165 stream->AddData(buffer, buffer->size()); |
| 166 stream->Finalize(); | 166 stream->Finalize(); |
| 167 | 167 |
| 168 net::HttpRequestHeaders extra_headers; | 168 net::HttpRequestHeaders extra_headers; |
| 169 extra_headers.SetHeader(net::HttpRequestHeaders::kRange, "bytes=1-3"); | 169 extra_headers.SetHeader(net::HttpRequestHeaders::kRange, "bytes=1-3"); |
| 170 TestRequest("GET", kStreamURL, extra_headers, 405, std::string()); | 170 TestRequest("GET", kStreamURL, extra_headers, 405, std::string()); |
| 171 } | 171 } |
| 172 | 172 |
| 173 } // namespace content | 173 } // namespace content |
| OLD | NEW |