blob: c38baa3fc5f5cf5bb3baa01ecfd06d1aecfa9450 [file] [log] [blame]
// Copyright 2013 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "base/compiler_specific.h"
#include "base/memory/scoped_ptr.h"
#include "base/strings/string_number_conversions.h"
#include "base/strings/utf_string_conversions.h"
#include "base/test/test_reg_util_win.h"
#include "base/time/time.h"
#include "testing/gtest/include/gtest/gtest.h"
namespace registry_util {
namespace {
const wchar_t kTestKeyPath[] = L"Software\\Chromium\\Foo\\Baz\\TestKey";
const wchar_t kTestValueName[] = L"TestValue";
} // namespace
class RegistryOverrideManagerTest : public testing::Test {
protected:
RegistryOverrideManagerTest() {
// We assign a fake test key path to our test RegistryOverrideManager
// so we don't interfere with any actual RegistryOverrideManagers running
// on the system. This fake path will be auto-deleted by other
// RegistryOverrideManagers in case we crash.
fake_test_key_root_ = registry_util::GenerateTempKeyPath();
// Ensure a clean test environment.
base::win::RegKey key(HKEY_CURRENT_USER);
key.DeleteKey(fake_test_key_root_.c_str());
key.DeleteKey(kTestKeyPath);
}
~RegistryOverrideManagerTest() override {
base::win::RegKey key(HKEY_CURRENT_USER);
key.DeleteKey(fake_test_key_root_.c_str());
}
void AssertKeyExists(const base::string16& key_path) {
base::win::RegKey key;
ASSERT_EQ(ERROR_SUCCESS,
key.Open(HKEY_CURRENT_USER, key_path.c_str(), KEY_READ))
<< key_path << " does not exist.";
}
void AssertKeyAbsent(const base::string16& key_path) {
base::win::RegKey key;
ASSERT_NE(ERROR_SUCCESS,
key.Open(HKEY_CURRENT_USER, key_path.c_str(), KEY_READ))
<< key_path << " exists but it should not.";
}
void CreateKey(const base::string16& key_path) {
base::win::RegKey key;
EXPECT_EQ(ERROR_SUCCESS,
key.Create(HKEY_CURRENT_USER, key_path.c_str(), KEY_ALL_ACCESS));
}
base::string16 FakeOverrideManagerPath(const base::Time& time) {
return fake_test_key_root_ + L"\\" +
base::Int64ToString16(time.ToInternalValue());
}
void CreateManager(const base::Time& timestamp) {
manager_.reset(new RegistryOverrideManager(timestamp, fake_test_key_root_));
manager_->OverrideRegistry(HKEY_CURRENT_USER);
}
base::string16 fake_test_key_root_;
scoped_ptr<RegistryOverrideManager> manager_;
};
TEST_F(RegistryOverrideManagerTest, Basic) {
CreateManager(base::Time::Now());
base::win::RegKey create_key;
EXPECT_EQ(ERROR_SUCCESS,
create_key.Create(HKEY_CURRENT_USER, kTestKeyPath, KEY_ALL_ACCESS));
EXPECT_TRUE(create_key.Valid());
EXPECT_EQ(ERROR_SUCCESS, create_key.WriteValue(kTestValueName, 42));
create_key.Close();
AssertKeyExists(kTestKeyPath);
DWORD value;
base::win::RegKey read_key;
EXPECT_EQ(ERROR_SUCCESS,
read_key.Open(HKEY_CURRENT_USER, kTestKeyPath, KEY_READ));
EXPECT_TRUE(read_key.Valid());
EXPECT_EQ(ERROR_SUCCESS, read_key.ReadValueDW(kTestValueName, &value));
EXPECT_EQ(42u, value);
read_key.Close();
manager_.reset();
AssertKeyAbsent(kTestKeyPath);
}
TEST_F(RegistryOverrideManagerTest, DeleteStaleKeys) {
base::Time::Exploded kTestTimeExploded = {2013, 11, 1, 4, 0, 0, 0, 0};
base::Time kTestTime = base::Time::FromUTCExploded(kTestTimeExploded);
base::string16 path_garbage = fake_test_key_root_ + L"\\Blah";
base::string16 path_very_stale =
FakeOverrideManagerPath(kTestTime - base::TimeDelta::FromDays(100));
base::string16 path_stale =
FakeOverrideManagerPath(kTestTime - base::TimeDelta::FromDays(5));
base::string16 path_current =
FakeOverrideManagerPath(kTestTime - base::TimeDelta::FromMinutes(1));
base::string16 path_future =
FakeOverrideManagerPath(kTestTime + base::TimeDelta::FromMinutes(1));
CreateKey(path_garbage);
CreateKey(path_very_stale);
CreateKey(path_stale);
CreateKey(path_current);
CreateKey(path_future);
CreateManager(kTestTime);
manager_.reset();
AssertKeyAbsent(path_garbage);
AssertKeyAbsent(path_very_stale);
AssertKeyAbsent(path_stale);
AssertKeyExists(path_current);
AssertKeyExists(path_future);
}
} // namespace registry_util