Revert 243967 "Use a Finch Experiment to control the Browser Bla..."

trunk/src/chrome_elf/blacklist/test/blacklist_test.cc

Index: trunk/src/chrome_elf/blacklist/test/blacklist_test.cc
===================================================================
--- trunk/src/chrome_elf/blacklist/test/blacklist_test.cc	(revision 244024)
+++ trunk/src/chrome_elf/blacklist/test/blacklist_test.cc	(working copy)
@@ -14,7 +14,6 @@
 #include "chrome_elf/blacklist/blacklist.h"
 #include "chrome_elf/blacklist/test/blacklist_test_main_dll.h"
 #include "testing/gtest/include/gtest/gtest.h"
-#include "version.h"  // NOLINT
 
 const wchar_t kTestDllName1[] = L"blacklist_test_dll_1.dll";
 const wchar_t kTestDllName2[] = L"blacklist_test_dll_2.dll";
@@ -38,6 +37,9 @@
   virtual void SetUp() {
     // Force an import from blacklist_test_main_dll.
     InitBlacklistTestDll();
+
+    // Ensure that the beacon state starts off cleared.
+    blacklist::ClearBeacon();
   }
 
   virtual void TearDown() {
@@ -50,50 +52,17 @@
   registry_util::RegistryOverrideManager override_manager;
   override_manager.OverrideRegistry(HKEY_CURRENT_USER, L"beacon_test");
 
-  // Ensure that the beacon state starts off enabled for this version.
-  DWORD blacklist_state = blacklist::BLACKLIST_ENABLED;
-  LONG result = ::RegSetKeyValue(HKEY_CURRENT_USER,
-                                 blacklist::kRegistryBeaconPath,
-                                 blacklist::kBeaconState,
-                                 REG_DWORD,
-                                 &blacklist_state,
-                                 sizeof(blacklist_state));
-  ASSERT_EQ(result, ERROR_SUCCESS);
+  // First call should succeed as the beacon is newly created.
+  EXPECT_TRUE(blacklist::CreateBeacon());
 
-  result = ::RegSetKeyValue(HKEY_CURRENT_USER,
-                            blacklist::kRegistryBeaconPath,
-                            blacklist::kBeaconVersion,
-                            REG_SZ,
-                            TEXT(CHROME_VERSION_STRING),
-                            sizeof(TEXT(CHROME_VERSION_STRING)));
-  ASSERT_EQ(result, ERROR_SUCCESS);
+  // Second call should fail indicating the beacon already existed.
+  EXPECT_FALSE(blacklist::CreateBeacon());
 
-  // First call should find the beacon and reset it.
-  EXPECT_TRUE(blacklist::ResetBeacon());
+  // First call should find the beacon and delete it.
+  EXPECT_TRUE(blacklist::ClearBeacon());
 
-  // First call should succeed as the beacon is enabled.
-  EXPECT_TRUE(blacklist::LeaveSetupBeacon());
-
-  // Second call should fail indicating the beacon wasn't set as enabled.
-  EXPECT_FALSE(blacklist::LeaveSetupBeacon());
-
-  // Resetting the beacon should work when setup beacon is present.
-  EXPECT_TRUE(blacklist::ResetBeacon());
-
-  // Change the version and ensure that the setup fails due to the version
-  // mismatch.
-  base::string16 different_version(L"other_version");
-  ASSERT_NE(different_version, TEXT(CHROME_VERSION_STRING));
-
-  result = ::RegSetKeyValue(HKEY_CURRENT_USER,
-                            blacklist::kRegistryBeaconPath,
-                            blacklist::kBeaconVersion,
-                            REG_SZ,
-                            different_version.c_str(),
-                            sizeof(different_version));
-  ASSERT_EQ(result, ERROR_SUCCESS);
-
-  EXPECT_FALSE(blacklist::LeaveSetupBeacon());
+  // Second call should fail to find the beacon and delete it.
+  EXPECT_FALSE(blacklist::ClearBeacon());
 }
 
 TEST_F(BlacklistTest, AddAndRemoveModules) {