headless: Implement screenshot capturing

headless/test/headless_browser_test.cc

 // Copyright 2016 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 "headless/test/headless_browser_test.h"
 
 #include "base/files/file_path.h"
 #include "base/memory/ptr_util.h"
 #include "base/message_loop/message_loop.h"
 #include "base/run_loop.h"
 #include "content/public/browser/browser_thread.h"
 #include "content/public/browser/render_process_host.h"
 #include "content/public/common/url_constants.h"
 #include "headless/lib/browser/headless_browser_impl.h"
 #include "headless/lib/headless_content_main_delegate.h"
 #include "headless/public/domains/network.h"
 #include "headless/public/domains/page.h"
 #include "headless/public/domains/runtime.h"
 #include "headless/public/headless_devtools_client.h"
 #include "headless/public/headless_devtools_target.h"
 #include "headless/public/headless_web_contents.h"
+#include "ui/gfx/geometry/size.h"
+#include "url/gurl.h"
 
 namespace headless {
 namespace {
 
 class WaitForLoadObserver : public page::Observer, public network::Observer {
  public:
   WaitForLoadObserver(HeadlessBrowserTest* browser_test,
                       HeadlessWebContents* web_contents)
       : browser_test_(browser_test),
         web_contents_(web_contents),
@@ skipping 134 matching lines @@
   EXPECT_FALSE(run_loop_);
   run_loop_ = base::WrapUnique(new base::RunLoop());
   run_loop_->Run();
   run_loop_ = nullptr;
 }
 
 void HeadlessBrowserTest::FinishAsynchronousTest() {
   run_loop_->Quit();
 }
 
+HeadlessAsyncDevTooledBrowserTest::HeadlessAsyncDevTooledBrowserTest()
+: web_contents_(nullptr),
+  devtools_client_(HeadlessDevToolsClient::Create()) {}
+
+HeadlessAsyncDevTooledBrowserTest::~HeadlessAsyncDevTooledBrowserTest() {}
+
+void HeadlessAsyncDevTooledBrowserTest::DevToolsTargetReady() {

[Sami, 2016/06/03 11:46:24]

Is this correctly indented? (git cl format should fix this for you).

[Eric Seckler, 2016/06/03 13:40:38]

Done.

+    EXPECT_TRUE(web_contents_->GetDevToolsTarget());
+    web_contents_->GetDevToolsTarget()->AttachClient(devtools_client_.get());
+    RunDevTooledTest();
+  }
+
+void HeadlessAsyncDevTooledBrowserTest::RunTest() {
+  web_contents_ =
+      browser()->CreateWebContents(GURL("about:blank"), gfx::Size(800, 600));
+  web_contents_->AddObserver(this);
+
+  RunAsynchronousTest();
+
+  web_contents_->GetDevToolsTarget()->DetachClient(devtools_client_.get());
+  web_contents_->RemoveObserver(this);
+  web_contents_->Close();
+  web_contents_ = nullptr;
+}
+
 }  // namespace headless