This source file includes following definitions.
- SetUp
- TearDown
- TEST_F
- TEST_F
#include <windows.h>
#include "base/files/file_path.h"
#include "base/memory/scoped_ptr.h"
#include "base/strings/string_util.h"
#include "base/win/registry.h"
#include "chrome/installer/util/delete_reg_value_work_item.h"
#include "chrome/installer/util/work_item.h"
#include "testing/gtest/include/gtest/gtest.h"
using base::win::RegKey;
namespace {
const wchar_t kTestRoot[] = L"DeleteRegValueWorkItemTest";
const wchar_t kWriteNew[] = L"WriteNew";
const wchar_t kNameStr[] = L"name_str";
const wchar_t kNameDword[] = L"name_dword";
class DeleteRegValueWorkItemTest : public testing::Test {
protected:
virtual void SetUp() {
RegKey key(HKEY_CURRENT_USER, L"", KEY_ALL_ACCESS);
key.DeleteKey(kTestRoot);
ASSERT_NE(ERROR_SUCCESS, key.Open(HKEY_CURRENT_USER, kTestRoot, KEY_READ));
ASSERT_EQ(ERROR_SUCCESS,
key.Create(HKEY_CURRENT_USER, kTestRoot, KEY_READ));
}
virtual void TearDown() {
logging::CloseLogFile();
RegKey key(HKEY_CURRENT_USER, L"", KEY_ALL_ACCESS);
ASSERT_EQ(ERROR_SUCCESS, key.DeleteKey(kTestRoot));
}
};
}
TEST_F(DeleteRegValueWorkItemTest, DeleteExistingValue) {
RegKey key;
std::wstring parent_key(kTestRoot);
parent_key.push_back(base::FilePath::kSeparators[0]);
parent_key.append(kWriteNew);
ASSERT_EQ(ERROR_SUCCESS,
key.Create(HKEY_CURRENT_USER, parent_key.c_str(), KEY_READ | KEY_WRITE));
std::wstring name_str(kNameStr);
std::wstring data_str(L"data_111");
ASSERT_EQ(ERROR_SUCCESS, key.WriteValue(name_str.c_str(), data_str.c_str()));
std::wstring name_dword(kNameDword);
DWORD data_dword = 100;
ASSERT_EQ(ERROR_SUCCESS, key.WriteValue(name_dword.c_str(), data_dword));
std::wstring name_empty(L"name_empty");
ASSERT_EQ(ERROR_SUCCESS, RegSetValueEx(key.Handle(), name_empty.c_str(), NULL,
REG_SZ, NULL, 0));
scoped_ptr<DeleteRegValueWorkItem> work_item1(
WorkItem::CreateDeleteRegValueWorkItem(HKEY_CURRENT_USER, parent_key,
name_str));
scoped_ptr<DeleteRegValueWorkItem> work_item2(
WorkItem::CreateDeleteRegValueWorkItem(HKEY_CURRENT_USER, parent_key,
name_dword));
scoped_ptr<DeleteRegValueWorkItem> work_item3(
WorkItem::CreateDeleteRegValueWorkItem(HKEY_CURRENT_USER, parent_key,
name_empty));
EXPECT_TRUE(work_item1->Do());
EXPECT_TRUE(work_item2->Do());
EXPECT_TRUE(work_item3->Do());
EXPECT_FALSE(key.HasValue(name_str.c_str()));
EXPECT_FALSE(key.HasValue(name_dword.c_str()));
EXPECT_FALSE(key.HasValue(name_empty.c_str()));
work_item1->Rollback();
work_item2->Rollback();
work_item3->Rollback();
std::wstring read_str;
DWORD read_dword;
EXPECT_EQ(ERROR_SUCCESS, key.ReadValue(name_str.c_str(), &read_str));
EXPECT_EQ(ERROR_SUCCESS, key.ReadValueDW(name_dword.c_str(), &read_dword));
EXPECT_EQ(read_str, data_str);
EXPECT_EQ(read_dword, data_dword);
DWORD type = 0;
DWORD size = 0;
EXPECT_EQ(ERROR_SUCCESS, key.ReadValue(name_empty.c_str(), NULL, &size,
&type));
EXPECT_EQ(REG_SZ, type);
EXPECT_EQ(0, size);
}
TEST_F(DeleteRegValueWorkItemTest, DeleteNonExistentValue) {
RegKey key;
std::wstring parent_key(kTestRoot);
parent_key.push_back(base::FilePath::kSeparators[0]);
parent_key.append(kWriteNew);
ASSERT_EQ(ERROR_SUCCESS,
key.Create(HKEY_CURRENT_USER, parent_key.c_str(), KEY_READ | KEY_WRITE));
std::wstring name_str(kNameStr);
std::wstring name_dword(kNameDword);
EXPECT_FALSE(key.HasValue(name_str.c_str()));
EXPECT_FALSE(key.HasValue(name_dword.c_str()));
scoped_ptr<DeleteRegValueWorkItem> work_item1(
WorkItem::CreateDeleteRegValueWorkItem(HKEY_CURRENT_USER, parent_key,
name_str));
scoped_ptr<DeleteRegValueWorkItem> work_item2(
WorkItem::CreateDeleteRegValueWorkItem(HKEY_CURRENT_USER, parent_key,
name_dword));
EXPECT_TRUE(work_item1->Do());
EXPECT_TRUE(work_item2->Do());
EXPECT_FALSE(key.HasValue(name_str.c_str()));
EXPECT_FALSE(key.HasValue(name_dword.c_str()));
work_item1->Rollback();
work_item2->Rollback();
EXPECT_FALSE(key.HasValue(name_str.c_str()));
EXPECT_FALSE(key.HasValue(name_dword.c_str()));
}