| OLD | NEW |
| 1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2011 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/files/file_path.h" | 5 #include "base/files/file_path.h" |
| 6 #include "base/memory/ref_counted.h" | 6 #include "base/memory/ref_counted.h" |
| 7 #include "base/memory/scoped_ptr.h" | 7 #include "base/memory/scoped_ptr.h" |
| 8 #include "base/message_loop.h" | 8 #include "base/message_loop.h" |
| 9 #include "base/time.h" | 9 #include "base/time.h" |
| 10 #include "testing/gtest/include/gtest/gtest.h" | 10 #include "testing/gtest/include/gtest/gtest.h" |
| (...skipping 10 matching lines...) Expand all Loading... |
| 21 item.SetToBytes("1", 1); | 21 item.SetToBytes("1", 1); |
| 22 EXPECT_TRUE(host->AppendBlobDataItem(id, item)); | 22 EXPECT_TRUE(host->AppendBlobDataItem(id, item)); |
| 23 EXPECT_TRUE(host->FinishBuildingBlob(id, "text/plain")); | 23 EXPECT_TRUE(host->FinishBuildingBlob(id, "text/plain")); |
| 24 EXPECT_FALSE(host->StartBuildingBlob(id)); | 24 EXPECT_FALSE(host->StartBuildingBlob(id)); |
| 25 } | 25 } |
| 26 } // namespace | 26 } // namespace |
| 27 | 27 |
| 28 TEST(BlobStorageContextTest, IncrementDecrementRef) { | 28 TEST(BlobStorageContextTest, IncrementDecrementRef) { |
| 29 BlobStorageContext context; | 29 BlobStorageContext context; |
| 30 BlobStorageHost host(&context); | 30 BlobStorageHost host(&context); |
| 31 MessageLoop fake_io_message_loop; | 31 base::MessageLoop fake_io_message_loop; |
| 32 | 32 |
| 33 // Build up a basic blob. | 33 // Build up a basic blob. |
| 34 const std::string kId("id"); | 34 const std::string kId("id"); |
| 35 SetupBasicBlob(&host, kId); | 35 SetupBasicBlob(&host, kId); |
| 36 | 36 |
| 37 // Make sure it's there, finish building implies a ref of one. | 37 // Make sure it's there, finish building implies a ref of one. |
| 38 scoped_ptr<BlobDataHandle> blob_data_handle; | 38 scoped_ptr<BlobDataHandle> blob_data_handle; |
| 39 blob_data_handle = context.GetBlobDataFromUUID(kId); | 39 blob_data_handle = context.GetBlobDataFromUUID(kId); |
| 40 EXPECT_TRUE(blob_data_handle); | 40 EXPECT_TRUE(blob_data_handle); |
| 41 blob_data_handle.reset(); | 41 blob_data_handle.reset(); |
| 42 | 42 |
| 43 // Make sure its still there after inc/dec. | 43 // Make sure its still there after inc/dec. |
| 44 EXPECT_TRUE(host.IncrementBlobRefCount(kId)); | 44 EXPECT_TRUE(host.IncrementBlobRefCount(kId)); |
| 45 EXPECT_TRUE(host.DecrementBlobRefCount(kId)); | 45 EXPECT_TRUE(host.DecrementBlobRefCount(kId)); |
| 46 blob_data_handle = context.GetBlobDataFromUUID(kId); | 46 blob_data_handle = context.GetBlobDataFromUUID(kId); |
| 47 EXPECT_TRUE(blob_data_handle); | 47 EXPECT_TRUE(blob_data_handle); |
| 48 blob_data_handle.reset(); | 48 blob_data_handle.reset(); |
| 49 | 49 |
| 50 // Make sure it goes away in the end. | 50 // Make sure it goes away in the end. |
| 51 EXPECT_TRUE(host.DecrementBlobRefCount(kId)); | 51 EXPECT_TRUE(host.DecrementBlobRefCount(kId)); |
| 52 blob_data_handle = context.GetBlobDataFromUUID(kId); | 52 blob_data_handle = context.GetBlobDataFromUUID(kId); |
| 53 EXPECT_FALSE(blob_data_handle); | 53 EXPECT_FALSE(blob_data_handle); |
| 54 EXPECT_FALSE(host.DecrementBlobRefCount(kId)); | 54 EXPECT_FALSE(host.DecrementBlobRefCount(kId)); |
| 55 EXPECT_FALSE(host.IncrementBlobRefCount(kId)); | 55 EXPECT_FALSE(host.IncrementBlobRefCount(kId)); |
| 56 } | 56 } |
| 57 | 57 |
| 58 TEST(BlobStorageContextTest, BlobDataHandle) { | 58 TEST(BlobStorageContextTest, BlobDataHandle) { |
| 59 BlobStorageContext context; | 59 BlobStorageContext context; |
| 60 BlobStorageHost host(&context); | 60 BlobStorageHost host(&context); |
| 61 MessageLoop fake_io_message_loop; | 61 base::MessageLoop fake_io_message_loop; |
| 62 | 62 |
| 63 // Build up a basic blob. | 63 // Build up a basic blob. |
| 64 const std::string kId("id"); | 64 const std::string kId("id"); |
| 65 SetupBasicBlob(&host, kId); | 65 SetupBasicBlob(&host, kId); |
| 66 | 66 |
| 67 // Get a handle to it. | 67 // Get a handle to it. |
| 68 scoped_ptr<BlobDataHandle> blob_data_handle = | 68 scoped_ptr<BlobDataHandle> blob_data_handle = |
| 69 context.GetBlobDataFromUUID(kId); | 69 context.GetBlobDataFromUUID(kId); |
| 70 EXPECT_TRUE(blob_data_handle); | 70 EXPECT_TRUE(blob_data_handle); |
| 71 | 71 |
| (...skipping 10 matching lines...) Expand all Loading... |
| 82 another_handle.reset(); | 82 another_handle.reset(); |
| 83 blob_data_handle = context.GetBlobDataFromUUID(kId); | 83 blob_data_handle = context.GetBlobDataFromUUID(kId); |
| 84 EXPECT_FALSE(blob_data_handle); | 84 EXPECT_FALSE(blob_data_handle); |
| 85 } | 85 } |
| 86 | 86 |
| 87 TEST(BlobStorageContextTest, CompoundBlobs) { | 87 TEST(BlobStorageContextTest, CompoundBlobs) { |
| 88 const std::string kId1("id1"); | 88 const std::string kId1("id1"); |
| 89 const std::string kId2("id2"); | 89 const std::string kId2("id2"); |
| 90 const std::string kId2Prime("id2.prime"); | 90 const std::string kId2Prime("id2.prime"); |
| 91 | 91 |
| 92 MessageLoop fake_io_message_loop; | 92 base::MessageLoop fake_io_message_loop; |
| 93 | 93 |
| 94 // Setup a set of blob data for testing. | 94 // Setup a set of blob data for testing. |
| 95 base::Time time1, time2; | 95 base::Time time1, time2; |
| 96 base::Time::FromString("Tue, 15 Nov 1994, 12:45:26 GMT", &time1); | 96 base::Time::FromString("Tue, 15 Nov 1994, 12:45:26 GMT", &time1); |
| 97 base::Time::FromString("Mon, 14 Nov 1994, 11:30:49 GMT", &time2); | 97 base::Time::FromString("Mon, 14 Nov 1994, 11:30:49 GMT", &time2); |
| 98 | 98 |
| 99 scoped_refptr<BlobData> blob_data1(new BlobData(kId1)); | 99 scoped_refptr<BlobData> blob_data1(new BlobData(kId1)); |
| 100 blob_data1->AppendData("Data1"); | 100 blob_data1->AppendData("Data1"); |
| 101 blob_data1->AppendData("Data2"); | 101 blob_data1->AppendData("Data2"); |
| 102 blob_data1->AppendFile(base::FilePath(FILE_PATH_LITERAL("File1.txt")), | 102 blob_data1->AppendFile(base::FilePath(FILE_PATH_LITERAL("File1.txt")), |
| (...skipping 24 matching lines...) Expand all Loading... |
| 127 | 127 |
| 128 // Test a blob composed in part with another blob. | 128 // Test a blob composed in part with another blob. |
| 129 blob_data_handle = context.AddFinishedBlob(blob_data2); | 129 blob_data_handle = context.AddFinishedBlob(blob_data2); |
| 130 ASSERT_TRUE(blob_data_handle.get()); | 130 ASSERT_TRUE(blob_data_handle.get()); |
| 131 EXPECT_TRUE(*(blob_data_handle->data()) == *canonicalized_blob_data2); | 131 EXPECT_TRUE(*(blob_data_handle->data()) == *canonicalized_blob_data2); |
| 132 } | 132 } |
| 133 | 133 |
| 134 TEST(BlobStorageContextTest, PublicBlobUrls) { | 134 TEST(BlobStorageContextTest, PublicBlobUrls) { |
| 135 BlobStorageContext context; | 135 BlobStorageContext context; |
| 136 BlobStorageHost host(&context); | 136 BlobStorageHost host(&context); |
| 137 MessageLoop fake_io_message_loop; | 137 base::MessageLoop fake_io_message_loop; |
| 138 | 138 |
| 139 // Build up a basic blob. | 139 // Build up a basic blob. |
| 140 const std::string kId("id"); | 140 const std::string kId("id"); |
| 141 SetupBasicBlob(&host, kId); | 141 SetupBasicBlob(&host, kId); |
| 142 | 142 |
| 143 // Now register a url for that blob. | 143 // Now register a url for that blob. |
| 144 GURL kUrl("blob:id"); | 144 GURL kUrl("blob:id"); |
| 145 EXPECT_TRUE(host.RegisterPublicBlobURL(kUrl, kId)); | 145 EXPECT_TRUE(host.RegisterPublicBlobURL(kUrl, kId)); |
| 146 scoped_ptr<BlobDataHandle> blob_data_handle = | 146 scoped_ptr<BlobDataHandle> blob_data_handle = |
| 147 context.GetBlobDataFromPublicURL(kUrl); | 147 context.GetBlobDataFromPublicURL(kUrl); |
| (...skipping 11 matching lines...) Expand all Loading... |
| 159 // Finally get rid of the url registration and the blob. | 159 // Finally get rid of the url registration and the blob. |
| 160 EXPECT_TRUE(host.RevokePublicBlobURL(kUrl)); | 160 EXPECT_TRUE(host.RevokePublicBlobURL(kUrl)); |
| 161 blob_data_handle = context.GetBlobDataFromPublicURL(kUrl); | 161 blob_data_handle = context.GetBlobDataFromPublicURL(kUrl); |
| 162 EXPECT_TRUE(!blob_data_handle.get()); | 162 EXPECT_TRUE(!blob_data_handle.get()); |
| 163 EXPECT_FALSE(host.RevokePublicBlobURL(kUrl)); | 163 EXPECT_FALSE(host.RevokePublicBlobURL(kUrl)); |
| 164 } | 164 } |
| 165 | 165 |
| 166 TEST(BlobStorageContextTest, HostCleanup) { | 166 TEST(BlobStorageContextTest, HostCleanup) { |
| 167 BlobStorageContext context; | 167 BlobStorageContext context; |
| 168 scoped_ptr<BlobStorageHost> host(new BlobStorageHost(&context)); | 168 scoped_ptr<BlobStorageHost> host(new BlobStorageHost(&context)); |
| 169 MessageLoop fake_io_message_loop; | 169 base::MessageLoop fake_io_message_loop; |
| 170 | 170 |
| 171 // Build up a basic blob and register a url | 171 // Build up a basic blob and register a url |
| 172 const std::string kId("id"); | 172 const std::string kId("id"); |
| 173 GURL kUrl("blob:id"); | 173 GURL kUrl("blob:id"); |
| 174 SetupBasicBlob(host.get(), kId); | 174 SetupBasicBlob(host.get(), kId); |
| 175 EXPECT_TRUE(host->RegisterPublicBlobURL(kUrl, kId)); | 175 EXPECT_TRUE(host->RegisterPublicBlobURL(kUrl, kId)); |
| 176 | 176 |
| 177 // All should disappear upon host deletion. | 177 // All should disappear upon host deletion. |
| 178 host.reset(); | 178 host.reset(); |
| 179 scoped_ptr<BlobDataHandle> handle = context.GetBlobDataFromPublicURL(kUrl); | 179 scoped_ptr<BlobDataHandle> handle = context.GetBlobDataFromPublicURL(kUrl); |
| 180 EXPECT_TRUE(!handle.get()); | 180 EXPECT_TRUE(!handle.get()); |
| 181 handle = context.GetBlobDataFromUUID(kId); | 181 handle = context.GetBlobDataFromUUID(kId); |
| 182 EXPECT_TRUE(!handle.get()); | 182 EXPECT_TRUE(!handle.get()); |
| 183 } | 183 } |
| 184 | 184 |
| 185 TEST(BlobStorageContextTest, EarlyContextDeletion) { | 185 TEST(BlobStorageContextTest, EarlyContextDeletion) { |
| 186 scoped_ptr<BlobStorageContext> context(new BlobStorageContext); | 186 scoped_ptr<BlobStorageContext> context(new BlobStorageContext); |
| 187 BlobStorageHost host(context.get()); | 187 BlobStorageHost host(context.get()); |
| 188 MessageLoop fake_io_message_loop; | 188 base::MessageLoop fake_io_message_loop; |
| 189 | 189 |
| 190 // Deleting the context should not induce crashes. | 190 // Deleting the context should not induce crashes. |
| 191 context.reset(); | 191 context.reset(); |
| 192 | 192 |
| 193 const std::string kId("id"); | 193 const std::string kId("id"); |
| 194 GURL kUrl("blob:id"); | 194 GURL kUrl("blob:id"); |
| 195 EXPECT_FALSE(host.StartBuildingBlob(kId)); | 195 EXPECT_FALSE(host.StartBuildingBlob(kId)); |
| 196 BlobData::Item item; | 196 BlobData::Item item; |
| 197 item.SetToBytes("1", 1); | 197 item.SetToBytes("1", 1); |
| 198 EXPECT_FALSE(host.AppendBlobDataItem(kId, item)); | 198 EXPECT_FALSE(host.AppendBlobDataItem(kId, item)); |
| 199 EXPECT_FALSE(host.FinishBuildingBlob(kId, "text/plain")); | 199 EXPECT_FALSE(host.FinishBuildingBlob(kId, "text/plain")); |
| 200 EXPECT_FALSE(host.RegisterPublicBlobURL(kUrl, kId)); | 200 EXPECT_FALSE(host.RegisterPublicBlobURL(kUrl, kId)); |
| 201 EXPECT_FALSE(host.IncrementBlobRefCount(kId)); | 201 EXPECT_FALSE(host.IncrementBlobRefCount(kId)); |
| 202 EXPECT_FALSE(host.DecrementBlobRefCount(kId)); | 202 EXPECT_FALSE(host.DecrementBlobRefCount(kId)); |
| 203 EXPECT_FALSE(host.RevokePublicBlobURL(kUrl)); | 203 EXPECT_FALSE(host.RevokePublicBlobURL(kUrl)); |
| 204 } | 204 } |
| 205 | 205 |
| 206 } // namespace webkit_blob | 206 } // namespace webkit_blob |
| OLD | NEW |