Remove multi-install from chrome/installer/setup.

chrome/installer/setup/setup_util_unittest.cc

 // Copyright (c) 2012 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/installer/setup/setup_util_unittest.h"
 
 #include <windows.h>
 #include <shlobj.h>
 
 #include <memory>
 #include <string>
 
 #include "base/command_line.h"
 #include "base/files/file_util.h"
 #include "base/files/scoped_temp_dir.h"
 #include "base/macros.h"
+#include "base/memory/ptr_util.h"
 #include "base/process/kill.h"
 #include "base/process/launch.h"
 #include "base/process/process_handle.h"
 #include "base/strings/string_util.h"
 #include "base/test/histogram_tester.h"
 #include "base/test/test_reg_util_win.h"
 #include "base/test/test_timeouts.h"
 #include "base/threading/platform_thread.h"
 #include "base/version.h"
 #include "base/win/registry.h"
@@ skipping 478 matching lines @@
 }
 
 TEST_F(FindArchiveToPatchTest, DesiredVersionNotFound) {
   base::FilePath patch_source(installer::FindArchiveToPatch(
     *original_state_, *installer_state_, base::Version("1.2.3.4")));
   EXPECT_EQ(base::FilePath().value(), patch_source.value());
 }
 
 #if defined(GOOGLE_CHROME_BUILD)
 namespace {
+
 const bool kSystemLevel = false;
 const HKEY kRootKey = kSystemLevel ? HKEY_LOCAL_MACHINE : HKEY_CURRENT_USER;
 const wchar_t kVersionString[] = L"30.0.1574.0";
 const wchar_t kMultiChannel[] = L"2.0-dev-multi-chromeframe";
 
 class MigrateMultiToSingleTest : public testing::Test {
  protected:
   void SetUp() override {
     registry_override_manager_.OverrideRegistry(kRootKey);
   }
 
  private:
   registry_util::RegistryOverrideManager registry_override_manager_;
 };
+
 }  // namespace
 
 // Test migrating Chrome Frame from multi to single.
 TEST_F(MigrateMultiToSingleTest, ChromeFrame) {
   installer::ProductState chrome_frame;
   installer::ProductState binaries;
   DWORD usagestats = 0;
 
   // Set up a config with dev-channel multi-install GCF.
   base::win::RegKey key;
@@ skipping 39 matching lines @@
                                       BrowserDistribution::CHROME_FRAME));
   EXPECT_TRUE(chrome_frame.GetUsageStats(&usagestats));
   EXPECT_EQ(1U, usagestats);
   EXPECT_EQ(L"2.0-dev", chrome_frame.channel().value());
 
   // Confirm that the binaries' channel no longer contains GCF.
   ASSERT_TRUE(binaries.Initialize(kSystemLevel,
                                   BrowserDistribution::CHROME_BINARIES));
   EXPECT_EQ(L"2.0-dev-multi", binaries.channel().value());
 }
-#endif
+#endif  // GOOGLE_CHROME_BUILD
 
 TEST(SetupUtilTest, ContainsUnsupportedSwitch) {
   EXPECT_FALSE(installer::ContainsUnsupportedSwitch(
       base::CommandLine::FromString(L"foo.exe")));
   EXPECT_FALSE(installer::ContainsUnsupportedSwitch(
       base::CommandLine::FromString(L"foo.exe --multi-install --chrome")));
   EXPECT_TRUE(installer::ContainsUnsupportedSwitch(
       base::CommandLine::FromString(L"foo.exe --chrome-frame")));
 }
 
@@ skipping 112 matching lines @@
     }
   }
 
   // Ensure that all values are absent.
   {
     base::win::RegistryValueIterator it(root_, path_.c_str());
     ASSERT_EQ(0u, it.ValueCount());
   }
 }
 
