| OLD | NEW |
| 1 // Copyright 2016 The Chromium Authors. All rights reserved. | 1 // Copyright 2016 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/dom_storage/local_storage_context_mojo.h" | 5 #include "content/browser/dom_storage/local_storage_context_mojo.h" |
| 6 | 6 |
| 7 #include "base/files/file_enumerator.h" | 7 #include "base/files/file_enumerator.h" |
| 8 #include "base/files/scoped_temp_dir.h" | 8 #include "base/files/scoped_temp_dir.h" |
| 9 #include "base/run_loop.h" | 9 #include "base/run_loop.h" |
| 10 #include "base/strings/utf_string_conversions.h" | 10 #include "base/strings/utf_string_conversions.h" |
| (...skipping 61 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 72 struct Observation { | 72 struct Observation { |
| 73 enum { kAdd, kChange, kDelete, kDeleteAll } type; | 73 enum { kAdd, kChange, kDelete, kDeleteAll } type; |
| 74 std::string key; | 74 std::string key; |
| 75 std::string old_value; | 75 std::string old_value; |
| 76 std::string new_value; | 76 std::string new_value; |
| 77 std::string source; | 77 std::string source; |
| 78 }; | 78 }; |
| 79 | 79 |
| 80 TestLevelDBObserver() : binding_(this) {} | 80 TestLevelDBObserver() : binding_(this) {} |
| 81 | 81 |
| 82 mojom::LevelDBObserverAssociatedPtrInfo Bind( | 82 mojom::LevelDBObserverAssociatedPtrInfo Bind() { |
| 83 mojo::AssociatedGroup* associated_group) { | |
| 84 mojom::LevelDBObserverAssociatedPtrInfo ptr_info; | 83 mojom::LevelDBObserverAssociatedPtrInfo ptr_info; |
| 85 binding_.Bind(&ptr_info, associated_group); | 84 binding_.Bind(&ptr_info); |
| 86 return ptr_info; | 85 return ptr_info; |
| 87 } | 86 } |
| 88 | 87 |
| 89 const std::vector<Observation>& observations() { return observations_; } | 88 const std::vector<Observation>& observations() { return observations_; } |
| 90 | 89 |
| 91 private: | 90 private: |
| 92 void KeyAdded(const std::vector<uint8_t>& key, | 91 void KeyAdded(const std::vector<uint8_t>& key, |
| 93 const std::vector<uint8_t>& value, | 92 const std::vector<uint8_t>& value, |
| 94 const std::string& source) override { | 93 const std::string& source) override { |
| 95 observations_.push_back({Observation::kAdd, Uint8VectorToStdString(key), "", | 94 observations_.push_back({Observation::kAdd, Uint8VectorToStdString(key), "", |
| (...skipping 75 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 171 run_loop.QuitClosure(), &result)); | 170 run_loop.QuitClosure(), &result)); |
| 172 run_loop.Run(); | 171 run_loop.Run(); |
| 173 return result; | 172 return result; |
| 174 } | 173 } |
| 175 | 174 |
| 176 private: | 175 private: |
| 177 TestBrowserThreadBundle thread_bundle_; | 176 TestBrowserThreadBundle thread_bundle_; |
| 178 base::ScopedTempDir temp_path_; | 177 base::ScopedTempDir temp_path_; |
| 179 std::map<std::vector<uint8_t>, std::vector<uint8_t>> mock_data_; | 178 std::map<std::vector<uint8_t>, std::vector<uint8_t>> mock_data_; |
| 180 MockLevelDBDatabase db_; | 179 MockLevelDBDatabase db_; |
| 181 mojo::AssociatedGroup associated_group_; | |
| 182 mojo::AssociatedBinding<leveldb::mojom::LevelDBDatabase> db_binding_; | 180 mojo::AssociatedBinding<leveldb::mojom::LevelDBDatabase> db_binding_; |
| 183 | 181 |
| 184 scoped_refptr<MockDOMStorageTaskRunner> task_runner_; | 182 scoped_refptr<MockDOMStorageTaskRunner> task_runner_; |
| 185 scoped_refptr<DOMStorageContextImpl> dom_storage_context_; | 183 scoped_refptr<DOMStorageContextImpl> dom_storage_context_; |
| 186 | 184 |
| 187 std::unique_ptr<LocalStorageContextMojo> context_; | 185 std::unique_ptr<LocalStorageContextMojo> context_; |
| 188 | 186 |
| 189 DISALLOW_COPY_AND_ASSIGN(LocalStorageContextMojoTest); | 187 DISALLOW_COPY_AND_ASSIGN(LocalStorageContextMojoTest); |
| 190 }; | 188 }; |
| 191 | 189 |
| (...skipping 262 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 454 context()->OpenLocalStorage(origin2, MakeRequest(&wrapper)); | 452 context()->OpenLocalStorage(origin2, MakeRequest(&wrapper)); |
| 455 wrapper->Put(key, value, "source", base::Bind(&NoOpSuccess)); | 453 wrapper->Put(key, value, "source", base::Bind(&NoOpSuccess)); |
| 456 wrapper.reset(); | 454 wrapper.reset(); |
| 457 | 455 |
| 458 // Make sure all data gets committed to disk. | 456 // Make sure all data gets committed to disk. |
| 459 base::RunLoop().RunUntilIdle(); | 457 base::RunLoop().RunUntilIdle(); |
| 460 EXPECT_FALSE(mock_data().empty()); | 458 EXPECT_FALSE(mock_data().empty()); |
| 461 | 459 |
| 462 TestLevelDBObserver observer; | 460 TestLevelDBObserver observer; |
| 463 context()->OpenLocalStorage(origin1, MakeRequest(&wrapper)); | 461 context()->OpenLocalStorage(origin1, MakeRequest(&wrapper)); |
| 464 wrapper->AddObserver(observer.Bind(wrapper.associated_group())); | 462 wrapper->AddObserver(observer.Bind()); |
| 465 base::RunLoop().RunUntilIdle(); | 463 base::RunLoop().RunUntilIdle(); |
| 466 | 464 |
| 467 context()->DeleteStorage(origin1); | 465 context()->DeleteStorage(origin1); |
| 468 base::RunLoop().RunUntilIdle(); | 466 base::RunLoop().RunUntilIdle(); |
| 469 | 467 |
| 470 ASSERT_EQ(1u, observer.observations().size()); | 468 ASSERT_EQ(1u, observer.observations().size()); |
| 471 EXPECT_EQ(TestLevelDBObserver::Observation::kDeleteAll, | 469 EXPECT_EQ(TestLevelDBObserver::Observation::kDeleteAll, |
| 472 observer.observations()[0].type); | 470 observer.observations()[0].type); |
| 473 | 471 |
| 474 // Data from origin2 should exist, including meta-data, but nothing should | 472 // Data from origin2 should exist, including meta-data, but nothing should |
| (...skipping 23 matching lines...) Expand all Loading... |
| 498 context()->OpenLocalStorage(origin2, MakeRequest(&wrapper)); | 496 context()->OpenLocalStorage(origin2, MakeRequest(&wrapper)); |
| 499 wrapper->Put(key, value, "source", base::Bind(&NoOpSuccess)); | 497 wrapper->Put(key, value, "source", base::Bind(&NoOpSuccess)); |
| 500 wrapper.reset(); | 498 wrapper.reset(); |
| 501 | 499 |
| 502 // Make sure all data gets committed to disk. | 500 // Make sure all data gets committed to disk. |
| 503 base::RunLoop().RunUntilIdle(); | 501 base::RunLoop().RunUntilIdle(); |
| 504 EXPECT_FALSE(mock_data().empty()); | 502 EXPECT_FALSE(mock_data().empty()); |
| 505 | 503 |
| 506 TestLevelDBObserver observer; | 504 TestLevelDBObserver observer; |
| 507 context()->OpenLocalStorage(origin1, MakeRequest(&wrapper)); | 505 context()->OpenLocalStorage(origin1, MakeRequest(&wrapper)); |
| 508 wrapper->AddObserver(observer.Bind(wrapper.associated_group())); | 506 wrapper->AddObserver(observer.Bind()); |
| 509 wrapper->Put(StdStringToUint8Vector("key2"), value, "source", | 507 wrapper->Put(StdStringToUint8Vector("key2"), value, "source", |
| 510 base::Bind(&NoOpSuccess)); | 508 base::Bind(&NoOpSuccess)); |
| 511 base::RunLoop().RunUntilIdle(); | 509 base::RunLoop().RunUntilIdle(); |
| 512 | 510 |
| 513 context()->DeleteStorage(origin1); | 511 context()->DeleteStorage(origin1); |
| 514 base::RunLoop().RunUntilIdle(); | 512 base::RunLoop().RunUntilIdle(); |
| 515 | 513 |
| 516 ASSERT_EQ(2u, observer.observations().size()); | 514 ASSERT_EQ(2u, observer.observations().size()); |
| 517 EXPECT_EQ(TestLevelDBObserver::Observation::kAdd, | 515 EXPECT_EQ(TestLevelDBObserver::Observation::kAdd, |
| 518 observer.observations()[0].type); | 516 observer.observations()[0].type); |
| (...skipping 325 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 844 base::RunLoop().RunUntilIdle(); | 842 base::RunLoop().RunUntilIdle(); |
| 845 | 843 |
| 846 // Data should have been preserved now. | 844 // Data should have been preserved now. |
| 847 context = base::MakeUnique<LocalStorageContextMojo>( | 845 context = base::MakeUnique<LocalStorageContextMojo>( |
| 848 connector(), nullptr, base::FilePath(), test_path); | 846 connector(), nullptr, base::FilePath(), test_path); |
| 849 EXPECT_TRUE(DoTestGet(context.get(), key, &result)); | 847 EXPECT_TRUE(DoTestGet(context.get(), key, &result)); |
| 850 EXPECT_EQ(value, result); | 848 EXPECT_EQ(value, result); |
| 851 } | 849 } |
| 852 | 850 |
| 853 } // namespace content | 851 } // namespace content |
| OLD | NEW |