Index: trunk/src/chrome/browser/chrome_elf_init_unittest_win.cc |
=================================================================== |
--- trunk/src/chrome/browser/chrome_elf_init_unittest_win.cc (revision 244024) |
+++ trunk/src/chrome/browser/chrome_elf_init_unittest_win.cc (working copy) |
@@ -1,111 +0,0 @@ |
-// Copyright 2014 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/browser/chrome_elf_init_win.h" |
- |
-#include "base/basictypes.h" |
-#include "base/strings/string16.h" |
-#include "base/strings/utf_string_conversions.h" |
-#include "base/test/test_reg_util_win.h" |
-#include "chrome/common/chrome_version_info.h" |
-#include "chrome_elf/blacklist/blacklist.h" |
-#include "testing/gtest/include/gtest/gtest.h" |
- |
-class ChromeBlacklistTrialTest : public testing::Test { |
- protected: |
- ChromeBlacklistTrialTest() : |
- blacklist_registry_key_(HKEY_CURRENT_USER, |
- blacklist::kRegistryBeaconPath, |
- KEY_QUERY_VALUE | KEY_SET_VALUE) {} |
- |
- virtual void SetUp() OVERRIDE { |
- testing::Test::SetUp(); |
- |
- registry_util::RegistryOverrideManager override_manager; |
- override_manager.OverrideRegistry(HKEY_CURRENT_USER, |
- L"browser_blacklist_test"); |
- |
- |
- } |
- |
- DWORD GetBlacklistState() { |
- DWORD blacklist_state = blacklist::BLACKLIST_STATE_MAX; |
- blacklist_registry_key_.ReadValueDW(blacklist::kBeaconState, |
- &blacklist_state); |
- |
- return blacklist_state; |
- } |
- |
- base::string16 GetBlacklistVersion() { |
- base::string16 blacklist_version; |
- blacklist_registry_key_.ReadValue(blacklist::kBeaconVersion, |
- &blacklist_version); |
- |
- return blacklist_version; |
- } |
- |
- base::win::RegKey blacklist_registry_key_; |
- |
- private: |
- DISALLOW_COPY_AND_ASSIGN(ChromeBlacklistTrialTest); |
-}; |
- |
- |
-// Ensure that the default trial deletes any existing blacklist beacons. |
-TEST_F(ChromeBlacklistTrialTest, DefaultRun) { |
- // Set some dummy values as beacons. |
- blacklist_registry_key_.WriteValue(blacklist::kBeaconState, |
- blacklist::BLACKLIST_ENABLED); |
- blacklist_registry_key_.WriteValue(blacklist::kBeaconVersion, L"Data"); |
- |
- // This setup code should result in the default group, which should remove |
- // all the beacon values. |
- InitializeChromeElf(); |
- |
- // Ensure that invalid values are returned to indicate that the |
- // beacon values are gone. |
- ASSERT_EQ(blacklist::BLACKLIST_STATE_MAX, GetBlacklistState()); |
- ASSERT_EQ(base::string16(), GetBlacklistVersion()); |
-} |
- |
-TEST_F(ChromeBlacklistTrialTest, VerifyFirstRun) { |
- BrowserBlacklistBeaconSetup(); |
- |
- // Verify the state is properly set after the first run. |
- ASSERT_EQ(blacklist::BLACKLIST_ENABLED, GetBlacklistState()); |
- |
- chrome::VersionInfo version_info; |
- base::string16 version(base::UTF8ToUTF16(version_info.Version())); |
- ASSERT_EQ(version, GetBlacklistVersion()); |
-} |
- |
-TEST_F(ChromeBlacklistTrialTest, SetupFailed) { |
- // Set the registry to indicate that the blacklist setup is running, |
- // which means it failed to run correctly last time. |
- blacklist_registry_key_.WriteValue(blacklist::kBeaconState, |
- blacklist::BLACKLIST_SETUP_RUNNING); |
- |
- BrowserBlacklistBeaconSetup(); |
- |
- // Since the blacklist setup failed, it should now be disabled. |
- ASSERT_EQ(blacklist::BLACKLIST_DISABLED, GetBlacklistState()); |
-} |
- |
-TEST_F(ChromeBlacklistTrialTest, VersionChanged) { |
- // Mark the blacklist as disabled for an older version, so it should |
- // get enabled for this new version. |
- blacklist_registry_key_.WriteValue(blacklist::kBeaconVersion, |
- L"old_version"); |
- blacklist_registry_key_.WriteValue(blacklist::kBeaconState, |
- blacklist::BLACKLIST_DISABLED); |
- |
- BrowserBlacklistBeaconSetup(); |
- |
- // The beacon should now be marked as enabled for the current version. |
- ASSERT_EQ(blacklist::BLACKLIST_ENABLED, GetBlacklistState()); |
- |
- chrome::VersionInfo version_info; |
- base::string16 expected_version(base::UTF8ToUTF16(version_info.Version())); |
- ASSERT_EQ(expected_version, GetBlacklistVersion()); |
-} |