| OLD | NEW |
| 1 // Copyright (c) 2010 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2010 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include <windows.h> | 5 #include <windows.h> |
| 6 | 6 |
| 7 #include "base/file_util.h" | 7 #include "base/file_util.h" |
| 8 #include "base/scoped_ptr.h" | 8 #include "base/scoped_ptr.h" |
| 9 #include "base/string_util.h" | 9 #include "base/string_util.h" |
| 10 #include "base/win/registry.h" | 10 #include "base/win/registry.h" |
| 11 #include "chrome/installer/util/delete_reg_value_work_item.h" | 11 #include "chrome/installer/util/delete_reg_value_work_item.h" |
| 12 #include "chrome/installer/util/work_item.h" | 12 #include "chrome/installer/util/work_item.h" |
| 13 #include "testing/gtest/include/gtest/gtest.h" | 13 #include "testing/gtest/include/gtest/gtest.h" |
| 14 | 14 |
| 15 using base::win::RegKey; | 15 using base::win::RegKey; |
| 16 | 16 |
| 17 namespace { | 17 namespace { |
| 18 | 18 |
| 19 wchar_t test_root[] = L"DeleteRegValueWorkItemTest"; | 19 wchar_t test_root[] = L"DeleteRegValueWorkItemTest"; |
| 20 | 20 |
| 21 class DeleteRegValueWorkItemTest : public testing::Test { | 21 class DeleteRegValueWorkItemTest : public testing::Test { |
| 22 protected: | 22 protected: |
| 23 virtual void SetUp() { | 23 virtual void SetUp() { |
| 24 // Create a temporary key for testing | 24 // Create a temporary key for testing |
| 25 RegKey key(HKEY_CURRENT_USER, L"", KEY_ALL_ACCESS); | 25 RegKey key(HKEY_CURRENT_USER, L"", KEY_ALL_ACCESS); |
| 26 key.DeleteKey(test_root); | 26 key.DeleteKey(test_root); |
| 27 ASSERT_FALSE(key.Open(HKEY_CURRENT_USER, test_root, KEY_READ)); | 27 ASSERT_NE(ERROR_SUCCESS, key.Open(HKEY_CURRENT_USER, test_root, KEY_READ)); |
| 28 ASSERT_TRUE(key.Create(HKEY_CURRENT_USER, test_root, KEY_READ)); | 28 ASSERT_EQ(ERROR_SUCCESS, |
| 29 key.Create(HKEY_CURRENT_USER, test_root, KEY_READ)); |
| 29 } | 30 } |
| 30 virtual void TearDown() { | 31 virtual void TearDown() { |
| 31 logging::CloseLogFile(); | 32 logging::CloseLogFile(); |
| 32 // Clean up the temporary key | 33 // Clean up the temporary key |
| 33 RegKey key(HKEY_CURRENT_USER, L"", KEY_ALL_ACCESS); | 34 RegKey key(HKEY_CURRENT_USER, L"", KEY_ALL_ACCESS); |
| 34 ASSERT_TRUE(key.DeleteKey(test_root)); | 35 ASSERT_EQ(ERROR_SUCCESS, key.DeleteKey(test_root)); |
| 35 } | 36 } |
| 36 }; | 37 }; |
| 37 | 38 |
| 38 } // namespace | 39 } // namespace |
| 39 | 40 |
| 40 // Delete a value. The value should get deleted after Do() and should be | 41 // Delete a value. The value should get deleted after Do() and should be |
| 41 // recreated after Rollback(). | 42 // recreated after Rollback(). |
| 42 TEST_F(DeleteRegValueWorkItemTest, DeleteExistingValue) { | 43 TEST_F(DeleteRegValueWorkItemTest, DeleteExistingValue) { |
| 43 RegKey key; | 44 RegKey key; |
| 44 std::wstring parent_key(test_root); | 45 std::wstring parent_key(test_root); |
| 45 file_util::AppendToPath(&parent_key, L"WriteNew"); | 46 file_util::AppendToPath(&parent_key, L"WriteNew"); |
| 46 ASSERT_TRUE(key.Create(HKEY_CURRENT_USER, parent_key.c_str(), | 47 ASSERT_EQ(ERROR_SUCCESS, |
| 47 KEY_READ | KEY_WRITE)); | 48 key.Create(HKEY_CURRENT_USER, parent_key.c_str(), KEY_READ | KEY_WRITE)); |
| 48 std::wstring name_str(L"name_str"); | 49 std::wstring name_str(L"name_str"); |
| 49 std::wstring data_str(L"data_111"); | 50 std::wstring data_str(L"data_111"); |
| 50 ASSERT_TRUE(key.WriteValue(name_str.c_str(), data_str.c_str())); | 51 ASSERT_EQ(ERROR_SUCCESS, key.WriteValue(name_str.c_str(), data_str.c_str())); |
| 51 std::wstring name_dword(L"name_dword"); | 52 std::wstring name_dword(L"name_dword"); |
| 52 DWORD data_dword = 100; | 53 DWORD data_dword = 100; |
| 53 ASSERT_TRUE(key.WriteValue(name_dword.c_str(), data_dword)); | 54 ASSERT_EQ(ERROR_SUCCESS, key.WriteValue(name_dword.c_str(), data_dword)); |
| 54 | 55 |
| 55 scoped_ptr<DeleteRegValueWorkItem> work_item1( | 56 scoped_ptr<DeleteRegValueWorkItem> work_item1( |
| 56 WorkItem::CreateDeleteRegValueWorkItem(HKEY_CURRENT_USER, parent_key, | 57 WorkItem::CreateDeleteRegValueWorkItem(HKEY_CURRENT_USER, parent_key, |
| 57 name_str, true)); | 58 name_str, true)); |
| 58 scoped_ptr<DeleteRegValueWorkItem> work_item2( | 59 scoped_ptr<DeleteRegValueWorkItem> work_item2( |
| 59 WorkItem::CreateDeleteRegValueWorkItem(HKEY_CURRENT_USER, parent_key, | 60 WorkItem::CreateDeleteRegValueWorkItem(HKEY_CURRENT_USER, parent_key, |
| 60 name_dword, false)); | 61 name_dword, false)); |
| 61 | 62 |
| 62 EXPECT_TRUE(work_item1->Do()); | 63 EXPECT_TRUE(work_item1->Do()); |
| 63 EXPECT_TRUE(work_item2->Do()); | 64 EXPECT_TRUE(work_item2->Do()); |
| 64 | 65 |
| 65 EXPECT_FALSE(key.ValueExists(name_str.c_str())); | 66 EXPECT_FALSE(key.ValueExists(name_str.c_str())); |
| 66 EXPECT_FALSE(key.ValueExists(name_dword.c_str())); | 67 EXPECT_FALSE(key.ValueExists(name_dword.c_str())); |
| 67 | 68 |
| 68 work_item1->Rollback(); | 69 work_item1->Rollback(); |
| 69 work_item2->Rollback(); | 70 work_item2->Rollback(); |
| 70 | 71 |
| 71 std::wstring read_str; | 72 std::wstring read_str; |
| 72 DWORD read_dword; | 73 DWORD read_dword; |
| 73 EXPECT_TRUE(key.ReadValue(name_str.c_str(), &read_str)); | 74 EXPECT_EQ(ERROR_SUCCESS, key.ReadValue(name_str.c_str(), &read_str)); |
| 74 EXPECT_TRUE(key.ReadValueDW(name_dword.c_str(), &read_dword)); | 75 EXPECT_EQ(ERROR_SUCCESS, key.ReadValueDW(name_dword.c_str(), &read_dword)); |
| 75 EXPECT_EQ(read_str, data_str); | 76 EXPECT_EQ(read_str, data_str); |
| 76 EXPECT_EQ(read_dword, data_dword); | 77 EXPECT_EQ(read_dword, data_dword); |
| 77 } | 78 } |
| 78 | 79 |
| 79 // Try deleting a value that doesn't exist. | 80 // Try deleting a value that doesn't exist. |
| 80 TEST_F(DeleteRegValueWorkItemTest, DeleteNonExistentValue) { | 81 TEST_F(DeleteRegValueWorkItemTest, DeleteNonExistentValue) { |
| 81 RegKey key; | 82 RegKey key; |
| 82 std::wstring parent_key(test_root); | 83 std::wstring parent_key(test_root); |
| 83 file_util::AppendToPath(&parent_key, L"WriteNew"); | 84 file_util::AppendToPath(&parent_key, L"WriteNew"); |
| 84 ASSERT_TRUE(key.Create(HKEY_CURRENT_USER, parent_key.c_str(), | 85 ASSERT_EQ(ERROR_SUCCESS, |
| 85 KEY_READ | KEY_WRITE)); | 86 key.Create(HKEY_CURRENT_USER, parent_key.c_str(), KEY_READ | KEY_WRITE)); |
| 86 std::wstring name_str(L"name_str"); | 87 std::wstring name_str(L"name_str"); |
| 87 std::wstring name_dword(L"name_dword"); | 88 std::wstring name_dword(L"name_dword"); |
| 88 EXPECT_FALSE(key.ValueExists(name_str.c_str())); | 89 EXPECT_FALSE(key.ValueExists(name_str.c_str())); |
| 89 EXPECT_FALSE(key.ValueExists(name_dword.c_str())); | 90 EXPECT_FALSE(key.ValueExists(name_dword.c_str())); |
| 90 | 91 |
| 91 scoped_ptr<DeleteRegValueWorkItem> work_item1( | 92 scoped_ptr<DeleteRegValueWorkItem> work_item1( |
| 92 WorkItem::CreateDeleteRegValueWorkItem(HKEY_CURRENT_USER, parent_key, | 93 WorkItem::CreateDeleteRegValueWorkItem(HKEY_CURRENT_USER, parent_key, |
| 93 name_str, true)); | 94 name_str, true)); |
| 94 scoped_ptr<DeleteRegValueWorkItem> work_item2( | 95 scoped_ptr<DeleteRegValueWorkItem> work_item2( |
| 95 WorkItem::CreateDeleteRegValueWorkItem(HKEY_CURRENT_USER, parent_key, | 96 WorkItem::CreateDeleteRegValueWorkItem(HKEY_CURRENT_USER, parent_key, |
| 96 name_dword, false)); | 97 name_dword, false)); |
| 97 | 98 |
| 98 EXPECT_TRUE(work_item1->Do()); | 99 EXPECT_TRUE(work_item1->Do()); |
| 99 EXPECT_TRUE(work_item2->Do()); | 100 EXPECT_TRUE(work_item2->Do()); |
| 100 | 101 |
| 101 EXPECT_FALSE(key.ValueExists(name_str.c_str())); | 102 EXPECT_FALSE(key.ValueExists(name_str.c_str())); |
| 102 EXPECT_FALSE(key.ValueExists(name_dword.c_str())); | 103 EXPECT_FALSE(key.ValueExists(name_dword.c_str())); |
| 103 | 104 |
| 104 work_item1->Rollback(); | 105 work_item1->Rollback(); |
| 105 work_item2->Rollback(); | 106 work_item2->Rollback(); |
| 106 | 107 |
| 107 EXPECT_FALSE(key.ValueExists(name_str.c_str())); | 108 EXPECT_FALSE(key.ValueExists(name_str.c_str())); |
| 108 EXPECT_FALSE(key.ValueExists(name_dword.c_str())); | 109 EXPECT_FALSE(key.ValueExists(name_dword.c_str())); |
| 109 } | 110 } |
| OLD | NEW |