Home | History | Annotate | Download | only in util 
      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   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().
     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 }
    134