OLD | NEW |
1 // Copyright 2014 The Chromium Authors. All rights reserved. | 1 // Copyright 2014 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/message_loop.h" | 8 #include "base/message_loop/message_loop.h" |
9 #include "base/time/time.h" | 9 #include "base/time/time.h" |
10 #include "content/browser/fileapi/blob_storage_host.h" | 10 #include "content/browser/fileapi/blob_storage_host_impl.h" |
11 #include "testing/gtest/include/gtest/gtest.h" | 11 #include "testing/gtest/include/gtest/gtest.h" |
12 #include "webkit/browser/blob/blob_data_handle.h" | 12 #include "webkit/browser/blob/blob_data_handle.h" |
13 #include "webkit/browser/blob/blob_storage_context.h" | 13 #include "webkit/browser/blob/blob_storage_context.h" |
14 | 14 |
15 using webkit_blob::BlobDataHandle; | 15 using webkit_blob::BlobDataHandle; |
16 | 16 |
17 namespace content { | 17 namespace content { |
18 | 18 |
19 namespace { | 19 namespace { |
20 void SetupBasicBlob(BlobStorageHost* host, const std::string& id) { | 20 void SetupBasicBlob(BlobStorageHost* host, const std::string& id) { |
21 EXPECT_TRUE(host->StartBuildingBlob(id)); | 21 EXPECT_TRUE(host->StartBuildingBlob(id)); |
22 BlobData::Item item; | 22 BlobData::Item item; |
23 item.SetToBytes("1", 1); | 23 item.SetToBytes("1", 1); |
24 EXPECT_TRUE(host->AppendBlobDataItem(id, item)); | 24 EXPECT_TRUE(host->AppendBlobDataItem(id, item)); |
25 EXPECT_TRUE(host->FinishBuildingBlob(id, "text/plain")); | 25 EXPECT_TRUE(host->FinishBuildingBlob(id, "text/plain")); |
26 EXPECT_FALSE(host->StartBuildingBlob(id)); | 26 EXPECT_FALSE(host->StartBuildingBlob(id)); |
27 } | 27 } |
28 } // namespace | 28 } // namespace |
29 | 29 |
30 TEST(BlobStorageContextTest, IncrementDecrementRef) { | 30 TEST(BlobStorageContextTest, IncrementDecrementRef) { |
31 BlobStorageContext context; | 31 BlobStorageContext context; |
32 BlobStorageHost host(&context); | 32 BlobStorageHostImpl host(&context); |
33 base::MessageLoop fake_io_message_loop; | 33 base::MessageLoop fake_io_message_loop; |
34 | 34 |
35 // Build up a basic blob. | 35 // Build up a basic blob. |
36 const std::string kId("id"); | 36 const std::string kId("id"); |
37 SetupBasicBlob(&host, kId); | 37 SetupBasicBlob(&host, kId); |
38 | 38 |
39 // Make sure it's there, finish building implies a ref of one. | 39 // Make sure it's there, finish building implies a ref of one. |
40 scoped_ptr<BlobDataHandle> blob_data_handle; | 40 scoped_ptr<BlobDataHandle> blob_data_handle; |
41 blob_data_handle = context.GetBlobDataFromUUID(kId); | 41 blob_data_handle = context.GetBlobDataFromUUID(kId); |
42 EXPECT_TRUE(blob_data_handle); | 42 EXPECT_TRUE(blob_data_handle); |
43 blob_data_handle.reset(); | 43 blob_data_handle.reset(); |
44 | 44 |
45 // Make sure its still there after inc/dec. | 45 // Make sure its still there after inc/dec. |
46 EXPECT_TRUE(host.IncrementBlobRefCount(kId)); | 46 EXPECT_TRUE(host.IncrementBlobRefCount(kId)); |
47 EXPECT_TRUE(host.DecrementBlobRefCount(kId)); | 47 EXPECT_TRUE(host.DecrementBlobRefCount(kId)); |
48 blob_data_handle = context.GetBlobDataFromUUID(kId); | 48 blob_data_handle = context.GetBlobDataFromUUID(kId); |
49 EXPECT_TRUE(blob_data_handle); | 49 EXPECT_TRUE(blob_data_handle); |
50 blob_data_handle.reset(); | 50 blob_data_handle.reset(); |
51 | 51 |
52 // Make sure it goes away in the end. | 52 // Make sure it goes away in the end. |
53 EXPECT_TRUE(host.DecrementBlobRefCount(kId)); | 53 EXPECT_TRUE(host.DecrementBlobRefCount(kId)); |
54 blob_data_handle = context.GetBlobDataFromUUID(kId); | 54 blob_data_handle = context.GetBlobDataFromUUID(kId); |
55 EXPECT_FALSE(blob_data_handle); | 55 EXPECT_FALSE(blob_data_handle); |
56 EXPECT_FALSE(host.DecrementBlobRefCount(kId)); | 56 EXPECT_FALSE(host.DecrementBlobRefCount(kId)); |
57 EXPECT_FALSE(host.IncrementBlobRefCount(kId)); | 57 EXPECT_FALSE(host.IncrementBlobRefCount(kId)); |
58 } | 58 } |
59 | 59 |
60 TEST(BlobStorageContextTest, BlobDataHandle) { | 60 TEST(BlobStorageContextTest, BlobDataHandle) { |
61 BlobStorageContext context; | 61 BlobStorageContext context; |
62 BlobStorageHost host(&context); | 62 BlobStorageHostImpl host(&context); |
63 base::MessageLoop fake_io_message_loop; | 63 base::MessageLoop fake_io_message_loop; |
64 | 64 |
65 // Build up a basic blob. | 65 // Build up a basic blob. |
66 const std::string kId("id"); | 66 const std::string kId("id"); |
67 SetupBasicBlob(&host, kId); | 67 SetupBasicBlob(&host, kId); |
68 | 68 |
69 // Get a handle to it. | 69 // Get a handle to it. |
70 scoped_ptr<BlobDataHandle> blob_data_handle = | 70 scoped_ptr<BlobDataHandle> blob_data_handle = |
71 context.GetBlobDataFromUUID(kId); | 71 context.GetBlobDataFromUUID(kId); |
72 EXPECT_TRUE(blob_data_handle); | 72 EXPECT_TRUE(blob_data_handle); |
(...skipping 55 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
128 EXPECT_TRUE(*(blob_data_handle->data()) == *blob_data1.get()); | 128 EXPECT_TRUE(*(blob_data_handle->data()) == *blob_data1.get()); |
129 | 129 |
130 // Test a blob composed in part with another blob. | 130 // Test a blob composed in part with another blob. |
131 blob_data_handle = context.AddFinishedBlob(blob_data2.get()); | 131 blob_data_handle = context.AddFinishedBlob(blob_data2.get()); |
132 ASSERT_TRUE(blob_data_handle.get()); | 132 ASSERT_TRUE(blob_data_handle.get()); |
133 EXPECT_TRUE(*(blob_data_handle->data()) == *canonicalized_blob_data2.get()); | 133 EXPECT_TRUE(*(blob_data_handle->data()) == *canonicalized_blob_data2.get()); |
134 } | 134 } |
135 | 135 |
136 TEST(BlobStorageContextTest, PublicBlobUrls) { | 136 TEST(BlobStorageContextTest, PublicBlobUrls) { |
137 BlobStorageContext context; | 137 BlobStorageContext context; |
138 BlobStorageHost host(&context); | 138 BlobStorageHostImpl host(&context); |
139 base::MessageLoop fake_io_message_loop; | 139 base::MessageLoop fake_io_message_loop; |
140 | 140 |
141 // Build up a basic blob. | 141 // Build up a basic blob. |
142 const std::string kId("id"); | 142 const std::string kId("id"); |
143 SetupBasicBlob(&host, kId); | 143 SetupBasicBlob(&host, kId); |
144 | 144 |
145 // Now register a url for that blob. | 145 // Now register a url for that blob. |
146 GURL kUrl("blob:id"); | 146 GURL kUrl("blob:id"); |
147 EXPECT_TRUE(host.RegisterPublicBlobURL(kUrl, kId)); | 147 EXPECT_TRUE(host.RegisterPublicBlobURL(kUrl, kId)); |
148 scoped_ptr<BlobDataHandle> blob_data_handle = | 148 scoped_ptr<BlobDataHandle> blob_data_handle = |
(...skipping 11 matching lines...) Expand all Loading... |
160 | 160 |
161 // Finally get rid of the url registration and the blob. | 161 // Finally get rid of the url registration and the blob. |
162 EXPECT_TRUE(host.RevokePublicBlobURL(kUrl)); | 162 EXPECT_TRUE(host.RevokePublicBlobURL(kUrl)); |
163 blob_data_handle = context.GetBlobDataFromPublicURL(kUrl); | 163 blob_data_handle = context.GetBlobDataFromPublicURL(kUrl); |
164 EXPECT_TRUE(!blob_data_handle.get()); | 164 EXPECT_TRUE(!blob_data_handle.get()); |
165 EXPECT_FALSE(host.RevokePublicBlobURL(kUrl)); | 165 EXPECT_FALSE(host.RevokePublicBlobURL(kUrl)); |
166 } | 166 } |
167 | 167 |
168 TEST(BlobStorageContextTest, HostCleanup) { | 168 TEST(BlobStorageContextTest, HostCleanup) { |
169 BlobStorageContext context; | 169 BlobStorageContext context; |
170 scoped_ptr<BlobStorageHost> host(new BlobStorageHost(&context)); | 170 scoped_ptr<BlobStorageHost> host(new BlobStorageHostImpl(&context)); |
171 base::MessageLoop fake_io_message_loop; | 171 base::MessageLoop fake_io_message_loop; |
172 | 172 |
173 // Build up a basic blob and register a url | 173 // Build up a basic blob and register a url |
174 const std::string kId("id"); | 174 const std::string kId("id"); |
175 GURL kUrl("blob:id"); | 175 GURL kUrl("blob:id"); |
176 SetupBasicBlob(host.get(), kId); | 176 SetupBasicBlob(host.get(), kId); |
177 EXPECT_TRUE(host->RegisterPublicBlobURL(kUrl, kId)); | 177 EXPECT_TRUE(host->RegisterPublicBlobURL(kUrl, kId)); |
178 | 178 |
179 // All should disappear upon host deletion. | 179 // All should disappear upon host deletion. |
180 host.reset(); | 180 host.reset(); |
181 scoped_ptr<BlobDataHandle> handle = context.GetBlobDataFromPublicURL(kUrl); | 181 scoped_ptr<BlobDataHandle> handle = context.GetBlobDataFromPublicURL(kUrl); |
182 EXPECT_TRUE(!handle.get()); | 182 EXPECT_TRUE(!handle.get()); |
183 handle = context.GetBlobDataFromUUID(kId); | 183 handle = context.GetBlobDataFromUUID(kId); |
184 EXPECT_TRUE(!handle.get()); | 184 EXPECT_TRUE(!handle.get()); |
185 } | 185 } |
186 | 186 |
187 TEST(BlobStorageContextTest, EarlyContextDeletion) { | 187 TEST(BlobStorageContextTest, EarlyContextDeletion) { |
188 scoped_ptr<BlobStorageContext> context(new BlobStorageContext); | 188 scoped_ptr<BlobStorageContext> context(new BlobStorageContext); |
189 BlobStorageHost host(context.get()); | 189 BlobStorageHostImpl host(context.get()); |
190 base::MessageLoop fake_io_message_loop; | 190 base::MessageLoop fake_io_message_loop; |
191 | 191 |
192 // Deleting the context should not induce crashes. | 192 // Deleting the context should not induce crashes. |
193 context.reset(); | 193 context.reset(); |
194 | 194 |
195 const std::string kId("id"); | 195 const std::string kId("id"); |
196 GURL kUrl("blob:id"); | 196 GURL kUrl("blob:id"); |
197 EXPECT_FALSE(host.StartBuildingBlob(kId)); | 197 EXPECT_FALSE(host.StartBuildingBlob(kId)); |
198 BlobData::Item item; | 198 BlobData::Item item; |
199 item.SetToBytes("1", 1); | 199 item.SetToBytes("1", 1); |
200 EXPECT_FALSE(host.AppendBlobDataItem(kId, item)); | 200 EXPECT_FALSE(host.AppendBlobDataItem(kId, item)); |
201 EXPECT_FALSE(host.FinishBuildingBlob(kId, "text/plain")); | 201 EXPECT_FALSE(host.FinishBuildingBlob(kId, "text/plain")); |
202 EXPECT_FALSE(host.RegisterPublicBlobURL(kUrl, kId)); | 202 EXPECT_FALSE(host.RegisterPublicBlobURL(kUrl, kId)); |
203 EXPECT_FALSE(host.IncrementBlobRefCount(kId)); | 203 EXPECT_FALSE(host.IncrementBlobRefCount(kId)); |
204 EXPECT_FALSE(host.DecrementBlobRefCount(kId)); | 204 EXPECT_FALSE(host.DecrementBlobRefCount(kId)); |
205 EXPECT_FALSE(host.RevokePublicBlobURL(kUrl)); | 205 EXPECT_FALSE(host.RevokePublicBlobURL(kUrl)); |
206 } | 206 } |
207 | 207 |
208 // TODO(michaeln): tests for the depcrecated url stuff | 208 // TODO(michaeln): tests for the depcrecated url stuff |
209 | 209 |
210 } // namespace content | 210 } // namespace content |
OLD | NEW |