OLD | NEW |
1 // Copyright 2015 The Chromium Authors. All rights reserved. | 1 // Copyright 2015 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 "storage/browser/blob/blob_storage_registry.h" | 5 #include "storage/browser/blob/blob_storage_registry.h" |
6 | 6 |
7 #include "base/callback.h" | 7 #include "base/callback.h" |
8 #include "testing/gtest/include/gtest/gtest.h" | 8 #include "testing/gtest/include/gtest/gtest.h" |
9 #include "url/gurl.h" | 9 #include "url/gurl.h" |
10 | 10 |
11 namespace storage { | 11 namespace storage { |
12 namespace { | 12 namespace { |
13 using Entry = BlobStorageRegistry::Entry; | 13 using Entry = BlobStorageRegistry::Entry; |
14 using BlobState = BlobStorageRegistry::BlobState; | 14 using BlobState = BlobStorageRegistry::BlobState; |
15 | 15 |
16 TEST(BlobStorageRegistry, UUIDRegistration) { | 16 TEST(BlobStorageRegistry, UUIDRegistration) { |
17 const std::string kBlob1 = "Blob1"; | 17 const std::string kBlob1 = "Blob1"; |
| 18 const std::string kType = "type1"; |
| 19 const std::string kDisposition = "disp1"; |
18 BlobStorageRegistry registry; | 20 BlobStorageRegistry registry; |
19 | 21 |
20 EXPECT_FALSE(registry.DeleteEntry(kBlob1)); | 22 EXPECT_FALSE(registry.DeleteEntry(kBlob1)); |
21 EXPECT_EQ(0u, registry.blob_count()); | 23 EXPECT_EQ(0u, registry.blob_count()); |
22 | 24 |
23 Entry* entry = registry.CreateEntry(kBlob1); | 25 Entry* entry = registry.CreateEntry(kBlob1, kType, kDisposition); |
24 ASSERT_NE(nullptr, entry); | 26 ASSERT_NE(nullptr, entry); |
25 EXPECT_EQ(BlobState::RESERVED, entry->state); | 27 EXPECT_EQ(BlobState::PENDING, entry->state); |
| 28 EXPECT_EQ(kType, entry->content_type); |
| 29 EXPECT_EQ(kDisposition, entry->content_disposition); |
26 EXPECT_EQ(1u, entry->refcount); | 30 EXPECT_EQ(1u, entry->refcount); |
27 EXPECT_FALSE(entry->exceeded_memory); | |
28 EXPECT_FALSE(entry->data.get() || entry->data_builder.get()); | 31 EXPECT_FALSE(entry->data.get() || entry->data_builder.get()); |
29 EXPECT_EQ(0u, entry->construction_complete_callbacks.size()); | 32 EXPECT_EQ(0u, entry->build_completion_callbacks.size()); |
30 | 33 |
31 EXPECT_EQ(entry, registry.GetEntry(kBlob1)); | 34 EXPECT_EQ(entry, registry.GetEntry(kBlob1)); |
32 EXPECT_TRUE(registry.DeleteEntry(kBlob1)); | 35 EXPECT_TRUE(registry.DeleteEntry(kBlob1)); |
33 entry = registry.CreateEntry(kBlob1); | 36 entry = registry.CreateEntry(kBlob1, kType, kDisposition); |
34 | 37 |
35 EXPECT_TRUE(entry->TestAndSetState(BlobState::RESERVED, | |
36 BlobState::ASYNC_TRANSPORTATION)); | |
37 EXPECT_FALSE( | |
38 entry->TestAndSetState(BlobState::CONSTRUCTION, BlobState::RESERVED)); | |
39 EXPECT_FALSE(entry->TestAndSetState(BlobState::ACTIVE, BlobState::RESERVED)); | |
40 EXPECT_TRUE(entry->TestAndSetState(BlobState::ASYNC_TRANSPORTATION, | |
41 BlobState::CONSTRUCTION)); | |
42 EXPECT_EQ(BlobState::CONSTRUCTION, entry->state); | |
43 EXPECT_EQ(1u, registry.blob_count()); | 38 EXPECT_EQ(1u, registry.blob_count()); |
44 } | 39 } |
45 | 40 |
46 TEST(BlobStorageRegistry, URLRegistration) { | 41 TEST(BlobStorageRegistry, URLRegistration) { |
47 const std::string kBlob = "Blob1"; | 42 const std::string kBlob = "Blob1"; |
| 43 const std::string kType = "type1"; |
| 44 const std::string kDisposition = "disp1"; |
48 const std::string kBlob2 = "Blob2"; | 45 const std::string kBlob2 = "Blob2"; |
49 const GURL kURL = GURL("blob://Blob1"); | 46 const GURL kURL = GURL("blob://Blob1"); |
50 const GURL kURL2 = GURL("blob://Blob2"); | 47 const GURL kURL2 = GURL("blob://Blob2"); |
51 BlobStorageRegistry registry; | 48 BlobStorageRegistry registry; |
52 | 49 |
53 EXPECT_FALSE(registry.IsURLMapped(kURL)); | 50 EXPECT_FALSE(registry.IsURLMapped(kURL)); |
54 EXPECT_EQ(nullptr, registry.GetEntryFromURL(kURL, nullptr)); | 51 EXPECT_EQ(nullptr, registry.GetEntryFromURL(kURL, nullptr)); |
55 EXPECT_FALSE(registry.DeleteURLMapping(kURL, nullptr)); | 52 EXPECT_FALSE(registry.DeleteURLMapping(kURL, nullptr)); |
56 EXPECT_FALSE(registry.CreateUrlMapping(kURL, kBlob)); | 53 EXPECT_FALSE(registry.CreateUrlMapping(kURL, kBlob)); |
57 EXPECT_EQ(0u, registry.url_count()); | 54 EXPECT_EQ(0u, registry.url_count()); |
58 Entry* entry = registry.CreateEntry(kBlob); | 55 Entry* entry = registry.CreateEntry(kBlob, kType, kDisposition); |
59 | 56 |
60 EXPECT_FALSE(registry.IsURLMapped(kURL)); | 57 EXPECT_FALSE(registry.IsURLMapped(kURL)); |
61 EXPECT_TRUE(registry.CreateUrlMapping(kURL, kBlob)); | 58 EXPECT_TRUE(registry.CreateUrlMapping(kURL, kBlob)); |
62 EXPECT_FALSE(registry.CreateUrlMapping(kURL, kBlob2)); | 59 EXPECT_FALSE(registry.CreateUrlMapping(kURL, kBlob2)); |
63 | 60 |
64 EXPECT_TRUE(registry.IsURLMapped(kURL)); | 61 EXPECT_TRUE(registry.IsURLMapped(kURL)); |
65 EXPECT_EQ(entry, registry.GetEntryFromURL(kURL, nullptr)); | 62 EXPECT_EQ(entry, registry.GetEntryFromURL(kURL, nullptr)); |
66 std::string uuid; | 63 std::string uuid; |
67 EXPECT_EQ(entry, registry.GetEntryFromURL(kURL, &uuid)); | 64 EXPECT_EQ(entry, registry.GetEntryFromURL(kURL, &uuid)); |
68 EXPECT_EQ(kBlob, uuid); | 65 EXPECT_EQ(kBlob, uuid); |
69 EXPECT_EQ(1u, registry.url_count()); | 66 EXPECT_EQ(1u, registry.url_count()); |
70 | 67 |
71 registry.CreateEntry(kBlob2); | 68 registry.CreateEntry(kBlob2, kType, kDisposition); |
72 EXPECT_TRUE(registry.CreateUrlMapping(kURL2, kBlob2)); | 69 EXPECT_TRUE(registry.CreateUrlMapping(kURL2, kBlob2)); |
73 EXPECT_EQ(2u, registry.url_count()); | 70 EXPECT_EQ(2u, registry.url_count()); |
74 EXPECT_TRUE(registry.DeleteURLMapping(kURL2, &uuid)); | 71 EXPECT_TRUE(registry.DeleteURLMapping(kURL2, &uuid)); |
75 EXPECT_EQ(kBlob2, uuid); | 72 EXPECT_EQ(kBlob2, uuid); |
76 EXPECT_FALSE(registry.IsURLMapped(kURL2)); | 73 EXPECT_FALSE(registry.IsURLMapped(kURL2)); |
77 | 74 |
78 // Both urls point to the same blob. | 75 // Both urls point to the same blob. |
79 EXPECT_TRUE(registry.CreateUrlMapping(kURL2, kBlob)); | 76 EXPECT_TRUE(registry.CreateUrlMapping(kURL2, kBlob)); |
80 std::string uuid2; | 77 std::string uuid2; |
81 EXPECT_EQ(registry.GetEntryFromURL(kURL, &uuid), | 78 EXPECT_EQ(registry.GetEntryFromURL(kURL, &uuid), |
82 registry.GetEntryFromURL(kURL2, &uuid2)); | 79 registry.GetEntryFromURL(kURL2, &uuid2)); |
83 EXPECT_EQ(uuid, uuid2); | 80 EXPECT_EQ(uuid, uuid2); |
84 } | 81 } |
85 | 82 |
86 } // namespace | 83 } // namespace |
87 } // namespace storage | 84 } // namespace storage |
OLD | NEW |