1 // Copyright 2014 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.
5 #include "chrome_elf/chrome_elf_util.h"
9 #include "base/test/test_reg_util_win.h"
10 #include "base/win/registry.h"
11 #include "testing/gtest/include/gtest/gtest.h"
12 #include "testing/platform_test.h"
16 const wchar_t kRegPathClientState
[] = L
"Software\\Google\\Update\\ClientState";
17 const wchar_t kRegPathClientStateMedium
[] =
18 L
"Software\\Google\\Update\\ClientStateMedium";
19 const wchar_t kRegValueUsageStats
[] = L
"usagestats";
20 const wchar_t kUninstallArgumentsField
[] = L
"UninstallArguments";
22 const wchar_t kAppGuidCanary
[] =
23 L
"{4ea16ac7-fd5a-47c3-875b-dbf4a2008c20}";
24 const wchar_t kAppGuidGoogleChrome
[] =
25 L
"{8A69D345-D564-463c-AFF1-A69D9E530F96}";
26 const wchar_t kAppGuidGoogleBinaries
[] =
27 L
"{4DC8B4CA-1BDA-483e-B5FA-D3C12E15B62D}";
29 const wchar_t kCanaryExePath
[] =
30 L
"C:\\Users\\user\\AppData\\Local\\Google\\Chrome SxS\\Application"
32 const wchar_t kChromeSystemExePath
[] =
33 L
"C:\\Program Files (x86)\\Google\\Chrome\\Application\\chrome.exe";
34 const wchar_t kChromeUserExePath
[] =
35 L
"C:\\Users\\user\\AppData\\Local\\Google\\Chrome\\Application\\chrome.exe";
36 const wchar_t kChromiumExePath
[] =
37 L
"C:\\Users\\user\\AppData\\Local\\Chromium\\Application\\chrome.exe";
40 TEST(ChromeElfUtilTest
, CanaryTest
) {
41 EXPECT_TRUE(IsCanary(kCanaryExePath
));
42 EXPECT_FALSE(IsCanary(kChromeUserExePath
));
43 EXPECT_FALSE(IsCanary(kChromiumExePath
));
46 TEST(ChromeElfUtilTest
, SystemInstallTest
) {
47 EXPECT_TRUE(IsSystemInstall(kChromeSystemExePath
));
48 EXPECT_FALSE(IsSystemInstall(kChromeUserExePath
));
51 // Parameterized test with paramters:
52 // 1: product: "canary" or "google"
53 // 2: install level: "user" or "system"
54 // 3: install mode: "single" or "multi"
55 class ChromeElfUtilTest
:
56 public testing::TestWithParam
<std::tuple
<const char*,
60 void SetUp() override
{
61 override_manager_
.OverrideRegistry(HKEY_LOCAL_MACHINE
);
62 override_manager_
.OverrideRegistry(HKEY_CURRENT_USER
);
66 std::tie(app
, level
, mode
) = GetParam();
67 is_canary_
= (std::string(app
) == "canary");
68 system_level_
= (std::string(level
) != "user");
69 multi_install_
= (std::string(mode
) != "single");
71 ASSERT_FALSE(system_level_
);
72 ASSERT_FALSE(multi_install_
);
73 app_guid_
= kAppGuidCanary
;
74 chrome_path_
= kCanaryExePath
;
76 app_guid_
= kAppGuidGoogleChrome
;
77 chrome_path_
= (system_level_
? kChromeSystemExePath
:
81 SetMultiInstallStateInRegistry(system_level_
, true);
82 app_guid_
= kAppGuidGoogleBinaries
;
86 base::string16
BuildKey(const wchar_t* path
, const wchar_t* guid
) {
87 base::string16
full_key_path(path
);
88 full_key_path
.append(1, L
'\\');
89 full_key_path
.append(guid
);
93 void SetUsageStat(DWORD value
, bool state_medium
) {
94 LONG result
= base::win::RegKey(
95 system_level_
? HKEY_LOCAL_MACHINE
: HKEY_CURRENT_USER
,
96 BuildKey(state_medium
? kRegPathClientStateMedium
: kRegPathClientState
,
98 KEY_SET_VALUE
).WriteValue(kRegValueUsageStats
, value
);
99 ASSERT_EQ(ERROR_SUCCESS
, result
);
102 void SetMultiInstallStateInRegistry(bool system_install
, bool multi
) {
103 base::win::RegKey
key(
104 system_install
? HKEY_LOCAL_MACHINE
: HKEY_CURRENT_USER
,
105 BuildKey(kRegPathClientState
, kAppGuidGoogleChrome
).c_str(),
109 result
= key
.WriteValue(kUninstallArgumentsField
,
110 L
"yadda yadda --multi-install yadda yadda");
112 result
= key
.DeleteValue(kUninstallArgumentsField
);
114 ASSERT_EQ(ERROR_SUCCESS
, result
);
117 const wchar_t* app_guid_
;
118 const wchar_t* chrome_path_
;
122 registry_util::RegistryOverrideManager override_manager_
;
125 TEST_P(ChromeElfUtilTest
, MultiInstallTest
) {
128 SetMultiInstallStateInRegistry(system_level_
, true);
129 EXPECT_TRUE(IsMultiInstall(system_level_
));
131 SetMultiInstallStateInRegistry(system_level_
, false);
132 EXPECT_FALSE(IsMultiInstall(system_level_
));
135 TEST_P(ChromeElfUtilTest
, UsageStatsAbsent
) {
136 EXPECT_FALSE(AreUsageStatsEnabled(chrome_path_
));
139 TEST_P(ChromeElfUtilTest
, UsageStatsZero
) {
140 SetUsageStat(0, false);
141 EXPECT_FALSE(AreUsageStatsEnabled(chrome_path_
));
144 TEST_P(ChromeElfUtilTest
, UsageStatsOne
) {
145 SetUsageStat(1, false);
146 EXPECT_TRUE(AreUsageStatsEnabled(chrome_path_
));
148 EXPECT_FALSE(AreUsageStatsEnabled(kChromeUserExePath
));
149 EXPECT_FALSE(AreUsageStatsEnabled(kChromeSystemExePath
));
150 } else if (system_level_
) {
151 EXPECT_FALSE(AreUsageStatsEnabled(kCanaryExePath
));
152 EXPECT_FALSE(AreUsageStatsEnabled(kChromeUserExePath
));
154 EXPECT_FALSE(AreUsageStatsEnabled(kCanaryExePath
));
155 EXPECT_FALSE(AreUsageStatsEnabled(kChromeSystemExePath
));
159 TEST_P(ChromeElfUtilTest
, UsageStatsZeroInStateMedium
) {
162 SetUsageStat(0, true);
163 EXPECT_FALSE(AreUsageStatsEnabled(chrome_path_
));
166 TEST_P(ChromeElfUtilTest
, UsageStatsOneInStateMedium
) {
169 SetUsageStat(1, true);
170 EXPECT_TRUE(AreUsageStatsEnabled(chrome_path_
));
171 EXPECT_FALSE(AreUsageStatsEnabled(kCanaryExePath
));
172 EXPECT_FALSE(AreUsageStatsEnabled(kChromeUserExePath
));
175 INSTANTIATE_TEST_CASE_P(Canary
, ChromeElfUtilTest
,
176 testing::Combine(testing::Values("canary"),
177 testing::Values("user"),
178 testing::Values("single")));
179 INSTANTIATE_TEST_CASE_P(GoogleChrome
, ChromeElfUtilTest
,
180 testing::Combine(testing::Values("google"),
181 testing::Values("user", "system"),
182 testing::Values("single", "multi")));