Index: chrome/installer/setup/setup_install_details_unittest.cc |
diff --git a/chrome/installer/setup/setup_install_details_unittest.cc b/chrome/installer/setup/setup_install_details_unittest.cc |
new file mode 100644 |
index 0000000000000000000000000000000000000000..829c29e056b5db7c742d9e4c9fe614c6d0f85aad |
--- /dev/null |
+++ b/chrome/installer/setup/setup_install_details_unittest.cc |
@@ -0,0 +1,290 @@ |
+// Copyright 2017 The Chromium Authors. All rights reserved. |
+// Use of this source code is governed by a BSD-style license that can be |
+// found in the LICENSE file. |
+ |
+#include "chrome/installer/setup/setup_install_details.h" |
+ |
+#include <windows.h> |
+ |
+#include "base/command_line.h" |
+#include "base/macros.h" |
+#include "base/memory/ptr_util.h" |
+#include "base/test/test_reg_util_win.h" |
+#include "chrome/install_static/install_details.h" |
+#include "chrome/install_static/install_modes.h" |
+#include "chrome/installer/util/master_preferences.h" |
+#include "chrome_elf/nt_registry/nt_registry.h" |
+ |
+#include "testing/gmock/include/gmock/gmock.h" |
+#include "testing/gtest/include/gtest/gtest.h" |
+ |
+using ::testing::Eq; |
+ |
+struct TestData { |
+ // Inputs: |
+ const wchar_t* command_line; |
+ const wchar_t* uninstall_args; |
+ const wchar_t* product_ap; |
+ const wchar_t* binaries_ap; |
+ |
+ // Expectations: |
+ install_static::InstallConstantIndex index; |
+ bool system_level; |
+ const wchar_t* channel; |
+}; |
+ |
+#if defined(GOOGLE_CHROME_BUILD) |
+constexpr TestData kTestData[] = { |
+ // User-level test cases. |
+ { |
+ L"setup.exe", // User-level, primary mode. |
+ L"", // New install. |
+ L"x64-stable", // Stable channel. |
+ L"1.1-beta", // Value ignored in binaries. |
+ install_static::STABLE_INDEX, // Expect primary mode. |
+ false, // Expect user-level. |
+ L"", // Expect stable channel. |
+ }, |
+ { |
+ L"setup.exe", // User-level, primary mode. |
+ L"--uninstall", // Updating an existing install. |
+ L"x64-stable", // Stable channel. |
+ L"1.1-beta", // Value ignored in binaries. |
+ install_static::STABLE_INDEX, // Expect primary mode. |
+ false, // Expect user-level. |
+ L"", // Expect stable channel. |
+ }, |
+ { |
+ L"setup.exe", // User-level, primary mode. |
+ L"--uninstall --multi-install", // Updating an existing multi-install. |
+ L"x64-stable", // Value ignored in product. |
+ L"1.1-beta", // Channel read from binaries. |
+ install_static::STABLE_INDEX, // Expect primary mode. |
+ false, // Expect user-level. |
+ L"beta", // Expect beta channel. |
+ }, |
+ { |
+ L"setup.exe", // User-level, primary mode. |
+ L"", // New install. |
+ L"1.1-beta", // Beta channel. |
+ L"x64-stable", // Value ignored in binaries. |
+ install_static::STABLE_INDEX, // Expect primary mode. |
+ false, // Expect user-level. |
+ L"beta", // Expect beta channel. |
+ }, |
+ { |
+ L"setup.exe --chrome-sxs", // User-level, secondary SxS canary mode. |
+ L"", // New install. |
+ L"", // Unused. |
+ L"", // Unused. |
+ install_static::CANARY_INDEX, // Expect SxS canary mode. |
+ false, // Expect user-level. |
+ L"canary", // Expect canary channel. |
+ }, |
+ { |
+ L"setup.exe --chrome-sxs", // User-level, secondary SxS canary mode. |
+ L"--uninstall --chrome-sxs", // Update. |
+ L"", // Unused. |
+ L"", // Unused. |
+ install_static::CANARY_INDEX, // Expect SxS canary mode. |
+ false, // Expect user-level. |
+ L"canary", // Expect canary channel. |
+ }, |
+ |
+ // System-level test cases. |
+ { |
+ L"setup.exe --system-level", // System-level, primary mode. |
+ L"", // New install. |
+ L"x64-stable", // Stable channel. |
+ L"1.1-beta", // Value ignored in binaries. |
+ install_static::STABLE_INDEX, // Expect primary mode. |
+ true, // Expect system-level. |
+ L"", // Expect stable channel. |
+ }, |
+ { |
+ L"setup.exe --system-level", // System-level, primary mode. |
+ L"--uninstall --system-level", // Updating an existing install. |
+ L"x64-stable", // Stable channel. |
+ L"1.1-beta", // Value ignored in binaries. |
+ install_static::STABLE_INDEX, // Expect primary mode. |
+ true, // Expect system-level. |
+ L"", // Expect stable channel. |
+ }, |
+ { |
+ L"setup.exe --system-level", // System-level, primary mode. |
+ // Updating an existing multi-install. |
+ L"--uninstall --system-level --multi-install", |
+ L"x64-stable", // Value ignored in product. |
+ L"1.1-beta", // Channel read from binaries. |
+ install_static::STABLE_INDEX, // Expect primary mode. |
+ true, // Expect system-level. |
+ L"beta", // Expect beta channel. |
+ }, |
+ { |
+ L"setup.exe --system-level", // System-level, primary mode. |
+ L"", // New install. |
+ L"1.1-beta", // Beta channel. |
+ L"x64-stable", // Value ignored in binaries. |
+ install_static::STABLE_INDEX, // Expect primary mode. |
+ true, // Expect system-level. |
+ L"beta", // Expect beta channel. |
+ }, |
+}; |
+#else // GOOGLE_CHROME_BUILD |
+constexpr TestData kTestData[] = { |
+ // User-level test cases. |
+ { |
+ L"setup.exe", // User-level, primary mode. |
+ L"", // New install. |
+ L"", // Channels are not supported. |
+ L"", // Channels are not supported. |
+ install_static::CHROMIUM_INDEX, // Expect primary mode. |
+ false, // Expect user-level. |
+ L"", // Expect empty channel. |
+ }, |
+ { |
+ L"setup.exe", // User-level, primary mode. |
+ L"--uninstall", // Updating an existing install. |
+ L"", // Channels are not supported. |
+ L"", // Channels are not supported. |
+ install_static::CHROMIUM_INDEX, // Expect primary mode. |
+ false, // Expect user-level. |
+ L"", // Expect empty channel. |
+ }, |
+ |
+ // System-level test cases. |
+ { |
+ L"setup.exe --system-level", // System-level, primary mode. |
+ L"", // New install. |
+ L"", // Channels are not supported. |
+ L"", // Channels are not supported. |
+ install_static::CHROMIUM_INDEX, // Expect primary mode. |
+ true, // Expect system-level. |
+ L"", // Expect empty channel. |
+ }, |
+ { |
+ L"setup.exe --system-level", // System-level, primary mode. |
+ L"--uninstall --system-level", // Updating an existing install. |
+ L"", // Channels are not supported. |
+ L"", // Channels are not supported. |
+ install_static::CHROMIUM_INDEX, // Expect primary mode. |
+ true, // Expect system-level. |
+ L"", // Expect empty channel. |
+ }, |
+}; |
+#endif // !GOOGLE_CHROME_BUILD |
+ |
+class MakeInstallDetailsTest : public testing::TestWithParam<TestData> { |
+ protected: |
+ MakeInstallDetailsTest() |
+ : test_data_(GetParam()), |
+ root_key_(test_data_.system_level ? HKEY_LOCAL_MACHINE |
+ : HKEY_CURRENT_USER), |
+ nt_root_key_(test_data_.system_level ? nt::HKLM : nt::HKCU), |
+ command_line_(base::CommandLine::NO_PROGRAM) { |
+ // Prepare the inputs from the process command line. |
+ command_line_.ParseFromString(test_data_.command_line); |
+ master_preferences_ = |
+ base::MakeUnique<installer::MasterPreferences>(command_line_); |
+ } |
+ |
+ void SetUp() override { |
+ base::string16 path; |
+ ASSERT_NO_FATAL_FAILURE( |
+ override_manager_.OverrideRegistry(root_key_, &path)); |
+ nt::SetTestingOverride(nt_root_key_, path); |
+ |
+ // Prepare the inputs from the machine's state. |
+ ASSERT_NO_FATAL_FAILURE(SetUninstallArguments( |
+ root_key_, install_static::kInstallModes[test_data_.index].app_guid, |
+ test_data_.uninstall_args)); |
+ if (install_static::kUseGoogleUpdateIntegration) { |
+ ASSERT_NO_FATAL_FAILURE(SetProductAp( |
+ root_key_, install_static::kInstallModes[test_data_.index].app_guid, |
+ test_data_.product_ap)); |
+ ASSERT_NO_FATAL_FAILURE(SetBinariesAp(root_key_, test_data_.binaries_ap)); |
+ } |
+ } |
+ |
+ void TearDown() override { |
+ nt::SetTestingOverride(nt_root_key_, base::string16()); |
+ } |
+ |
+ const TestData& test_data() const { return test_data_; } |
+ |
+ const base::CommandLine& command_line() const { return command_line_; } |
+ |
+ const installer::MasterPreferences& master_preferences() const { |
+ return *master_preferences_; |
+ } |
+ |
+ private: |
+ static void SetUninstallArguments(HKEY root_key, |
+ const wchar_t* app_guid, |
+ const wchar_t* uninstall_args) { |
+ // Do nothing if there's no value to write. |
+ if (!uninstall_args || !*uninstall_args) |
+ return; |
+ // Make it appear that the product is installed with the given uninstall |
+ // args. |
+ ASSERT_THAT( |
+ base::win::RegKey(root_key, |
+ install_static::GetClientsKeyPath(app_guid).c_str(), |
+ KEY_WOW64_32KEY | KEY_SET_VALUE) |
+ .WriteValue(L"pv", L"1.2.3.4"), |
+ Eq(ERROR_SUCCESS)); |
+ ASSERT_THAT( |
+ base::win::RegKey( |
+ root_key, install_static::GetClientStateKeyPath(app_guid).c_str(), |
+ KEY_WOW64_32KEY | KEY_SET_VALUE) |
+ .WriteValue(L"UninstallArguments", uninstall_args), |
+ Eq(ERROR_SUCCESS)); |
+ } |
+ |
+ static void SetProductAp(HKEY root_key, |
+ const wchar_t* app_guid, |
+ const wchar_t* ap) { |
+ // Do nothing if there's no value to write. |
+ if (!ap || !*ap) |
+ return; |
+ ASSERT_THAT( |
+ base::win::RegKey( |
+ root_key, install_static::GetClientStateKeyPath(app_guid).c_str(), |
+ KEY_WOW64_32KEY | KEY_SET_VALUE) |
+ .WriteValue(L"ap", ap), |
+ Eq(ERROR_SUCCESS)); |
+ } |
+ |
+ static void SetBinariesAp(HKEY root_key, const wchar_t* ap) { |
+ // Do nothing if there's no value to write. |
+ if (!ap || !*ap) |
+ return; |
+ ASSERT_THAT( |
+ base::win::RegKey( |
+ root_key, install_static::GetBinariesClientStateKeyPath().c_str(), |
+ KEY_WOW64_32KEY | KEY_SET_VALUE) |
+ .WriteValue(L"ap", ap), |
+ Eq(ERROR_SUCCESS)); |
+ } |
+ |
+ registry_util::RegistryOverrideManager override_manager_; |
+ const TestData& test_data_; |
+ HKEY root_key_; |
+ nt::ROOT_KEY nt_root_key_; |
+ base::CommandLine command_line_; |
+ std::unique_ptr<installer::MasterPreferences> master_preferences_; |
+ |
+ DISALLOW_COPY_AND_ASSIGN(MakeInstallDetailsTest); |
+}; |
+ |
+TEST_P(MakeInstallDetailsTest, Test) { |
+ std::unique_ptr<install_static::PrimaryInstallDetails> details( |
+ MakeInstallDetails(command_line(), master_preferences())); |
+ EXPECT_THAT(details->install_mode_index(), Eq(test_data().index)); |
+ EXPECT_THAT(details->system_level(), Eq(test_data().system_level)); |
+ EXPECT_THAT(details->channel(), Eq(test_data().channel)); |
+} |
+ |
+INSTANTIATE_TEST_CASE_P(All, |
+ MakeInstallDetailsTest, |
+ testing::ValuesIn(kTestData)); |