[headless] AddWindowTreeClient

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 "content/public/test/browser_test.h"
 #include "headless/public/headless_browser.h"
 #include "headless/public/headless_web_contents.h"
 #include "headless/test/headless_browser_test.h"
@@ skipping 21 matching lines @@
 
   bool navigation_succeeded() const { return navigation_succeeded_; }
 
  private:
   HeadlessWebContentsTest* browser_test_;  // Not owned.
   bool navigation_succeeded_;
 };
 
 IN_PROC_BROWSER_TEST_F(HeadlessWebContentsTest, Navigation) {
   EXPECT_TRUE(embedded_test_server()->Start());
-  std::unique_ptr<HeadlessWebContents> web_contents =
-      browser()->CreateWebContents(
-          embedded_test_server()->GetURL("/hello.html"), gfx::Size(800, 600));
+  HeadlessWebContents* web_contents = browser()->CreateWebContents(
+      embedded_test_server()->GetURL("/hello.html"), gfx::Size(800, 600));
   NavigationObserver observer(this);
   web_contents->AddObserver(&observer);
 
   RunAsynchronousTest();
 
   EXPECT_TRUE(observer.navigation_succeeded());
   web_contents->RemoveObserver(&observer);
+
+  std::vector<HeadlessWebContents*> all_web_contents =
+      browser()->GetAllWebContents();
+
+  EXPECT_EQ(all_web_contents.size(), static_cast<size_t>(1));
+  EXPECT_EQ(all_web_contents[0], web_contents);
+  EXPECT_TRUE(web_contents->IsDirectlyCreated());
+}
+
+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));
+  NavigationObserver observer(this);
+  web_contents->AddObserver(&observer);
+
+  RunAsynchronousTest();
+
+  EXPECT_TRUE(observer.navigation_succeeded());
+  web_contents->RemoveObserver(&observer);
+
+  std::vector<HeadlessWebContents*> all_web_contents =
+      browser()->GetAllWebContents();
+
+  EXPECT_EQ(all_web_contents.size(), static_cast<size_t>(2));
+
+  // Check that one WebContents was directly created, while other wasn't.
+  EXPECT_TRUE(all_web_contents[0]->IsDirectlyCreated() !=
+              all_web_contents[1]->IsDirectlyCreated());
 }
 
 }  // namespace headless