| 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_elf/nt_registry/nt_registry.h" | 5 #include "chrome_elf/nt_registry/nt_registry.h" |
| 6 | 6 |
| 7 #include <windows.h> | 7 #include <windows.h> |
| 8 #include <rpc.h> | 8 #include <rpc.h> |
| 9 #include <stddef.h> | 9 #include <stddef.h> |
| 10 | 10 |
| (...skipping 276 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 287 //------------------------------------------------------------------------------ | 287 //------------------------------------------------------------------------------ |
| 288 // NtRegistryTest class | 288 // NtRegistryTest class |
| 289 // | 289 // |
| 290 // Only use this class for tests that need testing registry redirection. | 290 // Only use this class for tests that need testing registry redirection. |
| 291 //------------------------------------------------------------------------------ | 291 //------------------------------------------------------------------------------ |
| 292 | 292 |
| 293 class NtRegistryTest : public testing::Test { | 293 class NtRegistryTest : public testing::Test { |
| 294 protected: | 294 protected: |
| 295 void SetUp() override { | 295 void SetUp() override { |
| 296 base::string16 temp; | 296 base::string16 temp; |
| 297 override_manager_.OverrideRegistry(HKEY_CURRENT_USER, &temp); | 297 ASSERT_NO_FATAL_FAILURE( |
| 298 override_manager_.OverrideRegistry(HKEY_CURRENT_USER, &temp)); |
| 298 ASSERT_TRUE(nt::SetTestingOverride(nt::HKCU, temp)); | 299 ASSERT_TRUE(nt::SetTestingOverride(nt::HKCU, temp)); |
| 299 override_manager_.OverrideRegistry(HKEY_LOCAL_MACHINE, &temp); | 300 ASSERT_NO_FATAL_FAILURE( |
| 301 override_manager_.OverrideRegistry(HKEY_LOCAL_MACHINE, &temp)); |
| 300 ASSERT_TRUE(nt::SetTestingOverride(nt::HKLM, temp)); | 302 ASSERT_TRUE(nt::SetTestingOverride(nt::HKLM, temp)); |
| 301 } | 303 } |
| 302 | 304 |
| 303 void TearDown() override { | 305 void TearDown() override { |
| 304 base::string16 temp; | 306 base::string16 temp; |
| 305 ASSERT_TRUE(nt::SetTestingOverride(nt::HKCU, temp)); | 307 ASSERT_TRUE(nt::SetTestingOverride(nt::HKCU, temp)); |
| 306 ASSERT_TRUE(nt::SetTestingOverride(nt::HKLM, temp)); | 308 ASSERT_TRUE(nt::SetTestingOverride(nt::HKLM, temp)); |
| 307 } | 309 } |
| 308 | 310 |
| 309 private: | 311 private: |
| (...skipping 146 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 456 nt::CloseRegKey(key_handle); | 458 nt::CloseRegKey(key_handle); |
| 457 | 459 |
| 458 // Subkey path can be null. | 460 // Subkey path can be null. |
| 459 ASSERT_TRUE(nt::CreateRegKey(nt::HKCU, nullptr, KEY_ALL_ACCESS, &key_handle)); | 461 ASSERT_TRUE(nt::CreateRegKey(nt::HKCU, nullptr, KEY_ALL_ACCESS, &key_handle)); |
| 460 ASSERT_NE(key_handle, INVALID_HANDLE_VALUE); | 462 ASSERT_NE(key_handle, INVALID_HANDLE_VALUE); |
| 461 ASSERT_NE(key_handle, nullptr); | 463 ASSERT_NE(key_handle, nullptr); |
| 462 nt::CloseRegKey(key_handle); | 464 nt::CloseRegKey(key_handle); |
| 463 } | 465 } |
| 464 | 466 |
| 465 } // namespace | 467 } // namespace |
| OLD | NEW |