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

Unified Diff: chrome/installer/util/delete_reg_value_work_item_unittest.cc

Issue 1309873005: Introduce DeleteRegValueWorkItemTest.DeleteValueInNonExistentKey and modernize associated fixture a… (Closed) Base URL: https://chromium.googlesource.com/chromium/src.git@a1_no_delegate_execute_win10
Patch Set: git cl format Created 5 years, 4 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 | « no previous file | no next file » | no next file with comments »
Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
Index: chrome/installer/util/delete_reg_value_work_item_unittest.cc
diff --git a/chrome/installer/util/delete_reg_value_work_item_unittest.cc b/chrome/installer/util/delete_reg_value_work_item_unittest.cc
index bf24bba976bda90b6084d7d0fdd8b4a7ade4372d..9a508aec4b23e00e6eb2745134b6b141e4295448 100644
--- a/chrome/installer/util/delete_reg_value_work_item_unittest.cc
+++ b/chrome/installer/util/delete_reg_value_work_item_unittest.cc
@@ -4,9 +4,9 @@
#include <windows.h>
-#include "base/files/file_path.h"
#include "base/memory/scoped_ptr.h"
-#include "base/strings/string_util.h"
+#include "base/strings/string16.h"
+#include "base/test/test_reg_util_win.h"
#include "base/win/registry.h"
#include "chrome/installer/util/delete_reg_value_work_item.h"
#include "chrome/installer/util/work_item.h"
@@ -16,27 +16,22 @@ using base::win::RegKey;
namespace {
-const wchar_t kTestRoot[] = L"DeleteRegValueWorkItemTest";
-const wchar_t kWriteNew[] = L"WriteNew";
+const wchar_t kTestKey[] = L"DeleteRegValueWorkItemTest";
const wchar_t kNameStr[] = L"name_str";
const wchar_t kNameDword[] = L"name_dword";
class DeleteRegValueWorkItemTest : public testing::Test {
protected:
+ DeleteRegValueWorkItemTest() {}
+
void SetUp() override {
- // 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));
- }
- void TearDown() override {
- logging::CloseLogFile();
- // Clean up the temporary key
- RegKey key(HKEY_CURRENT_USER, L"", KEY_ALL_ACCESS);
- ASSERT_EQ(ERROR_SUCCESS, key.DeleteKey(kTestRoot));
+ registry_override_manager_.OverrideRegistry(HKEY_CURRENT_USER);
}
+
+ private:
+ registry_util::RegistryOverrideManager registry_override_manager_;
+
+ DISALLOW_COPY_AND_ASSIGN(DeleteRegValueWorkItemTest);
};
} // namespace
@@ -45,56 +40,58 @@ class DeleteRegValueWorkItemTest : public testing::Test {
// 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));
+ key.Create(HKEY_CURRENT_USER, kTestKey, KEY_READ | KEY_WRITE));
+ const base::string16 data_str(L"data_111");
+ ASSERT_EQ(ERROR_SUCCESS, key.WriteValue(kNameStr, data_str.c_str()));
+ const DWORD data_dword = 100;
+ ASSERT_EQ(ERROR_SUCCESS, key.WriteValue(kNameDword, data_dword));
+
+ static const wchar_t kNameEmpty[](L"name_empty");
+ ASSERT_EQ(ERROR_SUCCESS,
+ RegSetValueEx(key.Handle(), kNameEmpty, NULL, REG_SZ, NULL, 0));
scoped_ptr<DeleteRegValueWorkItem> work_item1(
WorkItem::CreateDeleteRegValueWorkItem(
- HKEY_CURRENT_USER, parent_key, WorkItem::kWow64Default, name_str));
+ HKEY_CURRENT_USER, kTestKey, WorkItem::kWow64Default, kNameStr));
scoped_ptr<DeleteRegValueWorkItem> work_item2(
WorkItem::CreateDeleteRegValueWorkItem(
- HKEY_CURRENT_USER, parent_key, WorkItem::kWow64Default, name_dword));
+ HKEY_CURRENT_USER, kTestKey, WorkItem::kWow64Default, kNameDword));
scoped_ptr<DeleteRegValueWorkItem> work_item3(
WorkItem::CreateDeleteRegValueWorkItem(
- HKEY_CURRENT_USER, parent_key, WorkItem::kWow64Default, name_empty));
+ HKEY_CURRENT_USER, kTestKey, WorkItem::kWow64Default, kNameEmpty));
+
+ EXPECT_TRUE(key.HasValue(kNameStr));
+ EXPECT_TRUE(key.HasValue(kNameDword));
+ EXPECT_TRUE(key.HasValue(kNameEmpty));
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()));
+ EXPECT_FALSE(key.HasValue(kNameStr));
+ EXPECT_FALSE(key.HasValue(kNameDword));
+ EXPECT_FALSE(key.HasValue(kNameEmpty));
work_item1->Rollback();
work_item2->Rollback();
work_item3->Rollback();
- std::wstring read_str;
+ EXPECT_TRUE(key.HasValue(kNameStr));
+ EXPECT_TRUE(key.HasValue(kNameDword));
+ EXPECT_TRUE(key.HasValue(kNameEmpty));
+
+ base::string16 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(ERROR_SUCCESS, key.ReadValue(kNameStr, &read_str));
+ EXPECT_EQ(ERROR_SUCCESS, key.ReadValueDW(kNameDword, &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(ERROR_SUCCESS, key.ReadValue(kNameEmpty, NULL, &size, &type));
EXPECT_EQ(REG_SZ, type);
EXPECT_EQ(0, size);
}
@@ -102,32 +99,54 @@ TEST_F(DeleteRegValueWorkItemTest, DeleteExistingValue) {
// 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()));
+ key.Create(HKEY_CURRENT_USER, kTestKey, KEY_READ | KEY_WRITE));
+ EXPECT_FALSE(key.HasValue(kNameStr));
+ EXPECT_FALSE(key.HasValue(kNameDword));
+
+ scoped_ptr<DeleteRegValueWorkItem> work_item1(
+ WorkItem::CreateDeleteRegValueWorkItem(
+ HKEY_CURRENT_USER, kTestKey, WorkItem::kWow64Default, kNameStr));
+ scoped_ptr<DeleteRegValueWorkItem> work_item2(
+ WorkItem::CreateDeleteRegValueWorkItem(
+ HKEY_CURRENT_USER, kTestKey, WorkItem::kWow64Default, kNameDword));
+
+ EXPECT_TRUE(work_item1->Do());
+ EXPECT_TRUE(work_item2->Do());
+
+ EXPECT_FALSE(key.HasValue(kNameStr));
+ EXPECT_FALSE(key.HasValue(kNameDword));
+
+ work_item1->Rollback();
+ work_item2->Rollback();
+
+ EXPECT_FALSE(key.HasValue(kNameStr));
+ EXPECT_FALSE(key.HasValue(kNameDword));
+}
+
+// Try deleting a value whose key doesn't even exist.
+TEST_F(DeleteRegValueWorkItemTest, DeleteValueInNonExistentKey) {
+ RegKey key;
+ // Confirm the key doesn't exist.
+ ASSERT_EQ(ERROR_FILE_NOT_FOUND,
+ key.Open(HKEY_CURRENT_USER, kTestKey, KEY_READ));
scoped_ptr<DeleteRegValueWorkItem> work_item1(
WorkItem::CreateDeleteRegValueWorkItem(
- HKEY_CURRENT_USER, parent_key, WorkItem::kWow64Default, name_str));
+ HKEY_CURRENT_USER, kTestKey, WorkItem::kWow64Default, kNameStr));
scoped_ptr<DeleteRegValueWorkItem> work_item2(
WorkItem::CreateDeleteRegValueWorkItem(
- HKEY_CURRENT_USER, parent_key, WorkItem::kWow64Default, name_dword));
+ HKEY_CURRENT_USER, kTestKey, WorkItem::kWow64Default, kNameDword));
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()));
+ EXPECT_EQ(ERROR_FILE_NOT_FOUND,
+ key.Open(HKEY_CURRENT_USER, kTestKey, KEY_READ));
work_item1->Rollback();
work_item2->Rollback();
- EXPECT_FALSE(key.HasValue(name_str.c_str()));
- EXPECT_FALSE(key.HasValue(name_dword.c_str()));
+ EXPECT_EQ(ERROR_FILE_NOT_FOUND,
+ key.Open(HKEY_CURRENT_USER, kTestKey, KEY_READ));
}
« no previous file with comments | « no previous file | no next file » | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698