blob: 829c29e056b5db7c742d9e4c9fe614c6d0f85aad [file] [log] [blame]
// 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));