| OLD | NEW |
| (Empty) |
| 1 // Copyright (c) 2013 The Chromium Authors. All rights reserved. | |
| 2 // Use of this source code is governed by a BSD-style license that can be | |
| 3 // found in the LICENSE file. | |
| 4 | |
| 5 #include "content/browser/indexed_db/indexed_db_backing_store.h" | |
| 6 | |
| 7 #include "base/files/scoped_temp_dir.h" | |
| 8 #include "base/logging.h" | |
| 9 #include "base/string16.h" | |
| 10 #include "base/strings/utf_string_conversions.h" | |
| 11 #include "content/browser/indexed_db/indexed_db_factory_impl.h" | |
| 12 #include "content/browser/indexed_db/indexed_db_leveldb_coding.h" | |
| 13 #include "testing/gtest/include/gtest/gtest.h" | |
| 14 #include "third_party/WebKit/Source/Platform/chromium/public/WebData.h" | |
| 15 #include "third_party/WebKit/Source/WebKit/chromium/public/WebSecurityOrigin.h" | |
| 16 | |
| 17 using WebKit::WebSecurityOrigin; | |
| 18 using WebKit::WebIDBKey; | |
| 19 | |
| 20 namespace content { | |
| 21 | |
| 22 namespace { | |
| 23 | |
| 24 class IndexedDBBackingStoreTest : public testing::Test { | |
| 25 public: | |
| 26 IndexedDBBackingStoreTest() {} | |
| 27 virtual void SetUp() { | |
| 28 string16 file_identifier; | |
| 29 backing_store_ = IndexedDBBackingStore::OpenInMemory(file_identifier); | |
| 30 | |
| 31 // useful keys and values during tests | |
| 32 const char raw_value1[] = "value1"; | |
| 33 | |
| 34 const char raw_value2[] = "value2"; | |
| 35 const char raw_value3[] = "value3"; | |
| 36 m_value1.insert( | |
| 37 m_value1.end(), &raw_value1[0], &raw_value1[0] + sizeof(raw_value1)); | |
| 38 m_value2.insert( | |
| 39 m_value2.end(), &raw_value2[0], &raw_value2[0] + sizeof(raw_value2)); | |
| 40 m_value3.insert( | |
| 41 m_value3.end(), &raw_value3[0], &raw_value3[0] + sizeof(raw_value3)); | |
| 42 m_key1 = IndexedDBKey(99, WebIDBKey::NumberType); | |
| 43 m_key2 = IndexedDBKey(ASCIIToUTF16("key2")); | |
| 44 m_key3 = IndexedDBKey(ASCIIToUTF16("key3")); | |
| 45 } | |
| 46 | |
| 47 protected: | |
| 48 scoped_refptr<IndexedDBBackingStore> backing_store_; | |
| 49 | |
| 50 // Sample keys and values that are consistent. | |
| 51 IndexedDBKey m_key1; | |
| 52 IndexedDBKey m_key2; | |
| 53 IndexedDBKey m_key3; | |
| 54 std::vector<char> m_value1; | |
| 55 std::vector<char> m_value2; | |
| 56 std::vector<char> m_value3; | |
| 57 }; | |
| 58 | |
| 59 TEST_F(IndexedDBBackingStoreTest, PutGetConsistency) { | |
| 60 { | |
| 61 IndexedDBBackingStore::Transaction transaction1(backing_store_.get()); | |
| 62 transaction1.begin(); | |
| 63 IndexedDBBackingStore::RecordIdentifier record; | |
| 64 bool ok = backing_store_->PutRecord( | |
| 65 &transaction1, 1, 1, m_key1, m_value1, &record); | |
| 66 EXPECT_TRUE(ok); | |
| 67 transaction1.Commit(); | |
| 68 } | |
| 69 | |
| 70 { | |
| 71 IndexedDBBackingStore::Transaction transaction2(backing_store_.get()); | |
| 72 transaction2.begin(); | |
| 73 std::vector<char> result_value; | |
| 74 bool ok = | |
| 75 backing_store_->GetRecord(&transaction2, 1, 1, m_key1, result_value); | |
| 76 transaction2.Commit(); | |
| 77 EXPECT_TRUE(ok); | |
| 78 EXPECT_EQ(m_value1, result_value); | |
| 79 } | |
| 80 } | |
| 81 | |
| 82 // Make sure that using very high ( more than 32 bit ) values for database_id | |
| 83 // and object_store_id still work. | |
| 84 TEST_F(IndexedDBBackingStoreTest, HighIds) { | |
| 85 const int64 high_database_id = 1ULL << 35; | |
| 86 const int64 high_object_store_id = 1ULL << 39; | |
| 87 // index_ids are capped at 32 bits for storage purposes. | |
| 88 const int64 high_index_id = 1ULL << 29; | |
| 89 | |
| 90 const int64 invalid_high_index_id = 1ULL << 37; | |
| 91 | |
| 92 const IndexedDBKey& index_key = m_key2; | |
| 93 std::vector<char> index_key_raw = EncodeIDBKey(index_key); | |
| 94 { | |
| 95 IndexedDBBackingStore::Transaction transaction1(backing_store_.get()); | |
| 96 transaction1.begin(); | |
| 97 IndexedDBBackingStore::RecordIdentifier record; | |
| 98 bool ok = backing_store_->PutRecord(&transaction1, | |
| 99 high_database_id, | |
| 100 high_object_store_id, | |
| 101 m_key1, | |
| 102 m_value1, | |
| 103 &record); | |
| 104 EXPECT_TRUE(ok); | |
| 105 | |
| 106 ok = backing_store_->PutIndexDataForRecord(&transaction1, | |
| 107 high_database_id, | |
| 108 high_object_store_id, | |
| 109 invalid_high_index_id, | |
| 110 index_key, | |
| 111 record); | |
| 112 EXPECT_FALSE(ok); | |
| 113 | |
| 114 ok = backing_store_->PutIndexDataForRecord(&transaction1, | |
| 115 high_database_id, | |
| 116 high_object_store_id, | |
| 117 high_index_id, | |
| 118 index_key, | |
| 119 record); | |
| 120 EXPECT_TRUE(ok); | |
| 121 | |
| 122 ok = transaction1.Commit(); | |
| 123 EXPECT_TRUE(ok); | |
| 124 } | |
| 125 | |
| 126 { | |
| 127 IndexedDBBackingStore::Transaction transaction2(backing_store_.get()); | |
| 128 transaction2.begin(); | |
| 129 std::vector<char> result_value; | |
| 130 bool ok = backing_store_->GetRecord(&transaction2, | |
| 131 high_database_id, | |
| 132 high_object_store_id, | |
| 133 m_key1, | |
| 134 result_value); | |
| 135 EXPECT_TRUE(ok); | |
| 136 EXPECT_EQ(m_value1, result_value); | |
| 137 | |
| 138 scoped_ptr<IndexedDBKey> new_primary_key; | |
| 139 ok = backing_store_->GetPrimaryKeyViaIndex(&transaction2, | |
| 140 high_database_id, | |
| 141 high_object_store_id, | |
| 142 invalid_high_index_id, | |
| 143 index_key, | |
| 144 &new_primary_key); | |
| 145 EXPECT_FALSE(ok); | |
| 146 | |
| 147 ok = backing_store_->GetPrimaryKeyViaIndex(&transaction2, | |
| 148 high_database_id, | |
| 149 high_object_store_id, | |
| 150 high_index_id, | |
| 151 index_key, | |
| 152 &new_primary_key); | |
| 153 EXPECT_TRUE(ok); | |
| 154 EXPECT_TRUE(new_primary_key->IsEqual(m_key1)); | |
| 155 | |
| 156 ok = transaction2.Commit(); | |
| 157 EXPECT_TRUE(ok); | |
| 158 } | |
| 159 } | |
| 160 | |
| 161 // Make sure that other invalid ids do not crash. | |
| 162 TEST_F(IndexedDBBackingStoreTest, InvalidIds) { | |
| 163 // valid ids for use when testing invalid ids | |
| 164 const int64 database_id = 1; | |
| 165 const int64 object_store_id = 1; | |
| 166 const int64 index_id = kMinimumIndexId; | |
| 167 const int64 invalid_low_index_id = 19; // index_ids must be > kMinimumIndexId | |
| 168 | |
| 169 std::vector<char> result_value; | |
| 170 | |
| 171 IndexedDBBackingStore::Transaction transaction1(backing_store_.get()); | |
| 172 transaction1.begin(); | |
| 173 | |
| 174 IndexedDBBackingStore::RecordIdentifier record; | |
| 175 bool ok = backing_store_->PutRecord(&transaction1, | |
| 176 database_id, | |
| 177 KeyPrefix::kInvalidId, | |
| 178 m_key1, | |
| 179 m_value1, | |
| 180 &record); | |
| 181 EXPECT_FALSE(ok); | |
| 182 ok = backing_store_->PutRecord( | |
| 183 &transaction1, database_id, 0, m_key1, m_value1, &record); | |
| 184 EXPECT_FALSE(ok); | |
| 185 ok = backing_store_->PutRecord(&transaction1, | |
| 186 KeyPrefix::kInvalidId, | |
| 187 object_store_id, | |
| 188 m_key1, | |
| 189 m_value1, | |
| 190 &record); | |
| 191 EXPECT_FALSE(ok); | |
| 192 ok = backing_store_->PutRecord( | |
| 193 &transaction1, 0, object_store_id, m_key1, m_value1, &record); | |
| 194 EXPECT_FALSE(ok); | |
| 195 | |
| 196 ok = backing_store_->GetRecord( | |
| 197 &transaction1, database_id, KeyPrefix::kInvalidId, m_key1, result_value); | |
| 198 EXPECT_FALSE(ok); | |
| 199 ok = backing_store_->GetRecord( | |
| 200 &transaction1, database_id, 0, m_key1, result_value); | |
| 201 EXPECT_FALSE(ok); | |
| 202 ok = backing_store_->GetRecord(&transaction1, | |
| 203 KeyPrefix::kInvalidId, | |
| 204 object_store_id, | |
| 205 m_key1, | |
| 206 result_value); | |
| 207 EXPECT_FALSE(ok); | |
| 208 ok = backing_store_->GetRecord( | |
| 209 &transaction1, 0, object_store_id, m_key1, result_value); | |
| 210 EXPECT_FALSE(ok); | |
| 211 | |
| 212 scoped_ptr<IndexedDBKey> new_primary_key; | |
| 213 ok = backing_store_->GetPrimaryKeyViaIndex(&transaction1, | |
| 214 database_id, | |
| 215 object_store_id, | |
| 216 KeyPrefix::kInvalidId, | |
| 217 m_key1, | |
| 218 &new_primary_key); | |
| 219 EXPECT_FALSE(ok); | |
| 220 ok = backing_store_->GetPrimaryKeyViaIndex(&transaction1, | |
| 221 database_id, | |
| 222 object_store_id, | |
| 223 invalid_low_index_id, | |
| 224 m_key1, | |
| 225 &new_primary_key); | |
| 226 EXPECT_FALSE(ok); | |
| 227 ok = backing_store_->GetPrimaryKeyViaIndex( | |
| 228 &transaction1, database_id, object_store_id, 0, m_key1, &new_primary_key); | |
| 229 EXPECT_FALSE(ok); | |
| 230 | |
| 231 ok = backing_store_->GetPrimaryKeyViaIndex(&transaction1, | |
| 232 KeyPrefix::kInvalidId, | |
| 233 object_store_id, | |
| 234 index_id, | |
| 235 m_key1, | |
| 236 &new_primary_key); | |
| 237 EXPECT_FALSE(ok); | |
| 238 ok = backing_store_->GetPrimaryKeyViaIndex(&transaction1, | |
| 239 database_id, | |
| 240 KeyPrefix::kInvalidId, | |
| 241 index_id, | |
| 242 m_key1, | |
| 243 &new_primary_key); | |
| 244 EXPECT_FALSE(ok); | |
| 245 } | |
| 246 | |
| 247 TEST_F(IndexedDBBackingStoreTest, CreateDatabase) { | |
| 248 const string16 database_name(ASCIIToUTF16("db1")); | |
| 249 int64 database_id; | |
| 250 const string16 version(ASCIIToUTF16("old_string_version")); | |
| 251 const int64 int_version = 9; | |
| 252 | |
| 253 const int64 object_store_id = 99; | |
| 254 const string16 object_store_name(ASCIIToUTF16("object_store1")); | |
| 255 const bool auto_increment = true; | |
| 256 const IndexedDBKeyPath object_store_key_path( | |
| 257 ASCIIToUTF16("object_store_key")); | |
| 258 | |
| 259 const int64 index_id = 999; | |
| 260 const string16 index_name(ASCIIToUTF16("index1")); | |
| 261 const bool unique = true; | |
| 262 const bool multi_entry = true; | |
| 263 const IndexedDBKeyPath index_key_path(ASCIIToUTF16("index_key")); | |
| 264 | |
| 265 { | |
| 266 bool ok = backing_store_->CreateIDBDatabaseMetaData( | |
| 267 database_name, version, int_version, database_id); | |
| 268 EXPECT_TRUE(ok); | |
| 269 EXPECT_GT(database_id, 0); | |
| 270 | |
| 271 IndexedDBBackingStore::Transaction transaction(backing_store_.get()); | |
| 272 transaction.begin(); | |
| 273 | |
| 274 ok = backing_store_->CreateObjectStore(&transaction, | |
| 275 database_id, | |
| 276 object_store_id, | |
| 277 object_store_name, | |
| 278 object_store_key_path, | |
| 279 auto_increment); | |
| 280 EXPECT_TRUE(ok); | |
| 281 | |
| 282 ok = backing_store_->CreateIndex(&transaction, | |
| 283 database_id, | |
| 284 object_store_id, | |
| 285 index_id, | |
| 286 index_name, | |
| 287 index_key_path, | |
| 288 unique, | |
| 289 multi_entry); | |
| 290 EXPECT_TRUE(ok); | |
| 291 | |
| 292 ok = transaction.Commit(); | |
| 293 EXPECT_TRUE(ok); | |
| 294 } | |
| 295 | |
| 296 { | |
| 297 IndexedDBDatabaseMetadata database; | |
| 298 bool found; | |
| 299 bool ok = | |
| 300 backing_store_->GetIDBDatabaseMetaData(database_name, &database, found); | |
| 301 EXPECT_TRUE(ok); | |
| 302 EXPECT_TRUE(found); | |
| 303 | |
| 304 // database.name is not filled in by the implementation. | |
| 305 EXPECT_EQ(version, database.version); | |
| 306 EXPECT_EQ(int_version, database.int_version); | |
| 307 EXPECT_EQ(database_id, database.id); | |
| 308 | |
| 309 ok = backing_store_->GetObjectStores(database.id, &database.object_stores); | |
| 310 EXPECT_TRUE(ok); | |
| 311 | |
| 312 EXPECT_EQ(1UL, database.object_stores.size()); | |
| 313 IndexedDBObjectStoreMetadata object_store = | |
| 314 database.object_stores[object_store_id]; | |
| 315 EXPECT_EQ(object_store_name, object_store.name); | |
| 316 EXPECT_EQ(object_store_key_path, object_store.key_path); | |
| 317 EXPECT_EQ(auto_increment, object_store.auto_increment); | |
| 318 | |
| 319 EXPECT_EQ(1UL, object_store.indexes.size()); | |
| 320 IndexedDBIndexMetadata index = object_store.indexes[index_id]; | |
| 321 EXPECT_EQ(index_name, index.name); | |
| 322 EXPECT_EQ(index_key_path, index.key_path); | |
| 323 EXPECT_EQ(unique, index.unique); | |
| 324 EXPECT_EQ(multi_entry, index.multi_entry); | |
| 325 } | |
| 326 } | |
| 327 | |
| 328 class MockIDBFactory : public IndexedDBFactoryImpl { | |
| 329 public: | |
| 330 static scoped_refptr<MockIDBFactory> Create() { | |
| 331 return make_scoped_refptr(new MockIDBFactory()); | |
| 332 } | |
| 333 | |
| 334 scoped_refptr<IndexedDBBackingStore> TestOpenBackingStore( | |
| 335 const WebSecurityOrigin& origin, | |
| 336 const base::FilePath& data_directory) { | |
| 337 string16 path = UTF8ToUTF16(data_directory.AsUTF8Unsafe()); | |
| 338 return OpenBackingStore(origin.databaseIdentifier(), path); | |
| 339 } | |
| 340 | |
| 341 private: | |
| 342 virtual ~MockIDBFactory() {} | |
| 343 }; | |
| 344 | |
| 345 TEST(IndexedDBFactoryTest, BackingStoreLifetime) { | |
| 346 WebSecurityOrigin origin1 = | |
| 347 WebSecurityOrigin::createFromString("http://localhost:81"); | |
| 348 WebSecurityOrigin origin2 = | |
| 349 WebSecurityOrigin::createFromString("http://localhost:82"); | |
| 350 | |
| 351 scoped_refptr<MockIDBFactory> factory = MockIDBFactory::Create(); | |
| 352 | |
| 353 base::ScopedTempDir temp_directory; | |
| 354 ASSERT_TRUE(temp_directory.CreateUniqueTempDir()); | |
| 355 scoped_refptr<IndexedDBBackingStore> disk_store1 = | |
| 356 factory->TestOpenBackingStore(origin1, temp_directory.path()); | |
| 357 EXPECT_TRUE(disk_store1->HasOneRef()); | |
| 358 | |
| 359 scoped_refptr<IndexedDBBackingStore> disk_store2 = | |
| 360 factory->TestOpenBackingStore(origin1, temp_directory.path()); | |
| 361 EXPECT_EQ(disk_store1.get(), disk_store2.get()); | |
| 362 // TODO(alecflett): Verify refcount indirectly. | |
| 363 // EXPECT_EQ(2, disk_store2->RefCount()); | |
| 364 | |
| 365 scoped_refptr<IndexedDBBackingStore> disk_store3 = | |
| 366 factory->TestOpenBackingStore(origin2, temp_directory.path()); | |
| 367 EXPECT_TRUE(disk_store3->HasOneRef()); | |
| 368 // TODO(alecflett): Verify refcount indirectly. | |
| 369 // EXPECT_EQ(2, disk_store1->RefCount()); | |
| 370 } | |
| 371 | |
| 372 TEST(IndexedDBFactoryTest, MemoryBackingStoreLifetime) { | |
| 373 WebSecurityOrigin origin1 = | |
| 374 WebSecurityOrigin::createFromString("http://localhost:81"); | |
| 375 WebSecurityOrigin origin2 = | |
| 376 WebSecurityOrigin::createFromString("http://localhost:82"); | |
| 377 | |
| 378 scoped_refptr<MockIDBFactory> factory = MockIDBFactory::Create(); | |
| 379 scoped_refptr<IndexedDBBackingStore> mem_store1 = | |
| 380 factory->TestOpenBackingStore(origin1, base::FilePath()); | |
| 381 // TODO(alecflett): Verify refcount indirectly. | |
| 382 // EXPECT_EQ(2, mem_store1->RefCount()); | |
| 383 scoped_refptr<IndexedDBBackingStore> mem_store2 = | |
| 384 factory->TestOpenBackingStore(origin1, base::FilePath()); | |
| 385 EXPECT_EQ(mem_store1.get(), mem_store2.get()); | |
| 386 // TODO(alecflett): Verify refcount indirectly. | |
| 387 // EXPECT_EQ(3, mem_store2->RefCount()); | |
| 388 | |
| 389 scoped_refptr<IndexedDBBackingStore> mem_store3 = | |
| 390 factory->TestOpenBackingStore(origin2, base::FilePath()); | |
| 391 // TODO(alecflett): Verify refcount indirectly. | |
| 392 // EXPECT_EQ(2, mem_store3->RefCount()); | |
| 393 // EXPECT_EQ(3, mem_store1->RefCount()); | |
| 394 | |
| 395 factory = NULL; | |
| 396 // TODO(alecflett): Verify refcount indirectly. | |
| 397 // EXPECT_EQ(2, mem_store1->RefCount()); | |
| 398 // EXPECT_EQ(1, mem_store3->RefCount()); | |
| 399 } | |
| 400 | |
| 401 } // namespace | |
| 402 | |
| 403 } // namespace content | |
| OLD | NEW |