Index: chrome/installer/util/google_update_settings_unittest.cc |
=================================================================== |
--- chrome/installer/util/google_update_settings_unittest.cc (revision 49358) |
+++ chrome/installer/util/google_update_settings_unittest.cc (working copy) |
@@ -5,11 +5,8 @@ |
#include <windows.h> |
#include "base/registry.h" |
-#include "base/scoped_ptr.h" |
#include "chrome/installer/util/browser_distribution.h" |
-#include "chrome/installer/util/google_update_constants.h" |
#include "chrome/installer/util/google_update_settings.h" |
-#include "chrome/installer/util/work_item_list.h" |
#include "testing/gtest/include/gtest/gtest.h" |
namespace { |
@@ -18,8 +15,6 @@ |
const wchar_t* kHKLMReplacement = |
L"Software\\Google\\InstallUtilUnittest\\HKLM"; |
-const wchar_t kTestProductGuid[] = L"{89F1B351-B15D-48D4-8F10-1298721CF13D}"; |
- |
// This test fixture redirects the HKLM and HKCU registry hives for |
// the duration of the test to make it independent of the machine |
// and user settings. |
@@ -122,42 +117,6 @@ |
} |
} |
- // Creates "ap" key with the value given as parameter. Also adds work |
- // items to work_item_list given so that they can be rolled back later. |
- bool CreateApKey(WorkItemList* work_item_list, const std::wstring& value) { |
- HKEY reg_root = HKEY_CURRENT_USER; |
- std::wstring reg_key = GetApKeyPath(); |
- work_item_list->AddCreateRegKeyWorkItem(reg_root, reg_key); |
- work_item_list->AddSetRegValueWorkItem(reg_root, reg_key, |
- google_update::kRegApField, value.c_str(), true); |
- if (!work_item_list->Do()) { |
- work_item_list->Rollback(); |
- return false; |
- } |
- return true; |
- } |
- |
- // Returns the key path of "ap" key, e.g.: |
- // Google\Update\ClientState\<kTestProductGuid> |
- std::wstring GetApKeyPath() { |
- std::wstring reg_key(google_update::kRegPathClientState); |
- reg_key.append(L"\\"); |
- reg_key.append(kTestProductGuid); |
- return reg_key; |
- } |
- |
- // Utility method to read "ap" key value |
- std::wstring ReadApKeyValue() { |
- RegKey key; |
- std::wstring ap_key_value; |
- std::wstring reg_key = GetApKeyPath(); |
- if (key.Open(HKEY_CURRENT_USER, reg_key.c_str(), KEY_ALL_ACCESS) && |
- key.ReadValue(google_update::kRegApField, &ap_key_value)) { |
- return ap_key_value; |
- } |
- return std::wstring(); |
- } |
- |
RegKey hkcu_; |
RegKey hklm_; |
}; |
@@ -206,145 +165,3 @@ |
TEST_F(GoogleUpdateSettingsTest, CurrentChromeChannelVariousApValuesUser) { |
TestCurrentChromeChannelWithVariousApValues(USER_INSTALL); |
} |
- |
-TEST_F(GoogleUpdateSettingsTest, GetNewGoogleUpdateApKeyTest) { |
- installer_util::InstallStatus s = installer_util::FIRST_INSTALL_SUCCESS; |
- installer_util::InstallStatus f = installer_util::INSTALL_FAILED; |
- |
- // Incremental Installer that worked. |
- EXPECT_EQ(GoogleUpdateSettings::GetNewGoogleUpdateApKey(true, s, L""), L""); |
- EXPECT_EQ(GoogleUpdateSettings::GetNewGoogleUpdateApKey(true, s, L"1.1"), |
- L"1.1"); |
- EXPECT_EQ(GoogleUpdateSettings::GetNewGoogleUpdateApKey(true, s, L"1.1-dev"), |
- L"1.1-dev"); |
- EXPECT_EQ(GoogleUpdateSettings::GetNewGoogleUpdateApKey(true, s, L"-full"), |
- L""); |
- EXPECT_EQ(GoogleUpdateSettings::GetNewGoogleUpdateApKey(true, s, L"1.1-full"), |
- L"1.1"); |
- EXPECT_EQ(GoogleUpdateSettings::GetNewGoogleUpdateApKey(true, s, |
- L"1.1-dev-full"), |
- L"1.1-dev"); |
- |
- // Incremental Installer that failed. |
- EXPECT_EQ(GoogleUpdateSettings::GetNewGoogleUpdateApKey(true, f, L""), |
- L"-full"); |
- EXPECT_EQ(GoogleUpdateSettings::GetNewGoogleUpdateApKey(true, f, L"1.1"), |
- L"1.1-full"); |
- EXPECT_EQ(GoogleUpdateSettings::GetNewGoogleUpdateApKey(true, f, L"1.1-dev"), |
- L"1.1-dev-full"); |
- EXPECT_EQ(GoogleUpdateSettings::GetNewGoogleUpdateApKey(true, f, L"-full"), |
- L"-full"); |
- EXPECT_EQ(GoogleUpdateSettings::GetNewGoogleUpdateApKey(true, f, L"1.1-full"), |
- L"1.1-full"); |
- EXPECT_EQ(GoogleUpdateSettings::GetNewGoogleUpdateApKey(true, f, |
- L"1.1-dev-full"), |
- L"1.1-dev-full"); |
- |
- // Full Installer that worked. |
- EXPECT_EQ(GoogleUpdateSettings::GetNewGoogleUpdateApKey(false, s, L""), L""); |
- EXPECT_EQ(GoogleUpdateSettings::GetNewGoogleUpdateApKey(false, s, L"1.1"), |
- L"1.1"); |
- EXPECT_EQ(GoogleUpdateSettings::GetNewGoogleUpdateApKey(false, s, L"1.1-dev"), |
- L"1.1-dev"); |
- EXPECT_EQ(GoogleUpdateSettings::GetNewGoogleUpdateApKey(false, s, L"-full"), |
- L""); |
- EXPECT_EQ(GoogleUpdateSettings::GetNewGoogleUpdateApKey(false, s, |
- L"1.1-full"), |
- L"1.1"); |
- EXPECT_EQ(GoogleUpdateSettings::GetNewGoogleUpdateApKey(false, s, |
- L"1.1-dev-full"), |
- L"1.1-dev"); |
- |
- // Full Installer that failed. |
- EXPECT_EQ(GoogleUpdateSettings::GetNewGoogleUpdateApKey(false, f, L""), L""); |
- EXPECT_EQ(GoogleUpdateSettings::GetNewGoogleUpdateApKey(false, f, L"1.1"), |
- L"1.1"); |
- EXPECT_EQ(GoogleUpdateSettings::GetNewGoogleUpdateApKey(false, f, L"1.1-dev"), |
- L"1.1-dev"); |
- EXPECT_EQ(GoogleUpdateSettings::GetNewGoogleUpdateApKey(false, f, L"-full"), |
- L""); |
- EXPECT_EQ(GoogleUpdateSettings::GetNewGoogleUpdateApKey(false, f, |
- L"1.1-full"), |
- L"1.1"); |
- EXPECT_EQ(GoogleUpdateSettings::GetNewGoogleUpdateApKey(false, f, |
- L"1.1-dev-full"), |
- L"1.1-dev"); |
-} |
- |
-TEST_F(GoogleUpdateSettingsTest, UpdateDiffInstallStatusTest) { |
- scoped_ptr<WorkItemList> work_item_list(WorkItem::CreateWorkItemList()); |
- // Test incremental install failure |
- ASSERT_TRUE(CreateApKey(work_item_list.get(), L"")) |
- << "Failed to create ap key."; |
- GoogleUpdateSettings::UpdateDiffInstallStatus(false, true, |
- installer_util::INSTALL_FAILED, |
- kTestProductGuid); |
- EXPECT_STREQ(ReadApKeyValue().c_str(), L"-full"); |
- work_item_list->Rollback(); |
- |
- work_item_list.reset(WorkItem::CreateWorkItemList()); |
- // Test incremental install success |
- ASSERT_TRUE(CreateApKey(work_item_list.get(), L"")) |
- << "Failed to create ap key."; |
- GoogleUpdateSettings::UpdateDiffInstallStatus(false, true, |
- installer_util::FIRST_INSTALL_SUCCESS, |
- kTestProductGuid); |
- EXPECT_STREQ(ReadApKeyValue().c_str(), L""); |
- work_item_list->Rollback(); |
- |
- work_item_list.reset(WorkItem::CreateWorkItemList()); |
- // Test full install failure |
- ASSERT_TRUE(CreateApKey(work_item_list.get(), L"-full")) |
- << "Failed to create ap key."; |
- GoogleUpdateSettings::UpdateDiffInstallStatus(false, false, |
- installer_util::INSTALL_FAILED, |
- kTestProductGuid); |
- EXPECT_STREQ(ReadApKeyValue().c_str(), L""); |
- work_item_list->Rollback(); |
- |
- work_item_list.reset(WorkItem::CreateWorkItemList()); |
- // Test full install success |
- ASSERT_TRUE(CreateApKey(work_item_list.get(), L"-full")) |
- << "Failed to create ap key."; |
- GoogleUpdateSettings::UpdateDiffInstallStatus(false, false, |
- installer_util::FIRST_INSTALL_SUCCESS, |
- kTestProductGuid); |
- EXPECT_STREQ(ReadApKeyValue().c_str(), L""); |
- work_item_list->Rollback(); |
- |
- work_item_list.reset(WorkItem::CreateWorkItemList()); |
- // Test the case of when "ap" key doesnt exist at all |
- std::wstring ap_key_value = ReadApKeyValue(); |
- std::wstring reg_key = GetApKeyPath(); |
- HKEY reg_root = HKEY_CURRENT_USER; |
- bool ap_key_deleted = false; |
- RegKey key; |
- if (!key.Open(HKEY_CURRENT_USER, reg_key.c_str(), KEY_ALL_ACCESS)) { |
- work_item_list->AddCreateRegKeyWorkItem(reg_root, reg_key); |
- ASSERT_TRUE(work_item_list->Do()) << "Failed to create ClientState key."; |
- } else if (key.DeleteValue(google_update::kRegApField)) { |
- ap_key_deleted = true; |
- } |
- // try differential installer |
- GoogleUpdateSettings::UpdateDiffInstallStatus(false, true, |
- installer_util::INSTALL_FAILED, |
- kTestProductGuid); |
- EXPECT_STREQ(ReadApKeyValue().c_str(), L"-full"); |
- // try full installer now |
- GoogleUpdateSettings::UpdateDiffInstallStatus(false, false, |
- installer_util::INSTALL_FAILED, |
- kTestProductGuid); |
- EXPECT_STREQ(ReadApKeyValue().c_str(), L""); |
- // Now cleanup to leave the system in unchanged state. |
- // - Diff installer creates an ap key if it didnt exist, so delete this ap key |
- // - If we created any reg key path for ap, roll it back |
- // - Finally restore the original value of ap key. |
- key.Open(HKEY_CURRENT_USER, reg_key.c_str(), KEY_ALL_ACCESS); |
- key.DeleteValue(google_update::kRegApField); |
- work_item_list->Rollback(); |
- if (ap_key_deleted) { |
- work_item_list.reset(WorkItem::CreateWorkItemList()); |
- ASSERT_TRUE(CreateApKey(work_item_list.get(), ap_key_value)) |
- << "Failed to restore ap key."; |
- } |
-} |