OLD | NEW |
1 // Copyright 2013 The Chromium Authors. All rights reserved. | 1 // Copyright 2013 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 <stddef.h> | 5 #include <stddef.h> |
6 | 6 |
7 #include <memory> | 7 #include <memory> |
8 | 8 |
9 #include "base/environment.h" | 9 #include "base/environment.h" |
10 #include "base/files/file.h" | 10 #include "base/files/file.h" |
(...skipping 173 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
184 return; | 184 return; |
185 | 185 |
186 // We have to call this exported function every time this test setup runs. | 186 // We have to call this exported function every time this test setup runs. |
187 // If the tests are running in single process mode, the test DLL does not | 187 // If the tests are running in single process mode, the test DLL does not |
188 // get reloaded everytime - but we need to make sure it updates | 188 // get reloaded everytime - but we need to make sure it updates |
189 // appropriately. | 189 // appropriately. |
190 InitTestDll(); | 190 InitTestDll(); |
191 | 191 |
192 blacklist_registry_key_.reset( | 192 blacklist_registry_key_.reset( |
193 new base::win::RegKey(HKEY_CURRENT_USER, | 193 new base::win::RegKey(HKEY_CURRENT_USER, |
194 blacklist::kRegistryBeaconPath, | 194 install_static::GetRegistryPath() |
| 195 .append(kRegistryBeaconKeyName) |
| 196 .c_str(), |
195 KEY_QUERY_VALUE | KEY_SET_VALUE)); | 197 KEY_QUERY_VALUE | KEY_SET_VALUE)); |
196 | 198 |
197 // Find out how many dlls were blocked before the test starts. | 199 // Find out how many dlls were blocked before the test starts. |
198 TestDll_SuccessfullyBlocked(NULL, &num_initially_blocked_); | 200 TestDll_SuccessfullyBlocked(NULL, &num_initially_blocked_); |
199 } | 201 } |
200 | 202 |
201 void TearDown() override { | 203 void TearDown() override { |
202 TestDll_RemoveDllFromBlacklist(kTestDllName1); | 204 TestDll_RemoveDllFromBlacklist(kTestDllName1); |
203 TestDll_RemoveDllFromBlacklist(kTestDllName2); | 205 TestDll_RemoveDllFromBlacklist(kTestDllName2); |
204 TestDll_RemoveDllFromBlacklist(kTestDllName3); | 206 TestDll_RemoveDllFromBlacklist(kTestDllName3); |
(...skipping 175 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
380 &blacklist_state); | 382 &blacklist_state); |
381 EXPECT_EQ(blacklist_state, blacklist::BLACKLIST_SETUP_RUNNING); | 383 EXPECT_EQ(blacklist_state, blacklist::BLACKLIST_SETUP_RUNNING); |
382 | 384 |
383 DWORD attempt_count = blacklist::kBeaconMaxAttempts; | 385 DWORD attempt_count = blacklist::kBeaconMaxAttempts; |
384 blacklist_registry_key_->ReadValueDW(blacklist::kBeaconAttemptCount, | 386 blacklist_registry_key_->ReadValueDW(blacklist::kBeaconAttemptCount, |
385 &attempt_count); | 387 &attempt_count); |
386 EXPECT_EQ(static_cast<DWORD>(0), attempt_count); | 388 EXPECT_EQ(static_cast<DWORD>(0), attempt_count); |
387 } | 389 } |
388 | 390 |
389 } // namespace | 391 } // namespace |
OLD | NEW |