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 "webkit/glue/resource_request_body.h" | 5 #include "webkit/glue/resource_request_body.h" |
6 | 6 |
7 #include <algorithm> | 7 #include <algorithm> |
8 | 8 |
9 #include "base/file_path.h" | 9 #include "base/file_path.h" |
10 #include "base/file_util.h" | 10 #include "base/file_util.h" |
| 11 #include "base/message_loop_proxy.h" |
11 #include "base/time.h" | 12 #include "base/time.h" |
12 #include "googleurl/src/gurl.h" | 13 #include "googleurl/src/gurl.h" |
13 #include "net/base/upload_bytes_element_reader.h" | 14 #include "net/base/upload_bytes_element_reader.h" |
14 #include "net/base/upload_data_stream.h" | 15 #include "net/base/upload_data_stream.h" |
15 #include "net/base/upload_file_element_reader.h" | 16 #include "net/base/upload_file_element_reader.h" |
16 #include "testing/gtest/include/gtest/gtest.h" | 17 #include "testing/gtest/include/gtest/gtest.h" |
17 #include "webkit/blob/blob_storage_controller.h" | 18 #include "webkit/blob/blob_storage_controller.h" |
18 | 19 |
19 using webkit_blob::BlobData; | 20 using webkit_blob::BlobData; |
20 using webkit_blob::BlobStorageController; | 21 using webkit_blob::BlobStorageController; |
(...skipping 40 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
61 const uint64 kFileLength = 100U; | 62 const uint64 kFileLength = 100U; |
62 const base::Time kFileTime = base::Time::FromDoubleT(999); | 63 const base::Time kFileTime = base::Time::FromDoubleT(999); |
63 const int64 kIdentifier = 12345; | 64 const int64 kIdentifier = 12345; |
64 | 65 |
65 request_body->AppendBytes(kData, arraysize(kData) - 1); | 66 request_body->AppendBytes(kData, arraysize(kData) - 1); |
66 request_body->AppendFileRange(FilePath(kFilePath), | 67 request_body->AppendFileRange(FilePath(kFilePath), |
67 kFileOffset, kFileLength, kFileTime); | 68 kFileOffset, kFileLength, kFileTime); |
68 request_body->set_identifier(kIdentifier); | 69 request_body->set_identifier(kIdentifier); |
69 | 70 |
70 scoped_ptr<net::UploadDataStream> upload( | 71 scoped_ptr<net::UploadDataStream> upload( |
71 request_body->ResolveElementsAndCreateUploadDataStream(NULL)); | 72 request_body->ResolveElementsAndCreateUploadDataStream( |
| 73 NULL, base::MessageLoopProxy::current())); |
72 | 74 |
73 EXPECT_EQ(kIdentifier, upload->identifier()); | 75 EXPECT_EQ(kIdentifier, upload->identifier()); |
74 ASSERT_EQ(request_body->elements()->size(), upload->element_readers().size()); | 76 ASSERT_EQ(request_body->elements()->size(), upload->element_readers().size()); |
75 | 77 |
76 const net::UploadBytesElementReader* r1 = | 78 const net::UploadBytesElementReader* r1 = |
77 upload->element_readers()[0]->AsBytesReader(); | 79 upload->element_readers()[0]->AsBytesReader(); |
78 ASSERT_TRUE(r1); | 80 ASSERT_TRUE(r1); |
79 EXPECT_EQ(kData, std::string(r1->bytes(), r1->length())); | 81 EXPECT_EQ(kData, std::string(r1->bytes(), r1->length())); |
80 | 82 |
81 const net::UploadFileElementReader* r2 = | 83 const net::UploadFileElementReader* r2 = |
(...skipping 50 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
132 // Test no blob reference. | 134 // Test no blob reference. |
133 scoped_refptr<ResourceRequestBody> request_body(new ResourceRequestBody()); | 135 scoped_refptr<ResourceRequestBody> request_body(new ResourceRequestBody()); |
134 request_body->AppendBytes(upload_element1.bytes(), upload_element1.length()); | 136 request_body->AppendBytes(upload_element1.bytes(), upload_element1.length()); |
135 request_body->AppendFileRange(upload_element2.path(), | 137 request_body->AppendFileRange(upload_element2.path(), |
136 upload_element2.offset(), | 138 upload_element2.offset(), |
137 upload_element2.length(), | 139 upload_element2.length(), |
138 upload_element2.expected_modification_time()); | 140 upload_element2.expected_modification_time()); |
139 | 141 |
140 scoped_ptr<net::UploadDataStream> upload( | 142 scoped_ptr<net::UploadDataStream> upload( |
141 request_body->ResolveElementsAndCreateUploadDataStream( | 143 request_body->ResolveElementsAndCreateUploadDataStream( |
142 &blob_storage_controller)); | 144 &blob_storage_controller, base::MessageLoopProxy::current())); |
143 | 145 |
144 ASSERT_EQ(2U, upload->element_readers().size()); | 146 ASSERT_EQ(2U, upload->element_readers().size()); |
145 EXPECT_TRUE(AreElementsEqual(*upload->element_readers()[0], upload_element1)); | 147 EXPECT_TRUE(AreElementsEqual(*upload->element_readers()[0], upload_element1)); |
146 EXPECT_TRUE(AreElementsEqual(*upload->element_readers()[1], upload_element2)); | 148 EXPECT_TRUE(AreElementsEqual(*upload->element_readers()[1], upload_element2)); |
147 | 149 |
148 // Test having only one blob reference that refers to empty blob data. | 150 // Test having only one blob reference that refers to empty blob data. |
149 request_body = new ResourceRequestBody(); | 151 request_body = new ResourceRequestBody(); |
150 request_body->AppendBlob(blob_url0); | 152 request_body->AppendBlob(blob_url0); |
151 | 153 |
152 upload.reset(request_body->ResolveElementsAndCreateUploadDataStream( | 154 upload.reset(request_body->ResolveElementsAndCreateUploadDataStream( |
153 &blob_storage_controller)); | 155 &blob_storage_controller, base::MessageLoopProxy::current())); |
154 ASSERT_EQ(0U, upload->element_readers().size()); | 156 ASSERT_EQ(0U, upload->element_readers().size()); |
155 | 157 |
156 // Test having only one blob reference. | 158 // Test having only one blob reference. |
157 request_body = new ResourceRequestBody(); | 159 request_body = new ResourceRequestBody(); |
158 request_body->AppendBlob(blob_url1); | 160 request_body->AppendBlob(blob_url1); |
159 | 161 |
160 upload.reset(request_body->ResolveElementsAndCreateUploadDataStream( | 162 upload.reset(request_body->ResolveElementsAndCreateUploadDataStream( |
161 &blob_storage_controller)); | 163 &blob_storage_controller, base::MessageLoopProxy::current())); |
162 ASSERT_EQ(2U, upload->element_readers().size()); | 164 ASSERT_EQ(2U, upload->element_readers().size()); |
163 EXPECT_TRUE(AreElementsEqual(*upload->element_readers()[0], blob_element1)); | 165 EXPECT_TRUE(AreElementsEqual(*upload->element_readers()[0], blob_element1)); |
164 EXPECT_TRUE(AreElementsEqual(*upload->element_readers()[1], blob_element2)); | 166 EXPECT_TRUE(AreElementsEqual(*upload->element_readers()[1], blob_element2)); |
165 | 167 |
166 // Test having one blob reference at the beginning. | 168 // Test having one blob reference at the beginning. |
167 request_body = new ResourceRequestBody(); | 169 request_body = new ResourceRequestBody(); |
168 request_body->AppendBlob(blob_url1); | 170 request_body->AppendBlob(blob_url1); |
169 request_body->AppendBytes(upload_element1.bytes(), upload_element1.length()); | 171 request_body->AppendBytes(upload_element1.bytes(), upload_element1.length()); |
170 request_body->AppendFileRange(upload_element2.path(), | 172 request_body->AppendFileRange(upload_element2.path(), |
171 upload_element2.offset(), | 173 upload_element2.offset(), |
172 upload_element2.length(), | 174 upload_element2.length(), |
173 upload_element2.expected_modification_time()); | 175 upload_element2.expected_modification_time()); |
174 | 176 |
175 upload.reset(request_body->ResolveElementsAndCreateUploadDataStream( | 177 upload.reset(request_body->ResolveElementsAndCreateUploadDataStream( |
176 &blob_storage_controller)); | 178 &blob_storage_controller, base::MessageLoopProxy::current())); |
177 ASSERT_EQ(4U, upload->element_readers().size()); | 179 ASSERT_EQ(4U, upload->element_readers().size()); |
178 EXPECT_TRUE(AreElementsEqual(*upload->element_readers()[0], blob_element1)); | 180 EXPECT_TRUE(AreElementsEqual(*upload->element_readers()[0], blob_element1)); |
179 EXPECT_TRUE(AreElementsEqual(*upload->element_readers()[1], blob_element2)); | 181 EXPECT_TRUE(AreElementsEqual(*upload->element_readers()[1], blob_element2)); |
180 EXPECT_TRUE(AreElementsEqual(*upload->element_readers()[2], upload_element1)); | 182 EXPECT_TRUE(AreElementsEqual(*upload->element_readers()[2], upload_element1)); |
181 EXPECT_TRUE(AreElementsEqual(*upload->element_readers()[3], upload_element2)); | 183 EXPECT_TRUE(AreElementsEqual(*upload->element_readers()[3], upload_element2)); |
182 | 184 |
183 // Test having one blob reference at the end. | 185 // Test having one blob reference at the end. |
184 request_body = new ResourceRequestBody(); | 186 request_body = new ResourceRequestBody(); |
185 request_body->AppendBytes(upload_element1.bytes(), upload_element1.length()); | 187 request_body->AppendBytes(upload_element1.bytes(), upload_element1.length()); |
186 request_body->AppendFileRange(upload_element2.path(), | 188 request_body->AppendFileRange(upload_element2.path(), |
187 upload_element2.offset(), | 189 upload_element2.offset(), |
188 upload_element2.length(), | 190 upload_element2.length(), |
189 upload_element2.expected_modification_time()); | 191 upload_element2.expected_modification_time()); |
190 request_body->AppendBlob(blob_url1); | 192 request_body->AppendBlob(blob_url1); |
191 | 193 |
192 upload.reset(request_body->ResolveElementsAndCreateUploadDataStream( | 194 upload.reset(request_body->ResolveElementsAndCreateUploadDataStream( |
193 &blob_storage_controller)); | 195 &blob_storage_controller, base::MessageLoopProxy::current())); |
194 ASSERT_EQ(4U, upload->element_readers().size()); | 196 ASSERT_EQ(4U, upload->element_readers().size()); |
195 EXPECT_TRUE(AreElementsEqual(*upload->element_readers()[0], upload_element1)); | 197 EXPECT_TRUE(AreElementsEqual(*upload->element_readers()[0], upload_element1)); |
196 EXPECT_TRUE(AreElementsEqual(*upload->element_readers()[1], upload_element2)); | 198 EXPECT_TRUE(AreElementsEqual(*upload->element_readers()[1], upload_element2)); |
197 EXPECT_TRUE(AreElementsEqual(*upload->element_readers()[2], blob_element1)); | 199 EXPECT_TRUE(AreElementsEqual(*upload->element_readers()[2], blob_element1)); |
198 EXPECT_TRUE(AreElementsEqual(*upload->element_readers()[3], blob_element2)); | 200 EXPECT_TRUE(AreElementsEqual(*upload->element_readers()[3], blob_element2)); |
199 | 201 |
200 // Test having one blob reference in the middle. | 202 // Test having one blob reference in the middle. |
201 request_body = new ResourceRequestBody(); | 203 request_body = new ResourceRequestBody(); |
202 request_body->AppendBytes(upload_element1.bytes(), upload_element1.length()); | 204 request_body->AppendBytes(upload_element1.bytes(), upload_element1.length()); |
203 request_body->AppendBlob(blob_url1); | 205 request_body->AppendBlob(blob_url1); |
204 request_body->AppendFileRange(upload_element2.path(), | 206 request_body->AppendFileRange(upload_element2.path(), |
205 upload_element2.offset(), | 207 upload_element2.offset(), |
206 upload_element2.length(), | 208 upload_element2.length(), |
207 upload_element2.expected_modification_time()); | 209 upload_element2.expected_modification_time()); |
208 | 210 |
209 upload.reset(request_body->ResolveElementsAndCreateUploadDataStream( | 211 upload.reset(request_body->ResolveElementsAndCreateUploadDataStream( |
210 &blob_storage_controller)); | 212 &blob_storage_controller, base::MessageLoopProxy::current())); |
211 ASSERT_EQ(4U, upload->element_readers().size()); | 213 ASSERT_EQ(4U, upload->element_readers().size()); |
212 EXPECT_TRUE(AreElementsEqual(*upload->element_readers()[0], upload_element1)); | 214 EXPECT_TRUE(AreElementsEqual(*upload->element_readers()[0], upload_element1)); |
213 EXPECT_TRUE(AreElementsEqual(*upload->element_readers()[1], blob_element1)); | 215 EXPECT_TRUE(AreElementsEqual(*upload->element_readers()[1], blob_element1)); |
214 EXPECT_TRUE(AreElementsEqual(*upload->element_readers()[2], blob_element2)); | 216 EXPECT_TRUE(AreElementsEqual(*upload->element_readers()[2], blob_element2)); |
215 EXPECT_TRUE(AreElementsEqual(*upload->element_readers()[3], upload_element2)); | 217 EXPECT_TRUE(AreElementsEqual(*upload->element_readers()[3], upload_element2)); |
216 | 218 |
217 // Test having multiple blob references. | 219 // Test having multiple blob references. |
218 request_body = new ResourceRequestBody(); | 220 request_body = new ResourceRequestBody(); |
219 request_body->AppendBlob(blob_url1); | 221 request_body->AppendBlob(blob_url1); |
220 request_body->AppendBytes(upload_element1.bytes(), upload_element1.length()); | 222 request_body->AppendBytes(upload_element1.bytes(), upload_element1.length()); |
221 request_body->AppendBlob(blob_url2); | 223 request_body->AppendBlob(blob_url2); |
222 request_body->AppendBlob(blob_url3); | 224 request_body->AppendBlob(blob_url3); |
223 request_body->AppendFileRange(upload_element2.path(), | 225 request_body->AppendFileRange(upload_element2.path(), |
224 upload_element2.offset(), | 226 upload_element2.offset(), |
225 upload_element2.length(), | 227 upload_element2.length(), |
226 upload_element2.expected_modification_time()); | 228 upload_element2.expected_modification_time()); |
227 | 229 |
228 upload.reset(request_body->ResolveElementsAndCreateUploadDataStream( | 230 upload.reset(request_body->ResolveElementsAndCreateUploadDataStream( |
229 &blob_storage_controller)); | 231 &blob_storage_controller, base::MessageLoopProxy::current())); |
230 ASSERT_EQ(8U, upload->element_readers().size()); | 232 ASSERT_EQ(8U, upload->element_readers().size()); |
231 EXPECT_TRUE(AreElementsEqual(*upload->element_readers()[0], blob_element1)); | 233 EXPECT_TRUE(AreElementsEqual(*upload->element_readers()[0], blob_element1)); |
232 EXPECT_TRUE(AreElementsEqual(*upload->element_readers()[1], blob_element2)); | 234 EXPECT_TRUE(AreElementsEqual(*upload->element_readers()[1], blob_element2)); |
233 EXPECT_TRUE(AreElementsEqual(*upload->element_readers()[2], upload_element1)); | 235 EXPECT_TRUE(AreElementsEqual(*upload->element_readers()[2], upload_element1)); |
234 EXPECT_TRUE(AreElementsEqual(*upload->element_readers()[3], blob_element1)); | 236 EXPECT_TRUE(AreElementsEqual(*upload->element_readers()[3], blob_element1)); |
235 EXPECT_TRUE(AreElementsEqual(*upload->element_readers()[4], blob_element2)); | 237 EXPECT_TRUE(AreElementsEqual(*upload->element_readers()[4], blob_element2)); |
236 EXPECT_TRUE(AreElementsEqual(*upload->element_readers()[5], blob_element1)); | 238 EXPECT_TRUE(AreElementsEqual(*upload->element_readers()[5], blob_element1)); |
237 EXPECT_TRUE(AreElementsEqual(*upload->element_readers()[6], blob_element2)); | 239 EXPECT_TRUE(AreElementsEqual(*upload->element_readers()[6], blob_element2)); |
238 EXPECT_TRUE(AreElementsEqual(*upload->element_readers()[7], upload_element2)); | 240 EXPECT_TRUE(AreElementsEqual(*upload->element_readers()[7], upload_element2)); |
239 } | 241 } |
240 | 242 |
241 } // namespace webkit_glue | 243 } // namespace webkit_glue |
OLD | NEW |