Add test fixture for GPU browser tests which do image comparisons....

chrome/test/ui_test_utils.h

 // Copyright (c) 2010 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_TEST_UI_TEST_UTILS_H_
 #define CHROME_TEST_UI_TEST_UTILS_H_
 #pragma once
 
 #include <map>
+#include <queue>
 #include <string>
 #include <set>
 
 #include "app/keyboard_codes.h"
 #include "base/basictypes.h"
 #include "base/message_loop.h"
 #include "base/scoped_temp_dir.h"
 #include "base/string16.h"
 #include "chrome/browser/view_ids.h"
 #include "chrome/common/notification_observer.h"
 #include "chrome/common/notification_registrar.h"
 #include "chrome/common/notification_service.h"
 #include "chrome/test/automation/dom_element_proxy.h"
 #include "gfx/native_widget_types.h"
 
 class AppModalDialog;
 class BookmarkModel;
 class BookmarkNode;
 class Browser;
 class CommandLine;
 class DownloadManager;
 class ExtensionAction;
 class FilePath;
 class GURL;
 class MessageLoop;
 class NavigationController;
 class NotificationType;
 class Profile;
 class RenderViewHost;
+class RenderWidgetHost;
 class ScopedTempDir;
+class SkBitmap;
 class TabContents;
 class Value;
 
+namespace gfx {
+class Size;
+}
+
 // A collections of functions designed for use with InProcessBrowserTest.
 namespace ui_test_utils {
 
 // Turns on nestable tasks, runs the message loop, then resets nestable tasks
 // to what they were originally. Prefer this over MessageLoop::Run for in
 // process browser tests that need to block until a condition is met.
 void RunMessageLoop();
 
 // Turns on nestable tasks, runs all pending tasks in the message loop,
 // then resets nestable tasks to what they were originally. Prefer this
@@ skipping 362 matching lines @@
 
   DISALLOW_COPY_AND_ASSIGN(WindowedNotificationObserverWithDetails);
 };
 
 // Hide a native window.
 void HideNativeWindow(gfx::NativeWindow window);
 
 // Show and focus a native window.
 void ShowAndFocusNativeWindow(gfx::NativeWindow window);
 
+// Watches for responses from the DOMAutomationController and keeps them in a
+// queue. Useful for waiting for a message to be received.
+class DOMMessageQueue : public NotificationObserver {
+ public:
+  // Constructs a DOMMessageQueue and begins listening for messages from the
+  // DOMAutomationController. Do not construct this until the browser has
+  // started.
+  DOMMessageQueue();
+
+  // Wait for the next message to arrive. |message| will be set to the next
+  // message, if not null. Returns true on success.
+  bool WaitForMessage(std::string* message) WARN_UNUSED_RESULT;
+
+  // Overridden NotificationObserver methods.
+  virtual void Observe(NotificationType type,
+                       const NotificationSource& source,
+                       const NotificationDetails& details);
+
+ private:
+  NotificationRegistrar registrar_;
+  std::queue<std::string> message_queue_;
+  bool waiting_for_message_;
+
+  DISALLOW_COPY_AND_ASSIGN(DOMMessageQueue);
+};
+
+// Takes a snapshot of the given render widget, rendered at |page_size|. The
+// snapshot is set to |bitmap|. Returns true on success.
+bool TakeSnapshot(RenderWidgetHost* rwh,

[Paweł Hajdan Jr., 2010/11/11 10:33:36]

nit: The name is too generic. Should it be TakeRenderWidgetSnapshot?

[kkania, 2010/11/15 20:07:03]

Done.

+                  const gfx::Size& page_size,
+                  SkBitmap* bitmap) WARN_UNUSED_RESULT;
+
+// Takes a snapshot of the entire page, according to the width and height
+// properties of the DOM's document. Returns true on success. DOMAutomation
+// must be enabled.
+bool TakeEntirePageSnapshot(RenderViewHost* rvh,
+                            SkBitmap* bitmap) WARN_UNUSED_RESULT;
+
 }  // namespace ui_test_utils
 
 #endif  // CHROME_TEST_UI_TEST_UTILS_H_