+class LegacyCleanupsTest : public ::testing::Test {
+ protected:
+  LegacyCleanupsTest() = default;
+  void SetUp() override {
+    ASSERT_TRUE(temp_dir_.CreateUniqueTempDir());
+    registry_override_manager_.OverrideRegistry(HKEY_CURRENT_USER);
+    installer_state_ =
+        base::MakeUnique<FakeInstallerState>(temp_dir_.GetPath());
+    // Create the state to be cleared.
+    ASSERT_TRUE(base::win::RegKey(HKEY_CURRENT_USER, kBinariesClientsKeyPath,
+                                  KEY_WRITE | KEY_WOW64_32KEY)
+                    .Valid());
+#if defined(GOOGLE_CHROME_BUILD)
+    ASSERT_TRUE(base::win::RegKey(HKEY_CURRENT_USER, kGCFClientsKeyPath,
+                                  KEY_WRITE | KEY_WOW64_32KEY)
+                    .Valid());
+    ASSERT_TRUE(base::win::RegKey(HKEY_CURRENT_USER, kAppLauncherClientsKeyPath,
+                                  KEY_WRITE | KEY_WOW64_32KEY)
+                    .Valid());
+    ASSERT_GT(base::WriteFile(GetAppHostExePath(), "cha", 3), 0);
+    ASSERT_TRUE(
+        base::win::RegKey(HKEY_CURRENT_USER,
+                          GetChromeAppCommandPath(L"install-extension").c_str(),
+                          KEY_WRITE | KEY_WOW64_32KEY)
+            .Valid());
+#endif  // GOOGLE_CHROME_BUILD
+  }
+
+  const InstallerState& installer_state() const { return *installer_state_; }
+
+  bool HasBinariesVersionKey() const {
+    return base::win::RegKey(HKEY_CURRENT_USER, kBinariesClientsKeyPath,
+                             KEY_QUERY_VALUE | KEY_WOW64_32KEY)
+        .Valid();
+  }
+
+#if defined(GOOGLE_CHROME_BUILD)
+  bool HasMultiGCFVersionKey() const {
+    return base::win::RegKey(HKEY_CURRENT_USER, kGCFClientsKeyPath,
+                             KEY_QUERY_VALUE | KEY_WOW64_32KEY)
+        .Valid();
+  }
+
+  bool HasAppLauncherVersionKey() const {
+    return base::win::RegKey(HKEY_CURRENT_USER, kAppLauncherClientsKeyPath,
+                             KEY_QUERY_VALUE | KEY_WOW64_32KEY)
+        .Valid();
+  }
+
+  bool HasAppHostExe() const { return base::PathExists(GetAppHostExePath()); }
+
+  bool HasInstallExtensionCommand() const {
+    return base::win::RegKey(
+               HKEY_CURRENT_USER,
+               GetChromeAppCommandPath(L"install-extension").c_str(),
+               KEY_QUERY_VALUE | KEY_WOW64_32KEY)
+        .Valid();
+  }
+#endif  // GOOGLE_CHROME_BUILD
+
+ private:
+  // An InstallerState for a per-user install of Chrome in a given directory.
+  class FakeInstallerState : public InstallerState {
+   public:
+    explicit FakeInstallerState(const base::FilePath& target_path) {
+      BrowserDistribution* dist = BrowserDistribution::GetDistribution();
+      operation_ = InstallerState::SINGLE_INSTALL_OR_UPDATE;
+      target_path_ = target_path;
+      state_key_ = dist->GetStateKey();
+      state_type_ = dist->GetType();
+      product_ = base::MakeUnique<Product>(dist);
+      level_ = InstallerState::USER_LEVEL;
+      root_key_ = HKEY_CURRENT_USER;
+    }
+  };
+
+#if defined(GOOGLE_CHROME_BUILD)
+  base::FilePath GetAppHostExePath() const {
+    return installer_state_->target_path().AppendASCII("app_host.exe");
+  }
+
+  base::string16 GetChromeAppCommandPath(const wchar_t* command) const {
+    return base::string16(
+               L"SOFTWARE\\Google\\Update\\Clients\\"
+               L"{8A69D345-D564-463c-AFF1-A69D9E530F96}\\Commands\\") +
+           command;
+  }
+#endif  // GOOGLE_CHROME_BUILD
+
+  static const wchar_t kBinariesClientsKeyPath[];
+#if defined(GOOGLE_CHROME_BUILD)
+  static const wchar_t kGCFClientsKeyPath[];
+  static const wchar_t kAppLauncherClientsKeyPath[];
+#endif
+
+  base::ScopedTempDir temp_dir_;
+  registry_util::RegistryOverrideManager registry_override_manager_;
+  std::unique_ptr<FakeInstallerState> installer_state_;
+  DISALLOW_COPY_AND_ASSIGN(LegacyCleanupsTest);
+};
+
+#if defined(GOOGLE_CHROME_BUILD)
+const wchar_t LegacyCleanupsTest::kBinariesClientsKeyPath[] =
+    L"SOFTWARE\\Google\\Update\\Clients\\"
+    L"{4DC8B4CA-1BDA-483e-B5FA-D3C12E15B62D}";
+const wchar_t LegacyCleanupsTest::kGCFClientsKeyPath[] =
+    L"SOFTWARE\\Google\\Update\\Clients\\"
+    L"{8BA986DA-5100-405E-AA35-86F34A02ACBF}";
+const wchar_t LegacyCleanupsTest::kAppLauncherClientsKeyPath[] =
+    L"SOFTWARE\\Google\\Update\\Clients\\"
+    L"{FDA71E6F-AC4C-4a00-8B70-9958A68906BF}";
+#else   // GOOGLE_CHROME_BUILD
+const wchar_t LegacyCleanupsTest::kBinariesClientsKeyPath[] =
+    L"SOFTWARE\\Chromium Binaries";
+#endif  // !GOOGLE_CHROME_BUILD
+
+TEST_F(LegacyCleanupsTest, NoOpOnFailedUpdate) {
+  DoLegacyCleanups(installer_state(), INSTALL_FAILED);
+  EXPECT_TRUE(HasBinariesVersionKey());
+#if defined(GOOGLE_CHROME_BUILD)
+  EXPECT_TRUE(HasMultiGCFVersionKey());
+  EXPECT_TRUE(HasAppLauncherVersionKey());
+  EXPECT_TRUE(HasAppHostExe());
+  EXPECT_TRUE(HasInstallExtensionCommand());
+#endif  // GOOGLE_CHROME_BUILD
+}
+
+TEST_F(LegacyCleanupsTest, Do) {
+  DoLegacyCleanups(installer_state(), NEW_VERSION_UPDATED);
+  EXPECT_FALSE(HasBinariesVersionKey());
+#if defined(GOOGLE_CHROME_BUILD)
+  EXPECT_FALSE(HasMultiGCFVersionKey());
+  EXPECT_FALSE(HasAppLauncherVersionKey());
+  EXPECT_FALSE(HasAppHostExe());
+  EXPECT_FALSE(HasInstallExtensionCommand());
+#endif  // GOOGLE_CHROME_BUILD
+}
+
 }  // namespace installer