| OLD | NEW |
| 1 // Copyright 2016 The Chromium Authors. All rights reserved. | 1 // Copyright 2016 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "chrome/install_static/install_util.h" | 5 #include "chrome/install_static/install_util.h" |
| 6 | 6 |
| 7 #include <tuple> | 7 #include <tuple> |
| 8 #include <utility> | 8 #include <utility> |
| 9 | 9 |
| 10 #include "base/memory/ptr_util.h" | 10 #include "base/memory/ptr_util.h" |
| (...skipping 255 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 266 | 266 |
| 267 class InstallStaticUtilTest | 267 class InstallStaticUtilTest |
| 268 : public ::testing::TestWithParam< | 268 : public ::testing::TestWithParam< |
| 269 std::tuple<InstallConstantIndex, const char*>> { | 269 std::tuple<InstallConstantIndex, const char*>> { |
| 270 protected: | 270 protected: |
| 271 InstallStaticUtilTest() | 271 InstallStaticUtilTest() |
| 272 : system_level_(std::string(std::get<1>(GetParam())) != "user"), | 272 : system_level_(std::string(std::get<1>(GetParam())) != "user"), |
| 273 scoped_install_details_(system_level_, std::get<0>(GetParam())), | 273 scoped_install_details_(system_level_, std::get<0>(GetParam())), |
| 274 mode_(&InstallDetails::Get().mode()), | 274 mode_(&InstallDetails::Get().mode()), |
| 275 root_key_(system_level_ ? HKEY_LOCAL_MACHINE : HKEY_CURRENT_USER), | 275 root_key_(system_level_ ? HKEY_LOCAL_MACHINE : HKEY_CURRENT_USER), |
| 276 nt_root_key_(system_level_ ? nt::HKLM : nt::HKCU) { | 276 nt_root_key_(system_level_ ? nt::HKLM : nt::HKCU) {} |
| 277 |
| 278 void SetUp() override { |
| 279 ASSERT_TRUE(!system_level_ || mode_->supports_system_level); |
| 277 base::string16 path; | 280 base::string16 path; |
| 278 override_manager_.OverrideRegistry(root_key_, &path); | 281 ASSERT_NO_FATAL_FAILURE( |
| 282 override_manager_.OverrideRegistry(root_key_, &path)); |
| 279 nt::SetTestingOverride(nt_root_key_, path); | 283 nt::SetTestingOverride(nt_root_key_, path); |
| 280 } | 284 } |
| 281 | 285 |
| 282 ~InstallStaticUtilTest() { | 286 void TearDown() override { |
| 283 nt::SetTestingOverride(nt_root_key_, base::string16()); | 287 nt::SetTestingOverride(nt_root_key_, base::string16()); |
| 284 } | 288 } |
| 285 | 289 |
| 286 void SetUp() override { | |
| 287 ASSERT_TRUE(!system_level_ || mode_->supports_system_level); | |
| 288 } | |
| 289 | |
| 290 bool system_level() const { return system_level_; } | 290 bool system_level() const { return system_level_; } |
| 291 | 291 |
| 292 const wchar_t* default_channel() const { return mode_->default_channel_name; } | 292 const wchar_t* default_channel() const { return mode_->default_channel_name; } |
| 293 | 293 |
| 294 void SetUsageStat(DWORD value, bool medium) { | 294 void SetUsageStat(DWORD value, bool medium) { |
| 295 ASSERT_TRUE(!medium || system_level_); | 295 ASSERT_TRUE(!medium || system_level_); |
| 296 ASSERT_EQ(ERROR_SUCCESS, | 296 ASSERT_EQ(ERROR_SUCCESS, |
| 297 base::win::RegKey(root_key_, GetUsageStatsKeyPath(medium).c_str(), | 297 base::win::RegKey(root_key_, GetUsageStatsKeyPath(medium).c_str(), |
| 298 KEY_SET_VALUE | KEY_WOW64_32KEY) | 298 KEY_SET_VALUE | KEY_WOW64_32KEY) |
| 299 .WriteValue(L"usagestats", value)); | 299 .WriteValue(L"usagestats", value)); |
| (...skipping 117 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 417 testing::Values("user"))); | 417 testing::Values("user"))); |
| 418 #else // GOOGLE_CHROME_BUILD | 418 #else // GOOGLE_CHROME_BUILD |
| 419 // Chromium supports user and system levels. | 419 // Chromium supports user and system levels. |
| 420 INSTANTIATE_TEST_CASE_P(Chromium, | 420 INSTANTIATE_TEST_CASE_P(Chromium, |
| 421 InstallStaticUtilTest, | 421 InstallStaticUtilTest, |
| 422 testing::Combine(testing::Values(CHROMIUM_INDEX), | 422 testing::Combine(testing::Values(CHROMIUM_INDEX), |
| 423 testing::Values("user", "system"))); | 423 testing::Values("user", "system"))); |
| 424 #endif // !GOOGLE_CHROME_BUILD | 424 #endif // !GOOGLE_CHROME_BUILD |
| 425 | 425 |
| 426 } // namespace install_static | 426 } // namespace install_static |
| OLD | NEW |