Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(1557)

Unified Diff: chrome/browser/value_store/leveldb_value_store_unittest.cc

Issue 175853002: Revert of Add a Restore() method to ValueStore and make StorageAPI use it (Closed) Base URL: https://chromium.googlesource.com/chromium/src.git@master
Patch Set: Created 6 years, 10 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View side-by-side diff with in-line comments
Download patch
« no previous file with comments | « chrome/browser/value_store/leveldb_value_store.cc ('k') | chrome/browser/value_store/testing_value_store.h » ('j') | no next file with comments »
Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
Index: chrome/browser/value_store/leveldb_value_store_unittest.cc
diff --git a/chrome/browser/value_store/leveldb_value_store_unittest.cc b/chrome/browser/value_store/leveldb_value_store_unittest.cc
index 67fc3821cffcc69676497c7f360672d6241fde6a..7a7e82ffd3474316e53ffa3eab9757e3f08abe82 100644
--- a/chrome/browser/value_store/leveldb_value_store_unittest.cc
+++ b/chrome/browser/value_store/leveldb_value_store_unittest.cc
@@ -4,17 +4,8 @@
#include "chrome/browser/value_store/value_store_unittest.h"
-#include "base/file_util.h"
-#include "base/files/file_enumerator.h"
-#include "base/files/scoped_temp_dir.h"
#include "base/memory/ref_counted.h"
-#include "base/message_loop/message_loop.h"
-#include "base/values.h"
#include "chrome/browser/value_store/leveldb_value_store.h"
-#include "content/public/test/test_browser_thread_bundle.h"
-#include "testing/gtest/include/gtest/gtest.h"
-#include "third_party/leveldatabase/src/include/leveldb/db.h"
-#include "third_party/leveldatabase/src/include/leveldb/write_batch.h"
namespace {
@@ -28,157 +19,3 @@
LeveldbValueStore,
ValueStoreTest,
testing::Values(&Param));
-
-class LeveldbValueStoreUnitTest : public testing::Test {
- public:
- LeveldbValueStoreUnitTest() {}
- virtual ~LeveldbValueStoreUnitTest() {}
-
- protected:
- virtual void SetUp() OVERRIDE {
- ASSERT_TRUE(database_dir_.CreateUniqueTempDir());
- OpenStore();
- ASSERT_FALSE(store_->Get()->HasError());
- }
-
- virtual void TearDown() OVERRIDE {
- store_->Clear();
- store_.reset();
- }
-
- void CloseStore() { store_.reset(); }
-
- void OpenStore() { store_.reset(new LeveldbValueStore(database_path())); }
-
- LeveldbValueStore* store() { return store_.get(); }
- const base::FilePath& database_path() { return database_dir_.path(); }
-
- private:
- scoped_ptr<LeveldbValueStore> store_;
- base::ScopedTempDir database_dir_;
-
- content::TestBrowserThreadBundle thread_bundle_;
-};
-
-// Check that we can restore a single corrupted key in the LeveldbValueStore.
-TEST_F(LeveldbValueStoreUnitTest, RestoreKeyTest) {
- const char kNotCorruptKey[] = "not-corrupt";
- const char kValue[] = "value";
-
- // Insert a valid pair.
- scoped_ptr<base::Value> value(base::Value::CreateStringValue(kValue));
- ASSERT_FALSE(
- store()->Set(ValueStore::DEFAULTS, kNotCorruptKey, *value)->HasError());
-
- // Insert a corrupt pair.
- const char kCorruptKey[] = "corrupt";
- leveldb::WriteBatch batch;
- batch.Put(kCorruptKey, "[{(.*+\"\'\\");
- ASSERT_TRUE(store()->WriteToDbForTest(&batch));
-
- // Verify corruption.
- ValueStore::ReadResult result = store()->Get(kCorruptKey);
- ASSERT_TRUE(result->HasError());
- ASSERT_EQ(ValueStore::CORRUPTION, result->error().code);
-
- // Restore and verify.
- ASSERT_TRUE(store()->RestoreKey(kCorruptKey));
- result = store()->Get(kCorruptKey);
- EXPECT_FALSE(result->HasError());
- EXPECT_TRUE(result->settings().empty());
-
- // Verify that the valid pair is still present.
- result = store()->Get(kNotCorruptKey);
- EXPECT_FALSE(result->HasError());
- EXPECT_TRUE(result->settings().HasKey(kNotCorruptKey));
- std::string value_string;
- EXPECT_TRUE(result->settings().GetString(kNotCorruptKey, &value_string));
- EXPECT_EQ(kValue, value_string);
-}
-
-// Test that the Restore() method does not just delete the entire database
-// (unless absolutely necessary), and instead only removes corrupted keys.
-TEST_F(LeveldbValueStoreUnitTest, RestoreDoesMinimumNecessary) {
- const char* kNotCorruptKeys[] = {"a", "n", "z"};
- const size_t kNotCorruptKeysSize = 3u;
- const char kCorruptKey1[] = "f";
- const char kCorruptKey2[] = "s";
- const char kValue[] = "value";
- const char kCorruptValue[] = "[{(.*+\"\'\\";
-
- // Insert a collection of non-corrupted pairs.
- scoped_ptr<base::Value> value(base::Value::CreateStringValue(kValue));
- for (size_t i = 0; i < kNotCorruptKeysSize; ++i) {
- ASSERT_FALSE(store()
- ->Set(ValueStore::DEFAULTS, kNotCorruptKeys[i], *value)
- ->HasError());
- }
-
- // Insert a few corrupted pairs.
- leveldb::WriteBatch batch;
- batch.Put(kCorruptKey1, kCorruptValue);
- batch.Put(kCorruptKey2, kCorruptValue);
- ASSERT_TRUE(store()->WriteToDbForTest(&batch));
-
- // Verify that we broke it, and then fix it.
- ValueStore::ReadResult result = store()->Get();
- ASSERT_TRUE(result->HasError());
- ASSERT_EQ(ValueStore::CORRUPTION, result->error().code);
-
- ASSERT_TRUE(store()->Restore());
-
- // We should still have all valid pairs present in the database.
- std::string value_string;
- for (size_t i = 0; i < kNotCorruptKeysSize; ++i) {
- result = store()->Get(kNotCorruptKeys[i]);
- EXPECT_FALSE(result->HasError());
- EXPECT_TRUE(result->settings().HasKey(kNotCorruptKeys[i]));
- EXPECT_TRUE(
- result->settings().GetString(kNotCorruptKeys[i], &value_string));
- EXPECT_EQ(kValue, value_string);
- }
-}
-
-// Test that the LeveldbValueStore can recover in the case of a CATastrophic
-// failure and we have total corruption. In this case, the database is plagued
-// by LolCats.
-// Full corruption has been known to happen occasionally in strange edge cases,
-// such as after users use Windows Restore. We can't prevent it, but we need to
-// be able to handle it smoothly.
-TEST_F(LeveldbValueStoreUnitTest, RestoreFullDatabase) {
- const std::string kLolCats("I can haz leveldb filez?");
- const char* kNotCorruptKeys[] = {"a", "n", "z"};
- const size_t kNotCorruptKeysSize = 3u;
- const char kValue[] = "value";
-
- // Generate a database.
- scoped_ptr<base::Value> value(base::Value::CreateStringValue(kValue));
- for (size_t i = 0; i < kNotCorruptKeysSize; ++i) {
- ASSERT_FALSE(store()
- ->Set(ValueStore::DEFAULTS, kNotCorruptKeys[i], *value)
- ->HasError());
- }
-
- // Close it (so we remove the lock), and replace all files with LolCats.
- CloseStore();
- base::FileEnumerator enumerator(
- database_path(), true /* recursive */, base::FileEnumerator::FILES);
- for (base::FilePath file = enumerator.Next(); !file.empty();
- file = enumerator.Next()) {
- // WriteFile() failure is a result of -1.
- ASSERT_NE(file_util::WriteFile(file, kLolCats.c_str(), kLolCats.length()),
- -1);
- }
- OpenStore();
-
- // We should definitely have an error.
- ValueStore::ReadResult result = store()->Get();
- ASSERT_TRUE(result->HasError());
- ASSERT_EQ(ValueStore::CORRUPTION, result->error().code);
-
- ASSERT_TRUE(store()->Restore());
- result = store()->Get();
- EXPECT_FALSE(result->HasError());
- // We couldn't recover anything, but we should be in a sane state again.
- EXPECT_EQ(0u, result->settings().size());
-}
« no previous file with comments | « chrome/browser/value_store/leveldb_value_store.cc ('k') | chrome/browser/value_store/testing_value_store.h » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698