chrome/installer/util/delete_reg_value_work_item_unittest.cc

   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
  15 using base::win::RegKey;
  16
  17 namespace {
  18
  19 const wchar_t kTestRoot[] = L"DeleteRegValueWorkItemTest";
  20 const wchar_t kWriteNew[] = L"WriteNew";
  21 const wchar_t kNameStr[] = L"name_str";
  22 const wchar_t kNameDword[] = L"name_dword";
  23
  24 class DeleteRegValueWorkItemTest : public testing::Test {
  25  protected:
  26   void SetUp() override {
  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   void TearDown() override {
  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().
  46 TEST_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.
 103 TEST_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 }