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 "components/leveldb_proto/proto_database_impl.h" | 5 #include "components/leveldb_proto/proto_database_impl.h" |
6 | 6 |
7 #include <map> | 7 #include <map> |
8 | 8 |
9 #include "base/bind.h" | 9 #include "base/bind.h" |
10 #include "base/files/file_util.h" | 10 #include "base/files/file_util.h" |
11 #include "base/files/scoped_temp_dir.h" | 11 #include "base/files/scoped_temp_dir.h" |
| 12 #include "base/location.h" |
12 #include "base/run_loop.h" | 13 #include "base/run_loop.h" |
13 #include "base/threading/thread.h" | 14 #include "base/threading/thread.h" |
14 #include "components/leveldb_proto/leveldb_database.h" | 15 #include "components/leveldb_proto/leveldb_database.h" |
15 #include "components/leveldb_proto/testing/proto/test.pb.h" | 16 #include "components/leveldb_proto/testing/proto/test.pb.h" |
16 #include "testing/gmock/include/gmock/gmock.h" | 17 #include "testing/gmock/include/gmock/gmock.h" |
17 #include "testing/gtest/include/gtest/gtest.h" | 18 #include "testing/gtest/include/gtest/gtest.h" |
18 #include "third_party/leveldatabase/src/include/leveldb/options.h" | 19 #include "third_party/leveldatabase/src/include/leveldb/options.h" |
19 | 20 |
20 using base::MessageLoop; | 21 using base::MessageLoop; |
21 using base::ScopedTempDir; | 22 using base::ScopedTempDir; |
(...skipping 57 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
79 std::string serialized_actual = actual[i].SerializeAsString(); | 80 std::string serialized_actual = actual[i].SerializeAsString(); |
80 EXPECT_EQ(serialized_expected, serialized_actual); | 81 EXPECT_EQ(serialized_expected, serialized_actual); |
81 expected.erase(expected_it); | 82 expected.erase(expected_it); |
82 } | 83 } |
83 } | 84 } |
84 | 85 |
85 class ProtoDatabaseImplTest : public testing::Test { | 86 class ProtoDatabaseImplTest : public testing::Test { |
86 public: | 87 public: |
87 void SetUp() override { | 88 void SetUp() override { |
88 main_loop_.reset(new MessageLoop()); | 89 main_loop_.reset(new MessageLoop()); |
89 db_.reset( | 90 db_.reset(new ProtoDatabaseImpl<TestProto>(main_loop_->task_runner())); |
90 new ProtoDatabaseImpl<TestProto>(main_loop_->message_loop_proxy())); | |
91 } | 91 } |
92 | 92 |
93 void TearDown() override { | 93 void TearDown() override { |
94 db_.reset(); | 94 db_.reset(); |
95 base::RunLoop().RunUntilIdle(); | 95 base::RunLoop().RunUntilIdle(); |
96 main_loop_.reset(); | 96 main_loop_.reset(); |
97 } | 97 } |
98 | 98 |
99 scoped_ptr<ProtoDatabaseImpl<TestProto> > db_; | 99 scoped_ptr<ProtoDatabaseImpl<TestProto> > db_; |
100 scoped_ptr<MessageLoop> main_loop_; | 100 scoped_ptr<MessageLoop> main_loop_; |
(...skipping 224 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
325 // threading violations. | 325 // threading violations. |
326 TEST(ProtoDatabaseImplThreadingTest, TestDBDestruction) { | 326 TEST(ProtoDatabaseImplThreadingTest, TestDBDestruction) { |
327 base::MessageLoop main_loop; | 327 base::MessageLoop main_loop; |
328 | 328 |
329 ScopedTempDir temp_dir; | 329 ScopedTempDir temp_dir; |
330 ASSERT_TRUE(temp_dir.CreateUniqueTempDir()); | 330 ASSERT_TRUE(temp_dir.CreateUniqueTempDir()); |
331 | 331 |
332 base::Thread db_thread("dbthread"); | 332 base::Thread db_thread("dbthread"); |
333 ASSERT_TRUE(db_thread.Start()); | 333 ASSERT_TRUE(db_thread.Start()); |
334 | 334 |
335 scoped_ptr<ProtoDatabaseImpl<TestProto> > db( | 335 scoped_ptr<ProtoDatabaseImpl<TestProto>> db( |
336 new ProtoDatabaseImpl<TestProto>(db_thread.message_loop_proxy())); | 336 new ProtoDatabaseImpl<TestProto>(db_thread.task_runner())); |
337 | 337 |
338 MockDatabaseCaller caller; | 338 MockDatabaseCaller caller; |
339 EXPECT_CALL(caller, InitCallback(_)); | 339 EXPECT_CALL(caller, InitCallback(_)); |
340 db->Init(temp_dir.path(), base::Bind(&MockDatabaseCaller::InitCallback, | 340 db->Init(temp_dir.path(), base::Bind(&MockDatabaseCaller::InitCallback, |
341 base::Unretained(&caller))); | 341 base::Unretained(&caller))); |
342 | 342 |
343 db.reset(); | 343 db.reset(); |
344 | 344 |
345 base::RunLoop run_loop; | 345 base::RunLoop run_loop; |
346 db_thread.message_loop_proxy()->PostTaskAndReply( | 346 db_thread.task_runner()->PostTaskAndReply( |
347 FROM_HERE, base::Bind(base::DoNothing), run_loop.QuitClosure()); | 347 FROM_HERE, base::Bind(base::DoNothing), run_loop.QuitClosure()); |
348 run_loop.Run(); | 348 run_loop.Run(); |
349 } | 349 } |
350 | 350 |
351 // Test that the LevelDB properly saves entries and that load returns the saved | 351 // Test that the LevelDB properly saves entries and that load returns the saved |
352 // entries. If |close_after_save| is true, the database will be closed after | 352 // entries. If |close_after_save| is true, the database will be closed after |
353 // saving and then re-opened to ensure that the data is properly persisted. | 353 // saving and then re-opened to ensure that the data is properly persisted. |
354 void TestLevelDBSaveAndLoad(bool close_after_save) { | 354 void TestLevelDBSaveAndLoad(bool close_after_save) { |
355 ScopedTempDir temp_dir; | 355 ScopedTempDir temp_dir; |
356 ASSERT_TRUE(temp_dir.CreateUniqueTempDir()); | 356 ASSERT_TRUE(temp_dir.CreateUniqueTempDir()); |
(...skipping 49 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
406 KeyValueVector save_entries; | 406 KeyValueVector save_entries; |
407 std::vector<std::string> load_entries; | 407 std::vector<std::string> load_entries; |
408 KeyVector remove_keys; | 408 KeyVector remove_keys; |
409 | 409 |
410 EXPECT_FALSE(db->InitWithOptions(temp_dir.path(), options)); | 410 EXPECT_FALSE(db->InitWithOptions(temp_dir.path(), options)); |
411 EXPECT_FALSE(db->Load(&load_entries)); | 411 EXPECT_FALSE(db->Load(&load_entries)); |
412 EXPECT_FALSE(db->Save(save_entries, remove_keys)); | 412 EXPECT_FALSE(db->Save(save_entries, remove_keys)); |
413 } | 413 } |
414 | 414 |
415 } // namespace leveldb_proto | 415 } // namespace leveldb_proto |
OLD | NEW |