headless: Introduce a browser context

headless/lib/headless_web_contents_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 <string>
 #include <vector>
 
 #include "content/public/test/browser_test.h"
 #include "headless/public/domains/page.h"
 #include "headless/public/headless_browser.h"
 #include "headless/public/headless_devtools_client.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 {
 
 class HeadlessWebContentsTest : public HeadlessBrowserTest {};
 
 IN_PROC_BROWSER_TEST_F(HeadlessWebContentsTest, Navigation) {
   EXPECT_TRUE(embedded_test_server()->Start());
 
-  HeadlessWebContents* web_contents = browser()->CreateWebContents(
-      embedded_test_server()->GetURL("/hello.html"), gfx::Size(800, 600));
+  HeadlessWebContents* web_contents =
+      browser()
+          ->CreateWebContentsBuilder()
+          .SetInitialURL(embedded_test_server()->GetURL("/hello.html"))
+          .Build();
   EXPECT_TRUE(WaitForLoad(web_contents));
 
   std::vector<HeadlessWebContents*> all_web_contents =
       browser()->GetAllWebContents();
 
   EXPECT_EQ(static_cast<size_t>(1), all_web_contents.size());
   EXPECT_EQ(web_contents, all_web_contents[0]);
 }
 
 IN_PROC_BROWSER_TEST_F(HeadlessWebContentsTest, WindowOpen) {
   EXPECT_TRUE(embedded_test_server()->Start());
 
-  HeadlessWebContents* web_contents = browser()->CreateWebContents(
-      embedded_test_server()->GetURL("/window_open.html"), gfx::Size(800, 600));
+  HeadlessWebContents* web_contents =
+      browser()
+          ->CreateWebContentsBuilder()
+          .SetInitialURL(embedded_test_server()->GetURL("/window_open.html"))
+          .Build();
   EXPECT_TRUE(WaitForLoad(web_contents));
 
   std::vector<HeadlessWebContents*> all_web_contents =
       browser()->GetAllWebContents();
 
   EXPECT_EQ(static_cast<size_t>(2), all_web_contents.size());
 }
 
 class HeadlessWebContentsScreenshotTest
     : public HeadlessAsyncDevTooledBrowserTest {
  public:
   void RunDevTooledTest() override {
     devtools_client_->GetPage()->GetExperimental()->CaptureScreenshot(
         page::CaptureScreenshotParams::Builder().Build(),
         base::Bind(&HeadlessWebContentsScreenshotTest::OnScreenshotCaptured,
                    base::Unretained(this)));
   }
 
   void OnScreenshotCaptured(
       std::unique_ptr<page::CaptureScreenshotResult> result) {
     EXPECT_LT(0U, result->GetData().length());
     FinishAsynchronousTest();
   }
 };
 
 HEADLESS_ASYNC_DEVTOOLED_TEST_F(HeadlessWebContentsScreenshotTest);
 
 }  // namespace headless