Index: components/leveldb/leveldb_service_unittest.cc |
diff --git a/components/leveldb/leveldb_service_unittest.cc b/components/leveldb/leveldb_service_unittest.cc |
index 893ea6f6d160db76eb4427beb25fc53654201b79..88b1050a54d2594ceb2d673c702efecf359e2ff0 100644 |
--- a/components/leveldb/leveldb_service_unittest.cc |
+++ b/components/leveldb/leveldb_service_unittest.cc |
@@ -4,6 +4,7 @@ |
#include "base/bind.h" |
#include "base/macros.h" |
+#include "base/run_loop.h" |
#include "components/filesystem/public/interfaces/directory.mojom.h" |
#include "components/filesystem/public/interfaces/file_system.mojom.h" |
#include "components/filesystem/public/interfaces/types.mojom.h" |
@@ -21,18 +22,72 @@ namespace { |
template <typename... Args> void IgnoreAllArgs(Args&&...) {} |
template <typename... Args> |
-void DoCaptures(Args*... out_args, Args... in_args) { |
+void DoCaptures(Args*... out_args, |
+ const base::Closure& quit_closure, |
+ Args... in_args) { |
IgnoreAllArgs((*out_args = std::move(in_args))...); |
+ quit_closure.Run(); |
} |
template <typename T1> |
-base::Callback<void(T1)> Capture(T1* t1) { |
- return base::Bind(&DoCaptures<T1>, t1); |
+base::Callback<void(T1)> Capture(T1* t1, const base::Closure& quit_closure) { |
+ return base::Bind(&DoCaptures<T1>, t1, quit_closure); |
} |
template <typename T1, typename T2> |
-base::Callback<void(T1, T2)> Capture(T1* t1, T2* t2) { |
- return base::Bind(&DoCaptures<T1, T2>, t1, t2); |
+base::Callback<void(T1, T2)> Capture(T1* t1, |
+ T2* t2, |
+ const base::Closure& quit_closure) { |
+ return base::Bind(&DoCaptures<T1, T2>, t1, t2, quit_closure); |
+} |
+ |
+void DatabaseSyncPut(mojom::LevelDBDatabase* database, |
+ mojo::Array<uint8_t> key, |
+ mojo::Array<uint8_t> value, |
+ mojom::DatabaseError* out_error) { |
+ base::RunLoop run_loop; |
+ database->Put(std::move(key), std::move(value), |
+ Capture(out_error, run_loop.QuitClosure())); |
+ run_loop.Run(); |
+} |
+ |
+void DatabaseSyncGet(mojom::LevelDBDatabase* database, |
+ mojo::Array<uint8_t> key, |
+ mojom::DatabaseError* out_error, |
+ mojo::Array<uint8_t>* out_value) { |
+ base::RunLoop run_loop; |
+ database->Get(std::move(key), |
+ Capture(out_error, out_value, run_loop.QuitClosure())); |
+ run_loop.Run(); |
+} |
+ |
+void DatabaseSyncGetPrefixed(mojom::LevelDBDatabase* database, |
+ mojo::Array<uint8_t> key_prefix, |
+ mojom::DatabaseError* out_error, |
+ mojo::Array<mojom::KeyValuePtr>* out_key_values) { |
+ base::RunLoop run_loop; |
+ database->GetPrefixed( |
+ std::move(key_prefix), |
+ Capture(out_error, out_key_values, run_loop.QuitClosure())); |
+ run_loop.Run(); |
+} |
+ |
+void DatabaseSyncDeletePrefixed(mojom::LevelDBDatabase* database, |
+ mojo::Array<uint8_t> key_prefix, |
+ mojom::DatabaseError* out_error) { |
+ base::RunLoop run_loop; |
+ database->DeletePrefixed(std::move(key_prefix), |
+ Capture(out_error, run_loop.QuitClosure())); |
+ run_loop.Run(); |
+} |
+ |
+void LevelDBSyncOpenInMemory(mojom::LevelDBService* leveldb, |
+ mojom::LevelDBDatabaseRequest database, |
+ mojom::DatabaseError* out_error) { |
+ base::RunLoop run_loop; |
+ leveldb->OpenInMemory(std::move(database), |
+ Capture(out_error, run_loop.QuitClosure())); |
+ run_loop.Run(); |
} |
class LevelDBServiceTest : public shell::test::ShellTest { |
@@ -58,8 +113,8 @@ class LevelDBServiceTest : public shell::test::ShellTest { |
// since |ASSERT_...()| doesn't work with return values. |
void GetTempDirectory(filesystem::mojom::DirectoryPtr* directory) { |
FileError error = FileError::FAILED; |
- files()->OpenTempDirectory(GetProxy(directory), Capture(&error)); |
- ASSERT_TRUE(files().WaitForIncomingResponse()); |
+ bool handled = files()->OpenTempDirectory(GetProxy(directory), &error); |
+ ASSERT_TRUE(handled); |
ASSERT_EQ(FileError::OK, error); |
} |
@@ -76,40 +131,39 @@ class LevelDBServiceTest : public shell::test::ShellTest { |
TEST_F(LevelDBServiceTest, Basic) { |
mojom::DatabaseError error; |
mojom::LevelDBDatabasePtr database; |
- leveldb()->OpenInMemory(GetProxy(&database), Capture(&error)); |
- ASSERT_TRUE(leveldb().WaitForIncomingResponse()); |
+ LevelDBSyncOpenInMemory(leveldb().get(), GetProxy(&database), &error); |
EXPECT_EQ(mojom::DatabaseError::OK, error); |
// Write a key to the database. |
error = mojom::DatabaseError::INVALID_ARGUMENT; |
- database->Put(mojo::Array<uint8_t>::From(std::string("key")), |
- mojo::Array<uint8_t>::From(std::string("value")), |
- Capture(&error)); |
- ASSERT_TRUE(database.WaitForIncomingResponse()); |
+ DatabaseSyncPut(database.get(), |
+ mojo::Array<uint8_t>::From(std::string("key")), |
+ mojo::Array<uint8_t>::From(std::string("value")), &error); |
EXPECT_EQ(mojom::DatabaseError::OK, error); |
// Read the key back from the database. |
error = mojom::DatabaseError::INVALID_ARGUMENT; |
mojo::Array<uint8_t> value; |
- database->Get(mojo::Array<uint8_t>::From(std::string("key")), |
- Capture(&error, &value)); |
- ASSERT_TRUE(database.WaitForIncomingResponse()); |
+ DatabaseSyncGet(database.get(), |
+ mojo::Array<uint8_t>::From(std::string("key")), &error, |
+ &value); |
EXPECT_EQ(mojom::DatabaseError::OK, error); |
EXPECT_EQ("value", value.To<std::string>()); |
// Delete the key from the database. |
error = mojom::DatabaseError::INVALID_ARGUMENT; |
+ base::RunLoop run_loop; |
database->Delete(mojo::Array<uint8_t>::From(std::string("key")), |
- Capture(&error)); |
- ASSERT_TRUE(database.WaitForIncomingResponse()); |
+ Capture(&error, run_loop.QuitClosure())); |
+ run_loop.Run(); |
EXPECT_EQ(mojom::DatabaseError::OK, error); |
// Read the key back from the database. |
error = mojom::DatabaseError::INVALID_ARGUMENT; |
value.SetToEmpty(); |
- database->Get(mojo::Array<uint8_t>::From(std::string("key")), |
- Capture(&error, &value)); |
- ASSERT_TRUE(database.WaitForIncomingResponse()); |
+ DatabaseSyncGet(database.get(), |
+ mojo::Array<uint8_t>::From(std::string("key")), &error, |
+ &value); |
EXPECT_EQ(mojom::DatabaseError::NOT_FOUND, error); |
EXPECT_EQ("", value.To<std::string>()); |
} |
@@ -117,15 +171,13 @@ TEST_F(LevelDBServiceTest, Basic) { |
TEST_F(LevelDBServiceTest, WriteBatch) { |
mojom::DatabaseError error; |
mojom::LevelDBDatabasePtr database; |
- leveldb()->OpenInMemory(GetProxy(&database), Capture(&error)); |
- ASSERT_TRUE(leveldb().WaitForIncomingResponse()); |
+ LevelDBSyncOpenInMemory(leveldb().get(), GetProxy(&database), &error); |
EXPECT_EQ(mojom::DatabaseError::OK, error); |
// Write a key to the database. |
- database->Put(mojo::Array<uint8_t>::From(std::string("key")), |
- mojo::Array<uint8_t>::From(std::string("value")), |
- Capture(&error)); |
- ASSERT_TRUE(database.WaitForIncomingResponse()); |
+ DatabaseSyncPut(database.get(), |
+ mojo::Array<uint8_t>::From(std::string("key")), |
+ mojo::Array<uint8_t>::From(std::string("value")), &error); |
EXPECT_EQ(mojom::DatabaseError::OK, error); |
// Create a batched operation which both deletes "key" and adds another write. |
@@ -141,37 +193,37 @@ TEST_F(LevelDBServiceTest, WriteBatch) { |
item->value = mojo::Array<uint8_t>::From(std::string("more")); |
operations.push_back(std::move(item)); |
- database->Write(std::move(operations), Capture(&error)); |
- ASSERT_TRUE(database.WaitForIncomingResponse()); |
+ base::RunLoop run_loop; |
+ database->Write(std::move(operations), |
+ Capture(&error, run_loop.QuitClosure())); |
+ run_loop.Run(); |
EXPECT_EQ(mojom::DatabaseError::OK, error); |
// Reading "key" should be invalid now. |
error = mojom::DatabaseError::INVALID_ARGUMENT; |
mojo::Array<uint8_t> value; |
- database->Get(mojo::Array<uint8_t>::From(std::string("key")), |
- Capture(&error, &value)); |
- ASSERT_TRUE(database.WaitForIncomingResponse()); |
+ DatabaseSyncGet(database.get(), |
+ mojo::Array<uint8_t>::From(std::string("key")), &error, |
+ &value); |
EXPECT_EQ(mojom::DatabaseError::NOT_FOUND, error); |
EXPECT_EQ("", value.To<std::string>()); |
// Reading "other" should return "more" |
error = mojom::DatabaseError::INVALID_ARGUMENT; |
- database->Get(mojo::Array<uint8_t>::From(std::string("other")), |
- Capture(&error, &value)); |
- ASSERT_TRUE(database.WaitForIncomingResponse()); |
+ DatabaseSyncGet(database.get(), |
+ mojo::Array<uint8_t>::From(std::string("other")), &error, |
+ &value); |
EXPECT_EQ(mojom::DatabaseError::OK, error); |
EXPECT_EQ("more", value.To<std::string>()); |
// Write a some prefixed keys to the database. |
- database->Put(mojo::Array<uint8_t>::From(std::string("prefix-key1")), |
- mojo::Array<uint8_t>::From(std::string("value")), |
- Capture(&error)); |
- ASSERT_TRUE(database.WaitForIncomingResponse()); |
+ DatabaseSyncPut(database.get(), |
+ mojo::Array<uint8_t>::From(std::string("prefix-key1")), |
+ mojo::Array<uint8_t>::From(std::string("value")), &error); |
EXPECT_EQ(mojom::DatabaseError::OK, error); |
- database->Put(mojo::Array<uint8_t>::From(std::string("prefix-key2")), |
- mojo::Array<uint8_t>::From(std::string("value")), |
- Capture(&error)); |
- ASSERT_TRUE(database.WaitForIncomingResponse()); |
+ DatabaseSyncPut(database.get(), |
+ mojo::Array<uint8_t>::From(std::string("prefix-key2")), |
+ mojo::Array<uint8_t>::From(std::string("value")), &error); |
EXPECT_EQ(mojom::DatabaseError::OK, error); |
// Create a batched operation to delete them. |
@@ -180,24 +232,26 @@ TEST_F(LevelDBServiceTest, WriteBatch) { |
item->type = mojom::BatchOperationType::DELETE_PREFIXED_KEY; |
item->key = mojo::Array<uint8_t>::From(std::string("prefix")); |
operations.push_back(std::move(item)); |
- database->Write(std::move(operations), Capture(&error)); |
- ASSERT_TRUE(database.WaitForIncomingResponse()); |
+ base::RunLoop run_loop2; |
+ database->Write(std::move(operations), |
+ Capture(&error, run_loop2.QuitClosure())); |
+ run_loop2.Run(); |
EXPECT_EQ(mojom::DatabaseError::OK, error); |
// Reading all "prefix" keys should be invalid now. |
error = mojom::DatabaseError::INVALID_ARGUMENT; |
value = nullptr; |
- database->Get(mojo::Array<uint8_t>::From(std::string("prefix-key1")), |
- Capture(&error, &value)); |
- ASSERT_TRUE(database.WaitForIncomingResponse()); |
+ DatabaseSyncGet(database.get(), |
+ mojo::Array<uint8_t>::From(std::string("prefix-key1")), |
+ &error, &value); |
EXPECT_EQ(mojom::DatabaseError::NOT_FOUND, error); |
EXPECT_EQ("", value.To<std::string>()); |
// Reading "key" should be invalid now. |
error = mojom::DatabaseError::INVALID_ARGUMENT; |
value = nullptr; |
- database->Get(mojo::Array<uint8_t>::From(std::string("prefix-key2")), |
- Capture(&error, &value)); |
- ASSERT_TRUE(database.WaitForIncomingResponse()); |
+ DatabaseSyncGet(database.get(), |
+ mojo::Array<uint8_t>::From(std::string("prefix-key2")), |
+ &error, &value); |
EXPECT_EQ(mojom::DatabaseError::NOT_FOUND, error); |
EXPECT_EQ("", value.To<std::string>()); |
} |
@@ -216,19 +270,18 @@ TEST_F(LevelDBServiceTest, Reconnect) { |
leveldb::mojom::OpenOptionsPtr options = leveldb::mojom::OpenOptions::New(); |
options->error_if_exists = true; |
options->create_if_missing = true; |
- leveldb()->OpenWithOptions(std::move(options), |
- std::move(directory), "test", |
+ base::RunLoop run_loop; |
+ leveldb()->OpenWithOptions(std::move(options), std::move(directory), "test", |
GetProxy(&database), |
- Capture(&error)); |
- ASSERT_TRUE(leveldb().WaitForIncomingResponse()); |
+ Capture(&error, run_loop.QuitClosure())); |
+ run_loop.Run(); |
EXPECT_EQ(mojom::DatabaseError::OK, error); |
// Write a key to the database. |
error = mojom::DatabaseError::INVALID_ARGUMENT; |
- database->Put(mojo::Array<uint8_t>::From(std::string("key")), |
- mojo::Array<uint8_t>::From(std::string("value")), |
- Capture(&error)); |
- ASSERT_TRUE(database.WaitForIncomingResponse()); |
+ DatabaseSyncPut(database.get(), |
+ mojo::Array<uint8_t>::From(std::string("key")), |
+ mojo::Array<uint8_t>::From(std::string("value")), &error); |
EXPECT_EQ(mojom::DatabaseError::OK, error); |
// The database should go out of scope here. |
@@ -240,17 +293,18 @@ TEST_F(LevelDBServiceTest, Reconnect) { |
// Reconnect to the database. |
mojom::LevelDBDatabasePtr database; |
+ base::RunLoop run_loop; |
leveldb()->Open(std::move(directory), "test", GetProxy(&database), |
- Capture(&error)); |
- ASSERT_TRUE(leveldb().WaitForIncomingResponse()); |
+ Capture(&error, run_loop.QuitClosure())); |
+ run_loop.Run(); |
EXPECT_EQ(mojom::DatabaseError::OK, error); |
// We should still be able to read the key back from the database. |
error = mojom::DatabaseError::INVALID_ARGUMENT; |
mojo::Array<uint8_t> value; |
- database->Get(mojo::Array<uint8_t>::From(std::string("key")), |
- Capture(&error, &value)); |
- ASSERT_TRUE(database.WaitForIncomingResponse()); |
+ DatabaseSyncGet(database.get(), |
+ mojo::Array<uint8_t>::From(std::string("key")), &error, |
+ &value); |
EXPECT_EQ(mojom::DatabaseError::OK, error); |
EXPECT_EQ("value", value.To<std::string>()); |
} |
@@ -259,61 +313,59 @@ TEST_F(LevelDBServiceTest, Reconnect) { |
TEST_F(LevelDBServiceTest, GetSnapshotSimple) { |
mojom::DatabaseError error; |
mojom::LevelDBDatabasePtr database; |
- leveldb()->OpenInMemory(GetProxy(&database), Capture(&error)); |
- ASSERT_TRUE(leveldb().WaitForIncomingResponse()); |
+ LevelDBSyncOpenInMemory(leveldb().get(), GetProxy(&database), &error); |
EXPECT_EQ(mojom::DatabaseError::OK, error); |
uint64_t snapshot_id = 0; |
- database->GetSnapshot(Capture(&snapshot_id)); |
- ASSERT_TRUE(database.WaitForIncomingResponse()); |
+ base::RunLoop run_loop; |
+ database->GetSnapshot(Capture(&snapshot_id, run_loop.QuitClosure())); |
+ run_loop.Run(); |
EXPECT_NE(static_cast<uint64_t>(0), snapshot_id); |
} |
TEST_F(LevelDBServiceTest, GetFromSnapshots) { |
mojom::DatabaseError error; |
mojom::LevelDBDatabasePtr database; |
- leveldb()->OpenInMemory(GetProxy(&database), Capture(&error)); |
- ASSERT_TRUE(leveldb().WaitForIncomingResponse()); |
+ LevelDBSyncOpenInMemory(leveldb().get(), GetProxy(&database), &error); |
EXPECT_EQ(mojom::DatabaseError::OK, error); |
// Write a key to the database. |
error = mojom::DatabaseError::INVALID_ARGUMENT; |
- database->Put(mojo::Array<uint8_t>::From(std::string("key")), |
- mojo::Array<uint8_t>::From(std::string("value")), |
- Capture(&error)); |
- ASSERT_TRUE(database.WaitForIncomingResponse()); |
+ DatabaseSyncPut(database.get(), |
+ mojo::Array<uint8_t>::From(std::string("key")), |
+ mojo::Array<uint8_t>::From(std::string("value")), &error); |
EXPECT_EQ(mojom::DatabaseError::OK, error); |
// Take a snapshot where key=value. |
uint64_t key_value_snapshot = 0; |
- database->GetSnapshot(Capture(&key_value_snapshot)); |
- ASSERT_TRUE(database.WaitForIncomingResponse()); |
+ base::RunLoop run_loop; |
+ database->GetSnapshot(Capture(&key_value_snapshot, run_loop.QuitClosure())); |
+ run_loop.Run(); |
// Change key to "yek". |
error = mojom::DatabaseError::INVALID_ARGUMENT; |
- database->Put(mojo::Array<uint8_t>::From(std::string("key")), |
- mojo::Array<uint8_t>::From(std::string("yek")), |
- Capture(&error)); |
- ASSERT_TRUE(database.WaitForIncomingResponse()); |
+ DatabaseSyncPut(database.get(), |
+ mojo::Array<uint8_t>::From(std::string("key")), |
+ mojo::Array<uint8_t>::From(std::string("yek")), &error); |
EXPECT_EQ(mojom::DatabaseError::OK, error); |
// (Ensure this change is live on the database.) |
error = mojom::DatabaseError::INVALID_ARGUMENT; |
mojo::Array<uint8_t> value; |
- database->Get(mojo::Array<uint8_t>::From(std::string("key")), |
- Capture(&error, &value)); |
- ASSERT_TRUE(database.WaitForIncomingResponse()); |
+ DatabaseSyncGet(database.get(), |
+ mojo::Array<uint8_t>::From(std::string("key")), &error, |
+ &value); |
EXPECT_EQ(mojom::DatabaseError::OK, error); |
EXPECT_EQ("yek", value.To<std::string>()); |
// But if we were to read from the snapshot, we'd still get value. |
error = mojom::DatabaseError::INVALID_ARGUMENT; |
value.SetToEmpty(); |
- database->GetFromSnapshot( |
- key_value_snapshot, |
- mojo::Array<uint8_t>::From(std::string("key")), |
- Capture(&error, &value)); |
- ASSERT_TRUE(database.WaitForIncomingResponse()); |
+ base::RunLoop run_loop2; |
+ database->GetFromSnapshot(key_value_snapshot, |
+ mojo::Array<uint8_t>::From(std::string("key")), |
+ Capture(&error, &value, run_loop2.QuitClosure())); |
+ run_loop2.Run(); |
EXPECT_EQ(mojom::DatabaseError::OK, error); |
EXPECT_EQ("value", value.To<std::string>()); |
} |
@@ -321,59 +373,57 @@ TEST_F(LevelDBServiceTest, GetFromSnapshots) { |
TEST_F(LevelDBServiceTest, InvalidArgumentOnInvalidSnapshot) { |
mojom::LevelDBDatabasePtr database; |
mojom::DatabaseError error = mojom::DatabaseError::INVALID_ARGUMENT; |
- leveldb()->OpenInMemory(GetProxy(&database), Capture(&error)); |
- ASSERT_TRUE(leveldb().WaitForIncomingResponse()); |
+ LevelDBSyncOpenInMemory(leveldb().get(), GetProxy(&database), &error); |
EXPECT_EQ(mojom::DatabaseError::OK, error); |
uint64_t invalid_snapshot = 8; |
error = mojom::DatabaseError::OK; |
mojo::Array<uint8_t> value; |
- database->GetFromSnapshot( |
- invalid_snapshot, |
- mojo::Array<uint8_t>::From(std::string("key")), |
- Capture(&error, &value)); |
- ASSERT_TRUE(database.WaitForIncomingResponse()); |
+ base::RunLoop run_loop; |
+ database->GetFromSnapshot(invalid_snapshot, |
+ mojo::Array<uint8_t>::From(std::string("key")), |
+ Capture(&error, &value, run_loop.QuitClosure())); |
+ run_loop.Run(); |
EXPECT_EQ(mojom::DatabaseError::INVALID_ARGUMENT, error); |
} |
TEST_F(LevelDBServiceTest, MemoryDBReadWrite) { |
mojom::LevelDBDatabasePtr database; |
mojom::DatabaseError error = mojom::DatabaseError::INVALID_ARGUMENT; |
- leveldb()->OpenInMemory(GetProxy(&database), Capture(&error)); |
- ASSERT_TRUE(leveldb().WaitForIncomingResponse()); |
+ LevelDBSyncOpenInMemory(leveldb().get(), GetProxy(&database), &error); |
EXPECT_EQ(mojom::DatabaseError::OK, error); |
// Write a key to the database. |
error = mojom::DatabaseError::INVALID_ARGUMENT; |
- database->Put(mojo::Array<uint8_t>::From(std::string("key")), |
- mojo::Array<uint8_t>::From(std::string("value")), |
- Capture(&error)); |
- ASSERT_TRUE(database.WaitForIncomingResponse()); |
+ DatabaseSyncPut(database.get(), |
+ mojo::Array<uint8_t>::From(std::string("key")), |
+ mojo::Array<uint8_t>::From(std::string("value")), &error); |
EXPECT_EQ(mojom::DatabaseError::OK, error); |
// Read the key back from the database. |
error = mojom::DatabaseError::INVALID_ARGUMENT; |
mojo::Array<uint8_t> value; |
- database->Get(mojo::Array<uint8_t>::From(std::string("key")), |
- Capture(&error, &value)); |
- ASSERT_TRUE(database.WaitForIncomingResponse()); |
+ DatabaseSyncGet(database.get(), |
+ mojo::Array<uint8_t>::From(std::string("key")), &error, |
+ &value); |
EXPECT_EQ(mojom::DatabaseError::OK, error); |
EXPECT_EQ("value", value.To<std::string>()); |
// Delete the key from the database. |
error = mojom::DatabaseError::INVALID_ARGUMENT; |
+ base::RunLoop run_loop; |
database->Delete(mojo::Array<uint8_t>::From(std::string("key")), |
- Capture(&error)); |
- ASSERT_TRUE(database.WaitForIncomingResponse()); |
+ Capture(&error, run_loop.QuitClosure())); |
+ run_loop.Run(); |
EXPECT_EQ(mojom::DatabaseError::OK, error); |
// Read the key back from the database. |
error = mojom::DatabaseError::INVALID_ARGUMENT; |
value.SetToEmpty(); |
- database->Get(mojo::Array<uint8_t>::From(std::string("key")), |
- Capture(&error, &value)); |
- ASSERT_TRUE(database.WaitForIncomingResponse()); |
+ DatabaseSyncGet(database.get(), |
+ mojo::Array<uint8_t>::From(std::string("key")), &error, |
+ &value); |
EXPECT_EQ(mojom::DatabaseError::NOT_FOUND, error); |
EXPECT_EQ("", value.To<std::string>()); |
} |
@@ -382,8 +432,7 @@ TEST_F(LevelDBServiceTest, Prefixed) { |
// Open an in memory database for speed. |
mojom::DatabaseError error = mojom::DatabaseError::INVALID_ARGUMENT; |
mojom::LevelDBDatabasePtr database; |
- leveldb()->OpenInMemory(GetProxy(&database), Capture(&error)); |
- ASSERT_TRUE(leveldb().WaitForIncomingResponse()); |
+ LevelDBSyncOpenInMemory(leveldb().get(), GetProxy(&database), &error); |
EXPECT_EQ(mojom::DatabaseError::OK, error); |
const std::string prefix("prefix"); |
@@ -391,60 +440,50 @@ TEST_F(LevelDBServiceTest, Prefixed) { |
// Completely empty database. |
error = mojom::DatabaseError::INVALID_ARGUMENT; |
- database->GetPrefixed(mojo::Array<uint8_t>::From(prefix), |
- Capture(&error, &key_values)); |
- ASSERT_TRUE(database.WaitForIncomingResponse()); |
+ DatabaseSyncGetPrefixed(database.get(), mojo::Array<uint8_t>::From(prefix), |
+ &error, &key_values); |
EXPECT_EQ(mojom::DatabaseError::OK, error); |
EXPECT_TRUE(key_values.empty()); |
// No values with our prefix, but values before and after. |
error = mojom::DatabaseError::INVALID_ARGUMENT; |
- database->Put(mojo::Array<uint8_t>::From(std::string("a-before-prefix")), |
- mojo::Array<uint8_t>::From(std::string("value")), |
- Capture(&error)); |
- ASSERT_TRUE(database.WaitForIncomingResponse()); |
+ DatabaseSyncPut(database.get(), |
+ mojo::Array<uint8_t>::From(std::string("a-before-prefix")), |
+ mojo::Array<uint8_t>::From(std::string("value")), &error); |
EXPECT_EQ(mojom::DatabaseError::OK, error); |
error = mojom::DatabaseError::INVALID_ARGUMENT; |
- database->Put(mojo::Array<uint8_t>::From(std::string("z-after-prefix")), |
- mojo::Array<uint8_t>::From(std::string("value")), |
- Capture(&error)); |
- ASSERT_TRUE(database.WaitForIncomingResponse()); |
+ DatabaseSyncPut(database.get(), |
+ mojo::Array<uint8_t>::From(std::string("z-after-prefix")), |
+ mojo::Array<uint8_t>::From(std::string("value")), &error); |
EXPECT_EQ(mojom::DatabaseError::OK, error); |
key_values.SetToEmpty(); |
error = mojom::DatabaseError::INVALID_ARGUMENT; |
- database->GetPrefixed(mojo::Array<uint8_t>::From(prefix), |
- Capture(&error, &key_values)); |
- ASSERT_TRUE(database.WaitForIncomingResponse()); |
+ DatabaseSyncGetPrefixed(database.get(), mojo::Array<uint8_t>::From(prefix), |
+ &error, &key_values); |
EXPECT_EQ(mojom::DatabaseError::OK, error); |
EXPECT_TRUE(key_values.empty()); |
// One value with the exact prefix. |
- database->Put(mojo::Array<uint8_t>::From(prefix), |
- mojo::Array<uint8_t>::From(std::string("value")), |
- Capture(&error)); |
- ASSERT_TRUE(database.WaitForIncomingResponse()); |
+ DatabaseSyncPut(database.get(), mojo::Array<uint8_t>::From(prefix), |
+ mojo::Array<uint8_t>::From(std::string("value")), &error); |
EXPECT_EQ(mojom::DatabaseError::OK, error); |
error = mojom::DatabaseError::INVALID_ARGUMENT; |
key_values.SetToEmpty(); |
- database->GetPrefixed(mojo::Array<uint8_t>::From(prefix), |
- Capture(&error, &key_values)); |
- ASSERT_TRUE(database.WaitForIncomingResponse()); |
+ DatabaseSyncGetPrefixed(database.get(), mojo::Array<uint8_t>::From(prefix), |
+ &error, &key_values); |
EXPECT_EQ(mojom::DatabaseError::OK, error); |
EXPECT_EQ(1u, key_values.size()); |
EXPECT_EQ("prefix", key_values[0]->key.To<std::string>()); |
EXPECT_EQ("value", key_values[0]->value.To<std::string>()); |
// Multiple values with starting with the prefix. |
- database->Put(mojo::Array<uint8_t>::From(prefix + "2"), |
- mojo::Array<uint8_t>::From(std::string("value2")), |
- Capture(&error)); |
- ASSERT_TRUE(database.WaitForIncomingResponse()); |
+ DatabaseSyncPut(database.get(), mojo::Array<uint8_t>::From(prefix + "2"), |
+ mojo::Array<uint8_t>::From(std::string("value2")), &error); |
EXPECT_EQ(mojom::DatabaseError::OK, error); |
error = mojom::DatabaseError::INVALID_ARGUMENT; |
key_values.SetToEmpty(); |
- database->GetPrefixed(mojo::Array<uint8_t>::From(prefix), |
- Capture(&error, &key_values)); |
- ASSERT_TRUE(database.WaitForIncomingResponse()); |
+ DatabaseSyncGetPrefixed(database.get(), mojo::Array<uint8_t>::From(prefix), |
+ &error, &key_values); |
EXPECT_EQ(mojom::DatabaseError::OK, error); |
EXPECT_EQ(2u, key_values.size()); |
EXPECT_EQ("prefix", key_values[0]->key.To<std::string>()); |
@@ -454,29 +493,27 @@ TEST_F(LevelDBServiceTest, Prefixed) { |
// Delete the prefixed values. |
error = mojom::DatabaseError::INVALID_ARGUMENT; |
- database->DeletePrefixed(mojo::Array<uint8_t>::From(prefix), |
- Capture(&error)); |
- ASSERT_TRUE(database.WaitForIncomingResponse()); |
+ DatabaseSyncDeletePrefixed(database.get(), mojo::Array<uint8_t>::From(prefix), |
+ &error); |
EXPECT_EQ(mojom::DatabaseError::OK, error); |
error = mojom::DatabaseError::INVALID_ARGUMENT; |
key_values.SetToEmpty(); |
- database->GetPrefixed(mojo::Array<uint8_t>::From(prefix), |
- Capture(&error, &key_values)); |
- ASSERT_TRUE(database.WaitForIncomingResponse()); |
+ DatabaseSyncGetPrefixed(database.get(), mojo::Array<uint8_t>::From(prefix), |
+ &error, &key_values); |
EXPECT_EQ(mojom::DatabaseError::OK, error); |
EXPECT_TRUE(key_values.empty()); |
// Make sure the others are not deleted. |
mojo::Array<uint8_t> value; |
- database->Get(mojo::Array<uint8_t>::From(std::string("a-before-prefix")), |
- Capture(&error, &value)); |
- ASSERT_TRUE(database.WaitForIncomingResponse()); |
+ DatabaseSyncGet(database.get(), |
+ mojo::Array<uint8_t>::From(std::string("a-before-prefix")), |
+ &error, &value); |
EXPECT_EQ(mojom::DatabaseError::OK, error); |
EXPECT_EQ("value", value.To<std::string>()); |
value.SetToEmpty(); |
- database->Get(mojo::Array<uint8_t>::From(std::string("z-after-prefix")), |
- Capture(&error, &value)); |
- ASSERT_TRUE(database.WaitForIncomingResponse()); |
+ DatabaseSyncGet(database.get(), |
+ mojo::Array<uint8_t>::From(std::string("z-after-prefix")), |
+ &error, &value); |
EXPECT_EQ(mojom::DatabaseError::OK, error); |
EXPECT_EQ("value", value.To<std::string>()); |
@@ -484,30 +521,25 @@ TEST_F(LevelDBServiceTest, Prefixed) { |
// Even thought there is no exact matching key, GetPrefixed |
// and DeletePrefixed still operate on the values. |
error = mojom::DatabaseError::INVALID_ARGUMENT; |
- database->Put(mojo::Array<uint8_t>::From(prefix + "2"), |
- mojo::Array<uint8_t>::From(std::string("value2")), |
- Capture(&error)); |
- ASSERT_TRUE(database.WaitForIncomingResponse()); |
+ DatabaseSyncPut(database.get(), mojo::Array<uint8_t>::From(prefix + "2"), |
+ mojo::Array<uint8_t>::From(std::string("value2")), &error); |
EXPECT_EQ(mojom::DatabaseError::OK, error); |
error = mojom::DatabaseError::INVALID_ARGUMENT; |
key_values.SetToEmpty(); |
- database->GetPrefixed(mojo::Array<uint8_t>::From(prefix), |
- Capture(&error, &key_values)); |
- ASSERT_TRUE(database.WaitForIncomingResponse()); |
+ DatabaseSyncGetPrefixed(database.get(), mojo::Array<uint8_t>::From(prefix), |
+ &error, &key_values); |
EXPECT_EQ(mojom::DatabaseError::OK, error); |
EXPECT_EQ(1u, key_values.size()); |
EXPECT_EQ("prefix2", key_values[0]->key.To<std::string>()); |
EXPECT_EQ("value2", key_values[0]->value.To<std::string>()); |
error = mojom::DatabaseError::INVALID_ARGUMENT; |
- database->DeletePrefixed(mojo::Array<uint8_t>::From(prefix), |
- Capture(&error)); |
- ASSERT_TRUE(database.WaitForIncomingResponse()); |
+ DatabaseSyncDeletePrefixed(database.get(), mojo::Array<uint8_t>::From(prefix), |
+ &error); |
EXPECT_EQ(mojom::DatabaseError::OK, error); |
error = mojom::DatabaseError::INVALID_ARGUMENT; |
key_values.SetToEmpty(); |
- database->GetPrefixed(mojo::Array<uint8_t>::From(prefix), |
- Capture(&error, &key_values)); |
- ASSERT_TRUE(database.WaitForIncomingResponse()); |
+ DatabaseSyncGetPrefixed(database.get(), mojo::Array<uint8_t>::From(prefix), |
+ &error, &key_values); |
EXPECT_EQ(mojom::DatabaseError::OK, error); |
EXPECT_TRUE(key_values.empty()); |
} |