1 // Copyright (c) 2011 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/logging.h" 8 #include "base/memory/scoped_ptr.h" 9 #include "base/win/registry.h" 10 #include "chrome/installer/util/registry_key_backup.h" 11 #include "chrome/installer/util/registry_test_data.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 class RegistryKeyBackupTest : public testing::Test { 18 protected: 19 static void TearDownTestCase() { 20 logging::CloseLogFile(); 21 } 22 23 virtual void SetUp() { 24 ASSERT_TRUE(test_data_.Initialize(HKEY_CURRENT_USER, L"SOFTWARE\\TmpTmp")); 25 destination_path_.assign(test_data_.base_path()).append(L"\\Destination"); 26 } 27 28 RegistryTestData test_data_; 29 std::wstring destination_path_; 30 }; 31 32 // Test that writing an uninitialized backup does nothing. 33 TEST_F(RegistryKeyBackupTest, Uninitialized) { 34 RegistryKeyBackup backup; 35 36 EXPECT_TRUE(backup.WriteTo(test_data_.root_key(), 37 destination_path_.c_str(), 38 WorkItem::kWow64Default)); 39 EXPECT_FALSE(RegKey(test_data_.root_key(), destination_path_.c_str(), 40 KEY_READ).Valid()); 41 } 42 43 // Test that initializing a backup with a non-existent key works, and that 44 // writing it back out does nothing. 45 TEST_F(RegistryKeyBackupTest, MissingKey) { 46 std::wstring non_existent_key_path(test_data_.base_path() + L"\\NoKeyHere"); 47 RegistryKeyBackup backup; 48 49 EXPECT_TRUE(backup.Initialize(test_data_.root_key(), 50 non_existent_key_path.c_str(), 51 WorkItem::kWow64Default)); 52 EXPECT_TRUE(backup.WriteTo(test_data_.root_key(), 53 destination_path_.c_str(), 54 WorkItem::kWow64Default)); 55 EXPECT_FALSE(RegKey(test_data_.root_key(), destination_path_.c_str(), 56 KEY_READ).Valid()); 57 } 58 59 // Test that reading some data then writing it out does the right thing. 60 TEST_F(RegistryKeyBackupTest, ReadWrite) { 61 RegistryKeyBackup backup; 62 63 EXPECT_TRUE(backup.Initialize(test_data_.root_key(), 64 test_data_.non_empty_key_path().c_str(), 65 WorkItem::kWow64Default)); 66 EXPECT_TRUE(backup.WriteTo(test_data_.root_key(), 67 destination_path_.c_str(), 68 WorkItem::kWow64Default)); 69 test_data_.ExpectMatchesNonEmptyKey(test_data_.root_key(), 70 destination_path_.c_str()); 71 } 72 73 // Test that reading some data, swapping, then writing it out does the right 74 // thing. 75 TEST_F(RegistryKeyBackupTest, Swap) { 76 RegistryKeyBackup backup; 77 RegistryKeyBackup other_backup; 78 79 EXPECT_TRUE(backup.Initialize(test_data_.root_key(), 80 test_data_.non_empty_key_path().c_str(), 81 WorkItem::kWow64Default)); 82 backup.swap(other_backup); 83 EXPECT_TRUE(other_backup.WriteTo(test_data_.root_key(), 84 destination_path_.c_str(), 85 WorkItem::kWow64Default)); 86 87 // Now make sure the one we started with is truly empty. 88 EXPECT_EQ(ERROR_SUCCESS, 89 RegKey(test_data_.root_key(), L"", KEY_QUERY_VALUE) 90 .DeleteKey(destination_path_.c_str())); 91 EXPECT_TRUE(backup.WriteTo(test_data_.root_key(), 92 destination_path_.c_str(), 93 WorkItem::kWow64Default)); 94 EXPECT_FALSE(RegKey(test_data_.root_key(), destination_path_.c_str(), 95 KEY_READ).Valid()); 96 } 97