| OLD | NEW |
| 1 // Copyright 2013 The Chromium Authors. All rights reserved. | 1 // Copyright 2013 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 "content/browser/loader/upload_data_stream_builder.h" | 5 #include "content/browser/loader/upload_data_stream_builder.h" |
| 6 | 6 |
| 7 #include <algorithm> | 7 #include <algorithm> |
| 8 | 8 |
| 9 #include "base/file_util.h" | 9 #include "base/file_util.h" |
| 10 #include "base/files/file_path.h" | 10 #include "base/files/file_path.h" |
| 11 #include "base/message_loop/message_loop.h" | 11 #include "base/message_loop/message_loop.h" |
| 12 #include "base/message_loop/message_loop_proxy.h" | 12 #include "base/message_loop/message_loop_proxy.h" |
| 13 #include "base/run_loop.h" |
| 13 #include "base/time/time.h" | 14 #include "base/time/time.h" |
| 14 #include "content/common/resource_request_body.h" | 15 #include "content/common/resource_request_body.h" |
| 15 #include "net/base/upload_bytes_element_reader.h" | 16 #include "net/base/upload_bytes_element_reader.h" |
| 16 #include "net/base/upload_data_stream.h" | 17 #include "net/base/upload_data_stream.h" |
| 17 #include "net/base/upload_file_element_reader.h" | 18 #include "net/base/upload_file_element_reader.h" |
| 18 #include "testing/gtest/include/gtest/gtest.h" | 19 #include "testing/gtest/include/gtest/gtest.h" |
| 19 #include "url/gurl.h" | 20 #include "url/gurl.h" |
| 20 #include "webkit/browser/blob/blob_storage_context.h" | 21 #include "webkit/browser/blob/blob_storage_context.h" |
| 21 | 22 |
| 22 using webkit_blob::BlobData; | 23 using webkit_blob::BlobData; |
| (...skipping 63 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 86 upload->element_readers()[1]->AsFileReader(); | 87 upload->element_readers()[1]->AsFileReader(); |
| 87 ASSERT_TRUE(r2); | 88 ASSERT_TRUE(r2); |
| 88 EXPECT_EQ(kFilePath, r2->path().value()); | 89 EXPECT_EQ(kFilePath, r2->path().value()); |
| 89 EXPECT_EQ(kFileOffset, r2->range_offset()); | 90 EXPECT_EQ(kFileOffset, r2->range_offset()); |
| 90 EXPECT_EQ(kFileLength, r2->range_length()); | 91 EXPECT_EQ(kFileLength, r2->range_length()); |
| 91 EXPECT_EQ(kFileTime, r2->expected_modification_time()); | 92 EXPECT_EQ(kFileTime, r2->expected_modification_time()); |
| 92 } | 93 } |
| 93 | 94 |
| 94 TEST(UploadDataStreamBuilderTest, ResolveBlobAndCreateUploadDataStream) { | 95 TEST(UploadDataStreamBuilderTest, ResolveBlobAndCreateUploadDataStream) { |
| 95 base::MessageLoop message_loop; | 96 base::MessageLoop message_loop; |
| 96 // Setup blob data for testing. | 97 { |
| 97 base::Time time1, time2; | 98 // Setup blob data for testing. |
| 98 base::Time::FromString("Tue, 15 Nov 1994, 12:45:26 GMT", &time1); | 99 base::Time time1, time2; |
| 99 base::Time::FromString("Mon, 14 Nov 1994, 11:30:49 GMT", &time2); | 100 base::Time::FromString("Tue, 15 Nov 1994, 12:45:26 GMT", &time1); |
| 100 | 101 base::Time::FromString("Mon, 14 Nov 1994, 11:30:49 GMT", &time2); |
| 101 BlobStorageContext blob_storage_context; | 102 |
| 102 | 103 BlobStorageContext blob_storage_context; |
| 103 const std::string blob_id0("id-0"); | 104 |
| 104 scoped_refptr<BlobData> blob_data(new BlobData(blob_id0)); | 105 const std::string blob_id0("id-0"); |
| 105 scoped_ptr<BlobDataHandle> handle1 = | 106 scoped_refptr<BlobData> blob_data(new BlobData(blob_id0)); |
| 106 blob_storage_context.AddFinishedBlob(blob_data); | 107 scoped_ptr<BlobDataHandle> handle1 = |
| 107 | 108 blob_storage_context.AddFinishedBlob(blob_data); |
| 108 const std::string blob_id1("id-1"); | 109 |
| 109 blob_data = new BlobData(blob_id1); | 110 const std::string blob_id1("id-1"); |
| 110 blob_data->AppendData("BlobData"); | 111 blob_data = new BlobData(blob_id1); |
| 111 blob_data->AppendFile( | 112 blob_data->AppendData("BlobData"); |
| 112 base::FilePath(FILE_PATH_LITERAL("BlobFile.txt")), 0, 20, time1); | 113 blob_data->AppendFile( |
| 113 scoped_ptr<BlobDataHandle> handle2 = | 114 base::FilePath(FILE_PATH_LITERAL("BlobFile.txt")), 0, 20, time1); |
| 114 blob_storage_context.AddFinishedBlob(blob_data); | 115 scoped_ptr<BlobDataHandle> handle2 = |
| 115 | 116 blob_storage_context.AddFinishedBlob(blob_data); |
| 116 // Setup upload data elements for comparison. | 117 |
| 117 ResourceRequestBody::Element blob_element1, blob_element2; | 118 // Setup upload data elements for comparison. |
| 118 blob_element1.SetToBytes( | 119 ResourceRequestBody::Element blob_element1, blob_element2; |
| 119 blob_data->items().at(0).bytes() + | 120 blob_element1.SetToBytes( |
| 120 static_cast<int>(blob_data->items().at(0).offset()), | 121 blob_data->items().at(0).bytes() + |
| 121 static_cast<int>(blob_data->items().at(0).length())); | 122 static_cast<int>(blob_data->items().at(0).offset()), |
| 122 blob_element2.SetToFilePathRange( | 123 static_cast<int>(blob_data->items().at(0).length())); |
| 123 blob_data->items().at(1).path(), | 124 blob_element2.SetToFilePathRange( |
| 124 blob_data->items().at(1).offset(), | 125 blob_data->items().at(1).path(), |
| 125 blob_data->items().at(1).length(), | 126 blob_data->items().at(1).offset(), |
| 126 blob_data->items().at(1).expected_modification_time()); | 127 blob_data->items().at(1).length(), |
| 127 | 128 blob_data->items().at(1).expected_modification_time()); |
| 128 ResourceRequestBody::Element upload_element1, upload_element2; | 129 |
| 129 upload_element1.SetToBytes("Hello", 5); | 130 ResourceRequestBody::Element upload_element1, upload_element2; |
| 130 upload_element2.SetToFilePathRange( | 131 upload_element1.SetToBytes("Hello", 5); |
| 131 base::FilePath(FILE_PATH_LITERAL("foo1.txt")), 0, 20, time2); | 132 upload_element2.SetToFilePathRange( |
| 132 | 133 base::FilePath(FILE_PATH_LITERAL("foo1.txt")), 0, 20, time2); |
| 133 // Test no blob reference. | 134 |
| 134 scoped_refptr<ResourceRequestBody> request_body(new ResourceRequestBody()); | 135 // Test no blob reference. |
| 135 request_body->AppendBytes(upload_element1.bytes(), upload_element1.length()); | 136 scoped_refptr<ResourceRequestBody> request_body(new ResourceRequestBody()); |
| 136 request_body->AppendFileRange(upload_element2.path(), | 137 request_body->AppendBytes( |
| 137 upload_element2.offset(), | 138 upload_element1.bytes(), |
| 138 upload_element2.length(), | 139 upload_element1.length()); |
| 139 upload_element2.expected_modification_time()); | 140 request_body->AppendFileRange( |
| 140 | 141 upload_element2.path(), |
| 141 scoped_ptr<net::UploadDataStream> upload( | 142 upload_element2.offset(), |
| 142 UploadDataStreamBuilder::Build(request_body.get(), | 143 upload_element2.length(), |
| 143 &blob_storage_context, | 144 upload_element2.expected_modification_time()); |
| 144 NULL, | 145 |
| 145 base::MessageLoopProxy::current().get())); | 146 scoped_ptr<net::UploadDataStream> upload( |
| 146 | 147 UploadDataStreamBuilder::Build( |
| 147 ASSERT_EQ(2U, upload->element_readers().size()); | 148 request_body.get(), |
| 148 EXPECT_TRUE(AreElementsEqual(*upload->element_readers()[0], upload_element1)); | 149 &blob_storage_context, |
| 149 EXPECT_TRUE(AreElementsEqual(*upload->element_readers()[1], upload_element2)); | 150 NULL, |
| 150 | 151 base::MessageLoopProxy::current().get())); |
| 151 // Test having only one blob reference that refers to empty blob data. | 152 |
| 152 request_body = new ResourceRequestBody(); | 153 ASSERT_EQ(2U, upload->element_readers().size()); |
| 153 request_body->AppendBlob(blob_id0); | 154 EXPECT_TRUE(AreElementsEqual( |
| 154 | 155 *upload->element_readers()[0], upload_element1)); |
| 155 upload = | 156 EXPECT_TRUE(AreElementsEqual( |
| 156 UploadDataStreamBuilder::Build(request_body.get(), | 157 *upload->element_readers()[1], upload_element2)); |
| 157 &blob_storage_context, | 158 |
| 158 NULL, | 159 // Test having only one blob reference that refers to empty blob data. |
| 159 base::MessageLoopProxy::current().get()); | 160 request_body = new ResourceRequestBody(); |
| 160 ASSERT_EQ(0U, upload->element_readers().size()); | 161 request_body->AppendBlob(blob_id0); |
| 161 | 162 |
| 162 // Test having only one blob reference. | 163 upload = UploadDataStreamBuilder::Build( |
| 163 request_body = new ResourceRequestBody(); | 164 request_body.get(), |
| 164 request_body->AppendBlob(blob_id1); | 165 &blob_storage_context, |
| 165 | 166 NULL, |
| 166 upload = | 167 base::MessageLoopProxy::current().get()); |
| 167 UploadDataStreamBuilder::Build(request_body.get(), | 168 ASSERT_EQ(0U, upload->element_readers().size()); |
| 168 &blob_storage_context, | 169 |
| 169 NULL, | 170 // Test having only one blob reference. |
| 170 base::MessageLoopProxy::current().get()); | 171 request_body = new ResourceRequestBody(); |
| 171 ASSERT_EQ(2U, upload->element_readers().size()); | 172 request_body->AppendBlob(blob_id1); |
| 172 EXPECT_TRUE(AreElementsEqual(*upload->element_readers()[0], blob_element1)); | 173 |
| 173 EXPECT_TRUE(AreElementsEqual(*upload->element_readers()[1], blob_element2)); | 174 upload = UploadDataStreamBuilder::Build( |
| 174 | 175 request_body.get(), |
| 175 // Test having one blob reference at the beginning. | 176 &blob_storage_context, |
| 176 request_body = new ResourceRequestBody(); | 177 NULL, |
| 177 request_body->AppendBlob(blob_id1); | 178 base::MessageLoopProxy::current().get()); |
| 178 request_body->AppendBytes(upload_element1.bytes(), upload_element1.length()); | 179 ASSERT_EQ(2U, upload->element_readers().size()); |
| 179 request_body->AppendFileRange(upload_element2.path(), | 180 EXPECT_TRUE(AreElementsEqual( |
| 180 upload_element2.offset(), | 181 *upload->element_readers()[0], blob_element1)); |
| 181 upload_element2.length(), | 182 EXPECT_TRUE(AreElementsEqual( |
| 182 upload_element2.expected_modification_time()); | 183 *upload->element_readers()[1], blob_element2)); |
| 183 | 184 |
| 184 upload = | 185 // Test having one blob reference at the beginning. |
| 185 UploadDataStreamBuilder::Build(request_body.get(), | 186 request_body = new ResourceRequestBody(); |
| 186 &blob_storage_context, | 187 request_body->AppendBlob(blob_id1); |
| 187 NULL, | 188 request_body->AppendBytes( |
| 188 base::MessageLoopProxy::current().get()); | 189 upload_element1.bytes(), |
| 189 ASSERT_EQ(4U, upload->element_readers().size()); | 190 upload_element1.length()); |
| 190 EXPECT_TRUE(AreElementsEqual(*upload->element_readers()[0], blob_element1)); | 191 request_body->AppendFileRange( |
| 191 EXPECT_TRUE(AreElementsEqual(*upload->element_readers()[1], blob_element2)); | 192 upload_element2.path(), |
| 192 EXPECT_TRUE(AreElementsEqual(*upload->element_readers()[2], upload_element1)); | 193 upload_element2.offset(), |
| 193 EXPECT_TRUE(AreElementsEqual(*upload->element_readers()[3], upload_element2)); | 194 upload_element2.length(), |
| 194 | 195 upload_element2.expected_modification_time()); |
| 195 // Test having one blob reference at the end. | 196 |
| 196 request_body = new ResourceRequestBody(); | 197 upload = UploadDataStreamBuilder::Build( |
| 197 request_body->AppendBytes(upload_element1.bytes(), upload_element1.length()); | 198 request_body.get(), |
| 198 request_body->AppendFileRange(upload_element2.path(), | 199 &blob_storage_context, |
| 199 upload_element2.offset(), | 200 NULL, |
| 200 upload_element2.length(), | 201 base::MessageLoopProxy::current().get()); |
| 201 upload_element2.expected_modification_time()); | 202 ASSERT_EQ(4U, upload->element_readers().size()); |
| 202 request_body->AppendBlob(blob_id1); | 203 EXPECT_TRUE(AreElementsEqual( |
| 203 | 204 *upload->element_readers()[0], blob_element1)); |
| 204 upload = | 205 EXPECT_TRUE(AreElementsEqual( |
| 205 UploadDataStreamBuilder::Build(request_body.get(), | 206 *upload->element_readers()[1], blob_element2)); |
| 206 &blob_storage_context, | 207 EXPECT_TRUE(AreElementsEqual( |
| 207 NULL, | 208 *upload->element_readers()[2], upload_element1)); |
| 208 base::MessageLoopProxy::current().get()); | 209 EXPECT_TRUE(AreElementsEqual( |
| 209 ASSERT_EQ(4U, upload->element_readers().size()); | 210 *upload->element_readers()[3], upload_element2)); |
| 210 EXPECT_TRUE(AreElementsEqual(*upload->element_readers()[0], upload_element1)); | 211 |
| 211 EXPECT_TRUE(AreElementsEqual(*upload->element_readers()[1], upload_element2)); | 212 // Test having one blob reference at the end. |
| 212 EXPECT_TRUE(AreElementsEqual(*upload->element_readers()[2], blob_element1)); | 213 request_body = new ResourceRequestBody(); |
| 213 EXPECT_TRUE(AreElementsEqual(*upload->element_readers()[3], blob_element2)); | 214 request_body->AppendBytes( |
| 214 | 215 upload_element1.bytes(), |
| 215 // Test having one blob reference in the middle. | 216 upload_element1.length()); |
| 216 request_body = new ResourceRequestBody(); | 217 request_body->AppendFileRange( |
| 217 request_body->AppendBytes(upload_element1.bytes(), upload_element1.length()); | 218 upload_element2.path(), |
| 218 request_body->AppendBlob(blob_id1); | 219 upload_element2.offset(), |
| 219 request_body->AppendFileRange(upload_element2.path(), | 220 upload_element2.length(), |
| 220 upload_element2.offset(), | 221 upload_element2.expected_modification_time()); |
| 221 upload_element2.length(), | 222 request_body->AppendBlob(blob_id1); |
| 222 upload_element2.expected_modification_time()); | 223 |
| 223 | 224 upload = |
| 224 upload = | 225 UploadDataStreamBuilder::Build(request_body.get(), |
| 225 UploadDataStreamBuilder::Build(request_body.get(), | 226 &blob_storage_context, |
| 226 &blob_storage_context, | 227 NULL, |
| 227 NULL, | 228 base::MessageLoopProxy::current().get()); |
| 228 base::MessageLoopProxy::current().get()); | 229 ASSERT_EQ(4U, upload->element_readers().size()); |
| 229 ASSERT_EQ(4U, upload->element_readers().size()); | 230 EXPECT_TRUE(AreElementsEqual( |
| 230 EXPECT_TRUE(AreElementsEqual(*upload->element_readers()[0], upload_element1)); | 231 *upload->element_readers()[0], upload_element1)); |
| 231 EXPECT_TRUE(AreElementsEqual(*upload->element_readers()[1], blob_element1)); | 232 EXPECT_TRUE(AreElementsEqual( |
| 232 EXPECT_TRUE(AreElementsEqual(*upload->element_readers()[2], blob_element2)); | 233 *upload->element_readers()[1], upload_element2)); |
| 233 EXPECT_TRUE(AreElementsEqual(*upload->element_readers()[3], upload_element2)); | 234 EXPECT_TRUE(AreElementsEqual( |
| 234 | 235 *upload->element_readers()[2], blob_element1)); |
| 235 // Test having multiple blob references. | 236 EXPECT_TRUE(AreElementsEqual( |
| 236 request_body = new ResourceRequestBody(); | 237 *upload->element_readers()[3], blob_element2)); |
| 237 request_body->AppendBlob(blob_id1); | 238 |
| 238 request_body->AppendBytes(upload_element1.bytes(), upload_element1.length()); | 239 // Test having one blob reference in the middle. |
| 239 request_body->AppendBlob(blob_id1); | 240 request_body = new ResourceRequestBody(); |
| 240 request_body->AppendBlob(blob_id1); | 241 request_body->AppendBytes( |
| 241 request_body->AppendFileRange(upload_element2.path(), | 242 upload_element1.bytes(), |
| 242 upload_element2.offset(), | 243 upload_element1.length()); |
| 243 upload_element2.length(), | 244 request_body->AppendBlob(blob_id1); |
| 244 upload_element2.expected_modification_time()); | 245 request_body->AppendFileRange( |
| 245 | 246 upload_element2.path(), |
| 246 upload = | 247 upload_element2.offset(), |
| 247 UploadDataStreamBuilder::Build(request_body.get(), | 248 upload_element2.length(), |
| 248 &blob_storage_context, | 249 upload_element2.expected_modification_time()); |
| 249 NULL, | 250 |
| 250 base::MessageLoopProxy::current().get()); | 251 upload = UploadDataStreamBuilder::Build( |
| 251 ASSERT_EQ(8U, upload->element_readers().size()); | 252 request_body.get(), |
| 252 EXPECT_TRUE(AreElementsEqual(*upload->element_readers()[0], blob_element1)); | 253 &blob_storage_context, |
| 253 EXPECT_TRUE(AreElementsEqual(*upload->element_readers()[1], blob_element2)); | 254 NULL, |
| 254 EXPECT_TRUE(AreElementsEqual(*upload->element_readers()[2], upload_element1)); | 255 base::MessageLoopProxy::current().get()); |
| 255 EXPECT_TRUE(AreElementsEqual(*upload->element_readers()[3], blob_element1)); | 256 ASSERT_EQ(4U, upload->element_readers().size()); |
| 256 EXPECT_TRUE(AreElementsEqual(*upload->element_readers()[4], blob_element2)); | 257 EXPECT_TRUE(AreElementsEqual( |
| 257 EXPECT_TRUE(AreElementsEqual(*upload->element_readers()[5], blob_element1)); | 258 *upload->element_readers()[0], upload_element1)); |
| 258 EXPECT_TRUE(AreElementsEqual(*upload->element_readers()[6], blob_element2)); | 259 EXPECT_TRUE(AreElementsEqual( |
| 259 EXPECT_TRUE(AreElementsEqual(*upload->element_readers()[7], upload_element2)); | 260 *upload->element_readers()[1], blob_element1)); |
| 261 EXPECT_TRUE(AreElementsEqual( |
| 262 *upload->element_readers()[2], blob_element2)); |
| 263 EXPECT_TRUE(AreElementsEqual( |
| 264 *upload->element_readers()[3], upload_element2)); |
| 265 |
| 266 // Test having multiple blob references. |
| 267 request_body = new ResourceRequestBody(); |
| 268 request_body->AppendBlob(blob_id1); |
| 269 request_body->AppendBytes( |
| 270 upload_element1.bytes(), |
| 271 upload_element1.length()); |
| 272 request_body->AppendBlob(blob_id1); |
| 273 request_body->AppendBlob(blob_id1); |
| 274 request_body->AppendFileRange( |
| 275 upload_element2.path(), |
| 276 upload_element2.offset(), |
| 277 upload_element2.length(), |
| 278 upload_element2.expected_modification_time()); |
| 279 |
| 280 upload = UploadDataStreamBuilder::Build( |
| 281 request_body.get(), |
| 282 &blob_storage_context, |
| 283 NULL, |
| 284 base::MessageLoopProxy::current().get()); |
| 285 ASSERT_EQ(8U, upload->element_readers().size()); |
| 286 EXPECT_TRUE(AreElementsEqual( |
| 287 *upload->element_readers()[0], blob_element1)); |
| 288 EXPECT_TRUE(AreElementsEqual( |
| 289 *upload->element_readers()[1], blob_element2)); |
| 290 EXPECT_TRUE(AreElementsEqual( |
| 291 *upload->element_readers()[2], upload_element1)); |
| 292 EXPECT_TRUE(AreElementsEqual( |
| 293 *upload->element_readers()[3], blob_element1)); |
| 294 EXPECT_TRUE(AreElementsEqual( |
| 295 *upload->element_readers()[4], blob_element2)); |
| 296 EXPECT_TRUE(AreElementsEqual( |
| 297 *upload->element_readers()[5], blob_element1)); |
| 298 EXPECT_TRUE(AreElementsEqual( |
| 299 *upload->element_readers()[6], blob_element2)); |
| 300 EXPECT_TRUE(AreElementsEqual( |
| 301 *upload->element_readers()[7], upload_element2)); |
| 302 } |
| 303 // Clean up for ASAN. |
| 304 base::RunLoop().RunUntilIdle(); |
| 260 } | 305 } |
| 261 | 306 |
| 262 } // namespace content | 307 } // namespace content |
| OLD | NEW |