1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
|
// Copyright (c) 2012 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 <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() {
// Create a temporary key for testing
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();
// Clean up the temporary key
RegKey key(HKEY_CURRENT_USER, L"", KEY_ALL_ACCESS);
ASSERT_EQ(ERROR_SUCCESS, key.DeleteKey(kTestRoot));
}
};
} // namespace
// Delete a value. The value should get deleted after Do() and should be
// recreated after Rollback().
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);
// Verify empty value.
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);
}
// Try deleting a value that doesn't exist.
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()));
}
|