Resize DevTools target frames.

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 "base/json/json_reader.h"
 #include "content/public/test/browser_test.h"
 #include "headless/public/domains/browser.h"
+#include "headless/public/domains/emulation.h"
 #include "headless/public/domains/network.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/test/headless_browser_test.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #include "url/gurl.h"
 
 namespace headless {
 
 namespace {
 
 std::vector<HeadlessWebContents*> GetAllWebContents(HeadlessBrowser* browser) {
   std::vector<HeadlessWebContents*> result;
@@ skipping 179 matching lines @@
             .SetWidth(1)
             .SetHeight(1)
             .Build(),
         base::Bind(&BrowserDomainCreateAndDeletePageTest::OnCreateTargetResult,
                    base::Unretained(this)));
   }
 
   void OnCreateTargetResult(
       std::unique_ptr<browser::CreateTargetResult> result) {
     EXPECT_EQ(2u, GetAllWebContents(browser()).size());
+    EXPECT_TRUE(GetAllWebContents(browser()).back()->GetDevToolsTarget());
+
+    if (GetAllWebContents(browser()).size() < 2 ||
+        !GetAllWebContents(browser()).back()->GetDevToolsTarget()) {
+      OnGetFrameSizeResult(result->GetTargetId(), nullptr);
+    }
+
+    target_devtools_client_ = HeadlessDevToolsClient::Create();
+    GetAllWebContents(browser()).back()->GetDevToolsTarget()->AttachClient(
+        target_devtools_client_.get());
+    target_devtools_client_->GetEmulation()->GetExperimental()->GetFrameSize(
+        emulation::GetFrameSizeParams::Builder().Build(),
+        base::Bind(&BrowserDomainCreateAndDeletePageTest::OnGetFrameSizeResult,
+                   base::Unretained(this), result->GetTargetId()));
+  }
+
+  void OnGetFrameSizeResult(
+      std::string target_id,
+      std::unique_ptr<emulation::GetFrameSizeResult> result) {
+    if (result) {
+      EXPECT_EQ(1, result->GetWidth());
+      EXPECT_EQ(1, result->GetHeight());
+
+      GetAllWebContents(browser()).back()->GetDevToolsTarget()->DetachClient(
+          target_devtools_client_.get());
+    }
 
     devtools_client_->GetBrowser()->GetExperimental()->CloseTarget(
-        browser::CloseTargetParams::Builder()
-            .SetTargetId(result->GetTargetId())
-            .Build(),
+        browser::CloseTargetParams::Builder().SetTargetId(target_id).Build(),
         base::Bind(&BrowserDomainCreateAndDeletePageTest::OnCloseTargetResult,
                    base::Unretained(this)));
   }
 
   void OnCloseTargetResult(std::unique_ptr<browser::CloseTargetResult> result) {
     EXPECT_TRUE(result->GetSuccess());
     EXPECT_EQ(1u, GetAllWebContents(browser()).size());
     FinishAsynchronousTest();
   }
+
+ private:
+  std::unique_ptr<HeadlessDevToolsClient> target_devtools_client_;
 };
 
 HEADLESS_ASYNC_DEVTOOLED_TEST_F(BrowserDomainCreateAndDeletePageTest);
 
 class BrowserDomainCreateAndDeleteBrowserContextTest
     : public HeadlessAsyncDevTooledBrowserTest {
   void RunDevTooledTest() override {
     EXPECT_TRUE(embedded_test_server()->Start());
 
     EXPECT_EQ(1u, GetAllWebContents(browser()).size());
@@ skipping 375 matching lines @@
     FinishAsynchronousTest();
   }
 
  private:
   bool navigation_requested_ = false;
 };
 
 HEADLESS_ASYNC_DEVTOOLED_TEST_F(HeadlessDevToolsNavigationControlTest);
 
 }  // namespace headless