delete_reg_value_work_item_unittest.cc revision cedac228d2dd51db4b79ea1e72c7f249408ee061 
1// Copyright (c) 2012 The Chromium Authors. All rights reserved.
2// Use of this source code is governed by a BSD-style license that can be
3// found in the LICENSE file.
4
5#include <windows.h>
6
7#include "base/files/file_path.h"
8#include "base/memory/scoped_ptr.h"
9#include "base/strings/string_util.h"
10#include "base/win/registry.h"
11#include "chrome/installer/util/delete_reg_value_work_item.h"
12#include "chrome/installer/util/work_item.h"
13#include "testing/gtest/include/gtest/gtest.h"
14
15using base::win::RegKey;
16
17namespace {
18
19const wchar_t kTestRoot[] = L"DeleteRegValueWorkItemTest";
20const wchar_t kWriteNew[] = L"WriteNew";
21const wchar_t kNameStr[] = L"name_str";
22const wchar_t kNameDword[] = L"name_dword";
23
24class DeleteRegValueWorkItemTest : public testing::Test {
25 protected:
26  virtual void SetUp() {
27    // Create a temporary key for testing
28    RegKey key(HKEY_CURRENT_USER, L"", KEY_ALL_ACCESS);
29    key.DeleteKey(kTestRoot);
30    ASSERT_NE(ERROR_SUCCESS, key.Open(HKEY_CURRENT_USER, kTestRoot, KEY_READ));
31    ASSERT_EQ(ERROR_SUCCESS,
32        key.Create(HKEY_CURRENT_USER, kTestRoot, KEY_READ));
33  }
34  virtual void TearDown() {
35    logging::CloseLogFile();
36    // Clean up the temporary key
37    RegKey key(HKEY_CURRENT_USER, L"", KEY_ALL_ACCESS);
38    ASSERT_EQ(ERROR_SUCCESS, key.DeleteKey(kTestRoot));
39  }
40};
41
42}  // namespace
43
44// Delete a value. The value should get deleted after Do() and should be
45// recreated after Rollback().
46TEST_F(DeleteRegValueWorkItemTest, DeleteExistingValue) {
47  RegKey key;
48  std::wstring parent_key(kTestRoot);
49  parent_key.push_back(base::FilePath::kSeparators[0]);
50  parent_key.append(kWriteNew);
51  ASSERT_EQ(ERROR_SUCCESS,
52      key.Create(HKEY_CURRENT_USER, parent_key.c_str(), KEY_READ | KEY_WRITE));
53  std::wstring name_str(kNameStr);
54  std::wstring data_str(L"data_111");
55  ASSERT_EQ(ERROR_SUCCESS, key.WriteValue(name_str.c_str(), data_str.c_str()));
56  std::wstring name_dword(kNameDword);
57  DWORD data_dword = 100;
58  ASSERT_EQ(ERROR_SUCCESS, key.WriteValue(name_dword.c_str(), data_dword));
59
60  std::wstring name_empty(L"name_empty");
61  ASSERT_EQ(ERROR_SUCCESS, RegSetValueEx(key.Handle(), name_empty.c_str(), NULL,
62                                         REG_SZ, NULL, 0));
63
64  scoped_ptr<DeleteRegValueWorkItem> work_item1(
65      WorkItem::CreateDeleteRegValueWorkItem(
66          HKEY_CURRENT_USER, parent_key, WorkItem::kWow64Default, name_str));
67  scoped_ptr<DeleteRegValueWorkItem> work_item2(
68      WorkItem::CreateDeleteRegValueWorkItem(
69          HKEY_CURRENT_USER, parent_key, WorkItem::kWow64Default, name_dword));
70  scoped_ptr<DeleteRegValueWorkItem> work_item3(
71      WorkItem::CreateDeleteRegValueWorkItem(
72          HKEY_CURRENT_USER, parent_key, WorkItem::kWow64Default, name_empty));
73
74  EXPECT_TRUE(work_item1->Do());
75  EXPECT_TRUE(work_item2->Do());
76  EXPECT_TRUE(work_item3->Do());
77
78  EXPECT_FALSE(key.HasValue(name_str.c_str()));
79  EXPECT_FALSE(key.HasValue(name_dword.c_str()));
80  EXPECT_FALSE(key.HasValue(name_empty.c_str()));
81
82  work_item1->Rollback();
83  work_item2->Rollback();
84  work_item3->Rollback();
85
86  std::wstring read_str;
87  DWORD read_dword;
88  EXPECT_EQ(ERROR_SUCCESS, key.ReadValue(name_str.c_str(), &read_str));
89  EXPECT_EQ(ERROR_SUCCESS, key.ReadValueDW(name_dword.c_str(), &read_dword));
90  EXPECT_EQ(read_str, data_str);
91  EXPECT_EQ(read_dword, data_dword);
92
93  // Verify empty value.
94  DWORD type = 0;
95  DWORD size = 0;
96  EXPECT_EQ(ERROR_SUCCESS, key.ReadValue(name_empty.c_str(), NULL, &size,
97                                         &type));
98  EXPECT_EQ(REG_SZ, type);
99  EXPECT_EQ(0, size);
100}
101
102// Try deleting a value that doesn't exist.
103TEST_F(DeleteRegValueWorkItemTest, DeleteNonExistentValue) {
104  RegKey key;
105  std::wstring parent_key(kTestRoot);
106  parent_key.push_back(base::FilePath::kSeparators[0]);
107  parent_key.append(kWriteNew);
108  ASSERT_EQ(ERROR_SUCCESS,
109      key.Create(HKEY_CURRENT_USER, parent_key.c_str(), KEY_READ | KEY_WRITE));
110  std::wstring name_str(kNameStr);
111  std::wstring name_dword(kNameDword);
112  EXPECT_FALSE(key.HasValue(name_str.c_str()));
113  EXPECT_FALSE(key.HasValue(name_dword.c_str()));
114
115  scoped_ptr<DeleteRegValueWorkItem> work_item1(
116      WorkItem::CreateDeleteRegValueWorkItem(
117          HKEY_CURRENT_USER, parent_key, WorkItem::kWow64Default, name_str));
118  scoped_ptr<DeleteRegValueWorkItem> work_item2(
119      WorkItem::CreateDeleteRegValueWorkItem(
120          HKEY_CURRENT_USER, parent_key, WorkItem::kWow64Default, name_dword));
121
122  EXPECT_TRUE(work_item1->Do());
123  EXPECT_TRUE(work_item2->Do());
124
125  EXPECT_FALSE(key.HasValue(name_str.c_str()));
126  EXPECT_FALSE(key.HasValue(name_dword.c_str()));
127
128  work_item1->Rollback();
129  work_item2->Rollback();
130
131  EXPECT_FALSE(key.HasValue(name_str.c_str()));
132  EXPECT_FALSE(key.HasValue(name_dword.c_str()));
133}
134