headless: Implement client API generation

headless/lib/headless_devtools_client_browsertest.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 <memory>
+
+#include "content/public/test/browser_test.h"
+#include "headless/public/domains/page.h"
+#include "headless/public/domains/runtime.h"
+#include "headless/public/headless_browser.h"
+#include "headless/public/headless_devtools_client.h"
+#include "headless/public/headless_devtools_target.h"
+#include "headless/public/headless_web_contents.h"
+#include "headless/test/headless_browser_test.h"
+#include "testing/gtest/include/gtest/gtest.h"
+#include "ui/gfx/geometry/size.h"
+#include "url/gurl.h"
+
+namespace headless {
+
+#define DEVTOOLS_CLIENT_TEST_F(TEST_FIXTURE_NAME)                        \
+  IN_PROC_BROWSER_TEST_F(TEST_FIXTURE_NAME, RunAsyncTest) { RunTest(); } \
+  class AsyncHeadlessBrowserTestNeedsSemicolon##TEST_FIXTURE_NAME {}
+
+// A test fixture which attaches a devtools client before starting the test.
+class HeadlessDevToolsClientTest : public HeadlessBrowserTest,
+                                   public HeadlessWebContents::Observer {
+ public:
+  HeadlessDevToolsClientTest()
+      : devtools_client_(HeadlessDevToolsClient::Create()) {}
+  ~HeadlessDevToolsClientTest() override {}
+
+  // HeadlessWebContentsObserver implementation:
+  void DevToolsTargetReady() override {
+    EXPECT_TRUE(web_contents_->GetDevToolsTarget());
+    web_contents_->GetDevToolsTarget()->AttachClient(devtools_client_.get());
+    RunDevToolsClientTest();
+  }
+
+  virtual void RunDevToolsClientTest() = 0;
+
+ protected:
+  void 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_ = nullptr;
+  }
+
+  std::unique_ptr<HeadlessWebContents> web_contents_;
+  std::unique_ptr<HeadlessDevToolsClient> devtools_client_;
+};
+
+class HeadlessDevToolsClientNavigationTest : public HeadlessDevToolsClientTest {
+ public:
+  void RunDevToolsClientTest() override {
+    EXPECT_TRUE(embedded_test_server()->Start());
+    std::unique_ptr<page::NavigateParams> params =
+        page::NavigateParams::Builder()
+            .SetUrl(embedded_test_server()->GetURL("/hello.html").spec())
+            .Build();
+    devtools_client_->GetPage()->Navigate(std::move(params));
+  }
+
+  // TODO(skyostil): Wait for a load event once we support them.
+  void DidFinishNavigation(bool success) override {
+    EXPECT_TRUE(success);
+    FinishAsynchronousTest();
+  }
+};
+
+DEVTOOLS_CLIENT_TEST_F(HeadlessDevToolsClientNavigationTest);
+
+class HeadlessDevToolsClientEvalTest : public HeadlessDevToolsClientTest {
+ public:
+  void RunDevToolsClientTest() override {
+    std::unique_ptr<runtime::EvaluateParams> params =
+        runtime::EvaluateParams::Builder().SetExpression("1 + 2").Build();
+    devtools_client_->GetRuntime()->Evaluate(
+        std::move(params), base::Bind(&HeadlessDevToolsClientEvalTest::OnResult,
+                                      base::Unretained(this)));
+  }
+
+  void OnResult(std::unique_ptr<runtime::EvaluateResult> result) {
+    int value;
+    EXPECT_TRUE(result->GetResult()->HasValue());
+    EXPECT_TRUE(result->GetResult()->GetValue()->GetAsInteger(&value));
+    EXPECT_EQ(3, value);
+    FinishAsynchronousTest();
+  }
+};
+
+DEVTOOLS_CLIENT_TEST_F(HeadlessDevToolsClientEvalTest);
+
+}  // namespace headless