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.h" | 5 #include "base/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 31 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
42 scoped_refptr<Stream> stream = registry_->GetStream(request->url()); | 42 scoped_refptr<Stream> stream = registry_->GetStream(request->url()); |
43 if (stream) | 43 if (stream) |
44 return new StreamURLRequestJob(request, network_delegate, stream); | 44 return new StreamURLRequestJob(request, network_delegate, stream); |
45 return NULL; | 45 return NULL; |
46 } | 46 } |
47 | 47 |
48 private: | 48 private: |
49 StreamRegistry* registry_; | 49 StreamRegistry* registry_; |
50 }; | 50 }; |
51 | 51 |
52 StreamURLRequestJobTest() | 52 StreamURLRequestJobTest() : message_loop_(base::MessageLoop::TYPE_IO) {} |
53 : message_loop_(MessageLoop::TYPE_IO) { | |
54 } | |
55 | 53 |
56 virtual void SetUp() { | 54 virtual void SetUp() { |
57 registry_.reset(new StreamRegistry()); | 55 registry_.reset(new StreamRegistry()); |
58 | 56 |
59 url_request_job_factory_.SetProtocolHandler( | 57 url_request_job_factory_.SetProtocolHandler( |
60 "blob", new MockProtocolHandler(registry_.get())); | 58 "blob", new MockProtocolHandler(registry_.get())); |
61 url_request_context_.set_job_factory(&url_request_job_factory_); | 59 url_request_context_.set_job_factory(&url_request_job_factory_); |
62 } | 60 } |
63 | 61 |
64 virtual void TearDown() { | 62 virtual void TearDown() { |
65 } | 63 } |
66 | 64 |
67 void TestSuccessRequest(const GURL& url, | 65 void TestSuccessRequest(const GURL& url, |
68 const std::string& expected_response) { | 66 const std::string& expected_response) { |
69 TestRequest("GET", url, net::HttpRequestHeaders(), 200, expected_response); | 67 TestRequest("GET", url, net::HttpRequestHeaders(), 200, expected_response); |
70 } | 68 } |
71 | 69 |
72 void TestRequest(const std::string& method, | 70 void TestRequest(const std::string& method, |
73 const GURL& url, | 71 const GURL& url, |
74 const net::HttpRequestHeaders& extra_headers, | 72 const net::HttpRequestHeaders& extra_headers, |
75 int expected_status_code, | 73 int expected_status_code, |
76 const std::string& expected_response) { | 74 const std::string& expected_response) { |
77 net::TestDelegate delegate; | 75 net::TestDelegate delegate; |
78 request_.reset(url_request_context_.CreateRequest(url, &delegate)); | 76 request_.reset(url_request_context_.CreateRequest(url, &delegate)); |
79 request_->set_method(method); | 77 request_->set_method(method); |
80 if (!extra_headers.IsEmpty()) | 78 if (!extra_headers.IsEmpty()) |
81 request_->SetExtraRequestHeaders(extra_headers); | 79 request_->SetExtraRequestHeaders(extra_headers); |
82 request_->Start(); | 80 request_->Start(); |
83 | 81 |
84 MessageLoop::current()->RunUntilIdle(); | 82 base::MessageLoop::current()->RunUntilIdle(); |
85 | 83 |
86 // Verify response. | 84 // Verify response. |
87 EXPECT_TRUE(request_->status().is_success()); | 85 EXPECT_TRUE(request_->status().is_success()); |
88 EXPECT_EQ(expected_status_code, | 86 EXPECT_EQ(expected_status_code, |
89 request_->response_headers()->response_code()); | 87 request_->response_headers()->response_code()); |
90 EXPECT_EQ(expected_response, delegate.data_received()); | 88 EXPECT_EQ(expected_response, delegate.data_received()); |
91 } | 89 } |
92 | 90 |
93 protected: | 91 protected: |
94 MessageLoop message_loop_; | 92 base::MessageLoop message_loop_; |
95 scoped_ptr<StreamRegistry> registry_; | 93 scoped_ptr<StreamRegistry> registry_; |
96 | 94 |
97 net::URLRequestContext url_request_context_; | 95 net::URLRequestContext url_request_context_; |
98 net::URLRequestJobFactoryImpl url_request_job_factory_; | 96 net::URLRequestJobFactoryImpl url_request_job_factory_; |
99 scoped_ptr<net::URLRequest> request_; | 97 scoped_ptr<net::URLRequest> request_; |
100 }; | 98 }; |
101 | 99 |
102 TEST_F(StreamURLRequestJobTest, TestGetSimpleDataRequest) { | 100 TEST_F(StreamURLRequestJobTest, TestGetSimpleDataRequest) { |
103 scoped_refptr<Stream> stream( | 101 scoped_refptr<Stream> stream( |
104 new Stream(registry_.get(), NULL, GURL(), kStreamURL)); | 102 new Stream(registry_.get(), NULL, GURL(), kStreamURL)); |
(...skipping 23 matching lines...) Expand all Loading... |
128 stream->Finalize(); | 126 stream->Finalize(); |
129 TestSuccessRequest(kStreamURL, large_data); | 127 TestSuccessRequest(kStreamURL, large_data); |
130 } | 128 } |
131 | 129 |
132 TEST_F(StreamURLRequestJobTest, TestGetNonExistentStreamRequest) { | 130 TEST_F(StreamURLRequestJobTest, TestGetNonExistentStreamRequest) { |
133 net::TestDelegate delegate; | 131 net::TestDelegate delegate; |
134 request_.reset(url_request_context_.CreateRequest(kStreamURL, &delegate)); | 132 request_.reset(url_request_context_.CreateRequest(kStreamURL, &delegate)); |
135 request_->set_method("GET"); | 133 request_->set_method("GET"); |
136 request_->Start(); | 134 request_->Start(); |
137 | 135 |
138 MessageLoop::current()->RunUntilIdle(); | 136 base::MessageLoop::current()->RunUntilIdle(); |
139 | 137 |
140 // Verify response. | 138 // Verify response. |
141 EXPECT_FALSE(request_->status().is_success()); | 139 EXPECT_FALSE(request_->status().is_success()); |
142 } | 140 } |
143 | 141 |
144 TEST_F(StreamURLRequestJobTest, TestRangeDataRequest) { | 142 TEST_F(StreamURLRequestJobTest, TestRangeDataRequest) { |
145 scoped_refptr<Stream> stream( | 143 scoped_refptr<Stream> stream( |
146 new Stream(registry_.get(), NULL, GURL(), kStreamURL)); | 144 new Stream(registry_.get(), NULL, GURL(), kStreamURL)); |
147 | 145 |
148 scoped_refptr<net::StringIOBuffer> buffer( | 146 scoped_refptr<net::StringIOBuffer> buffer( |
(...skipping 17 matching lines...) Expand all Loading... |
166 | 164 |
167 stream->AddData(buffer, buffer->size()); | 165 stream->AddData(buffer, buffer->size()); |
168 stream->Finalize(); | 166 stream->Finalize(); |
169 | 167 |
170 net::HttpRequestHeaders extra_headers; | 168 net::HttpRequestHeaders extra_headers; |
171 extra_headers.SetHeader(net::HttpRequestHeaders::kRange, "bytes=1-3"); | 169 extra_headers.SetHeader(net::HttpRequestHeaders::kRange, "bytes=1-3"); |
172 TestRequest("GET", kStreamURL, extra_headers, 405, std::string()); | 170 TestRequest("GET", kStreamURL, extra_headers, 405, std::string()); |
173 } | 171 } |
174 | 172 |
175 } // namespace content | 173 } // namespace content |
OLD | NEW |