Cause new_chrome.exe to always be written if Chrome Frame is in use, resulting in updates being d...

chrome/installer/util/installer_state_unittest.cc

 // Copyright (c) 2011 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 <windows.h>
 
 #include <fstream>
 
 #include "base/base_paths.h"
 #include "base/command_line.h"
@@ skipping 29 matching lines @@
  protected:
 };
 
 // An installer state on which we can tweak the target path.
 class MockInstallerState : public InstallerState {
  public:
   MockInstallerState() : InstallerState() { }
   void set_target_path(const FilePath& target_path) {
     target_path_ = target_path;
   }
+  bool IsFileInUse(const FilePath& file) {

[grt (UTC plus 2), 2011/07/15 13:58:09]

nit: what do you think about making this static, too?

[robertshield, 2011/07/15 17:19:12]

Done.

+    return InstallerState::IsFileInUse(file);
+  }
 };
 
 // Simple function to dump some text into a new file.
 void CreateTextFile(const std::wstring& filename,
                     const std::wstring& contents) {
   std::ofstream file;
   file.open(filename.c_str());
   ASSERT_TRUE(file.is_open());
   file << contents;
   file.close();
@@ skipping 390 matching lines @@
   CommandLine cmd_line(
       CommandLine::FromString(L"setup.exe --multi-install --chrome"));
   MasterPreferences prefs(cmd_line);
   InstallerState installer_state;
   installer_state.Initialize(cmd_line, prefs, machine_state);
 
   // Is the Chrome version picked up?
   scoped_ptr<Version> version(installer_state.GetCurrentVersion(machine_state));
   EXPECT_TRUE(version.get() != NULL);
 }
+
+TEST_F(InstallerStateTest, IsFileInUse) {
+  ScopedTempDir temp_dir;
+  ASSERT_TRUE(temp_dir.CreateUniqueTempDir());
+
+  FilePath temp_file;
+  ASSERT_TRUE(file_util::CreateTemporaryFileInDir(temp_dir.path(), &temp_file));
+
+  MockInstallerState mock_installer_state;
+  EXPECT_FALSE(mock_installer_state.IsFileInUse(temp_file));
+
+  {
+    // Open a handle to the file with the same access mode and sharing options
+    // as the loader.
+    base::win::ScopedHandle temp_handle(
+        CreateFile(temp_file.value().c_str(),
+                   SYNCHRONIZE | FILE_EXECUTE,
+                   FILE_SHARE_DELETE | FILE_SHARE_READ,
+                   NULL, OPEN_EXISTING, 0, 0));
+    ASSERT_TRUE(temp_handle != NULL);
+
+    // The file should now be in use.
+    EXPECT_TRUE(mock_installer_state.IsFileInUse(temp_file));
+  }
+
+  // And once the handle is gone, it should no longer be in use.
+  EXPECT_FALSE(mock_installer_state.IsFileInUse(temp_file));
+}