Attempt to fix flake in FullTabDownloadTest.CF_DownloadFileFromPost.

chrome_frame/test/chrome_frame_test_utils.h

 // 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.
 
 #ifndef CHROME_FRAME_TEST_CHROME_FRAME_TEST_UTILS_H_
 #define CHROME_FRAME_TEST_CHROME_FRAME_TEST_UTILS_H_
 
 #include <windows.h>
 
 #include <atlbase.h>
 #include <atlwin.h>
 
 #include <string>
 
 #include "base/basictypes.h"
 #include "base/cancelable_callback.h"
 #include "base/compiler_specific.h"
 #include "base/files/file_path.h"
 #include "base/memory/scoped_ptr.h"
 #include "base/message_loop.h"
 #include "base/process_util.h"
 #include "base/run_loop.h"
 #include "base/test/test_reg_util_win.h"
 #include "base/time.h"
 #include "base/time.h"
 #include "base/win/registry.h"
 #include "base/win/scoped_comptr.h"
-#include "chrome/test/base/ui_test_utils.h"
 #include "chrome_frame/chrome_tab.h"
 #include "chrome_frame/test/simulate_input.h"
 #include "chrome_frame/test_utils.h"
 #include "chrome_frame/utils.h"
 
 #include "gtest/gtest.h"
 
 // Needed for CreateFunctor.
 #define GMOCK_MUTANT_INCLUDE_LATE_OBJECT_BINDING
 #include "testing/gmock_mutant.h"
@@ skipping 12 matching lines @@
 base::ProcessHandle LaunchChrome(const std::wstring& url,
                                  const base::FilePath& user_data_dir);
 
 // Attempts to close all open IE windows.
 // The return value is the number of windows closed.
 // @note: this function requires COM to be initialized on the calling thread.
 // Since the caller might be running in either MTA or STA, the function does
 // not perform this initialization itself.
 int CloseAllIEWindows();
 
+// Saves a screen snapshot to the desktop and logs its location.
+bool TakeSnapshotAndLog();
+
 extern const wchar_t kIEImageName[];
 extern const wchar_t kIEBrokerImageName[];
 extern const char kChromeImageName[];
 extern const wchar_t kChromeLauncher[];
 extern const base::TimeDelta kChromeFrameLongNavigationTimeout;
 extern const base::TimeDelta kChromeFrameVeryLongNavigationTimeout;
 
 // Temporarily impersonate the current thread to low integrity for the lifetime
 // of the object. Destructor will automatically revert integrity level.
 class LowIntegrityToken {
@@ skipping 168 matching lines @@
   bool WasTimedOut() const {
     return !quit_loop_invoked_;
   }
 
   void RunUntilIdle() {
     loop_.RunUntilIdle();
   }
 
  private:
   static void SnapshotAndQuit() {
-    base::FilePath snapshot;
-    if (ui_test_utils::SaveScreenSnapshotToDesktop(&snapshot)) {
-      testing::UnitTest* unit_test = testing::UnitTest::GetInstance();
-      const testing::TestInfo* test_info = unit_test->current_test_info();
-      std::string name;
-      if (test_info != NULL) {
-        name.append(test_info->test_case_name())
-            .append(1, '.')
-            .append(test_info->name());
-      } else {
-        name = "unknown test";
-      }
-      LOG(ERROR) << name << " timed out. Screen snapshot saved to "
-                 << snapshot.value();
-    }
+    TakeSnapshotAndLog();
     MessageLoop::current()->Quit();
   }
 
   MessageLoopForUI loop_;
   base::CancelableClosure timeout_closure_;
   bool snapshot_on_timeout_;
   bool quit_loop_invoked_;
 };
 
 // Saves typing. It's somewhat hard to create a wrapper around
@@ skipping 103 matching lines @@
 }  // namespace chrome_frame_test
 
 // TODO(tommi): This is a temporary workaround while we're getting our
 // Singleton story straight.  Ideally each test should clear up any singletons
 // it might have created, but test cases do not implicitly have their own
 // AtExitManager, so we have this workaround method for tests that depend on
 // "fresh" singletons.  The implementation is in chrome_frame_unittest_main.cc.
 void DeleteAllSingletons();
 
 #endif  // CHROME_FRAME_TEST_CHROME_FRAME_TEST_UTILS_H_