Add testRunner.capturePixelsThen() method

content/shell/renderer/test_runner/test_runner.h

 // Copyright 2014 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 CONTENT_SHELL_RENDERER_TEST_RUNNER_TEST_RUNNER_H_
 #define CONTENT_SHELL_RENDERER_TEST_RUNNER_TEST_RUNNER_H_
 
 #include <deque>
 #include <set>
 #include <string>
 
 #include "base/memory/scoped_ptr.h"
 #include "base/memory/weak_ptr.h"
 #include "content/shell/renderer/test_runner/WebTask.h"
 #include "content/shell/renderer/test_runner/web_test_runner.h"
 #include "v8/include/v8.h"
 
+class SkBitmap;
+
 namespace blink {
 class WebFrame;
 class WebNotificationPresenter;
 class WebPermissionClient;
 class WebString;
 class WebView;
 }
 
 namespace gin {
 class ArrayBufferView;
@@ skipping 496 matching lines @@
   bool WasMockSpeechRecognitionAborted();
 
   // WebPageOverlay related functions. Permits the adding and removing of only
   // one opaque overlay.
   void AddWebPageOverlay();
   void RemoveWebPageOverlay();
 
   void DisplayAsync();
   void DisplayAsyncThen(v8::Handle<v8::Function> callback);
 
+  // Similar to DisplayAsyncThen(), but pass parameters of the captured
+  // snapshot (width, height, snapshot) to the callback.
+  void CapturePixelsAsyncThen(v8::Handle<v8::Function> callback);
+
   void SetMockPushClientSuccess(const std::string& end_point,
                                 const std::string& registration_id);
   void SetMockPushClientError(const std::string& message);
 
   ///////////////////////////////////////////////////////////////////////////
   // Internal helpers
+
+  void CapturePixelsCallback(scoped_ptr<InvokeCallbackTask> task,
+                             const SkBitmap& snapshot);
+
   void CheckResponseMimeType();
   void CompleteNotifyDone();
 
   void DidAcquirePointerLockInternal();
   void DidNotAcquirePointerLockInternal();
   void DidLosePointerLockInternal();
 
   // In the Mac code, this is called to trigger the end of a test after the
   // page has finished loading. From here, we can generate the dump for the
   // test.
@@ skipping 189 matching lines @@
   bool use_mock_theme_;
 
   base::WeakPtrFactory<TestRunner> weak_factory_;
 
   DISALLOW_COPY_AND_ASSIGN(TestRunner);
 };
 
 }  // namespace content
 
 #endif  // CONTENT_SHELL_RENDERER_TEST_RUNNER_TEST_RUNNER_H_