HeadlessWebContents:Observer to observe render process exit status

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/devtools/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 "testing/gtest/include/gtest/gtest.h"
 #include "ui/gfx/geometry/size.h"
 #include "url/gurl.h"
 
 namespace headless {
 namespace {
 
 class SynchronousLoadObserver {
  public:
   SynchronousLoadObserver(HeadlessBrowserTest* browser_test,
                           HeadlessWebContents* web_contents)
@@ skipping 147 matching lines @@
   run_loop_->Run();
   run_loop_ = nullptr;
 }
 
 void HeadlessBrowserTest::FinishAsynchronousTest() {
   run_loop_->Quit();
 }
 
 HeadlessAsyncDevTooledBrowserTest::HeadlessAsyncDevTooledBrowserTest()
     : web_contents_(nullptr),
-      devtools_client_(HeadlessDevToolsClient::Create()) {}
+      devtools_client_(HeadlessDevToolsClient::Create()),
+      render_process_exited_(false) {}
 
 HeadlessAsyncDevTooledBrowserTest::~HeadlessAsyncDevTooledBrowserTest() {}
 
 void HeadlessAsyncDevTooledBrowserTest::DevToolsTargetReady() {
   EXPECT_TRUE(web_contents_->GetDevToolsTarget());
   web_contents_->GetDevToolsTarget()->AttachClient(devtools_client_.get());
   RunDevTooledTest();
 }
 
+void HeadlessAsyncDevTooledBrowserTest::RenderProcessExited(
+    base::TerminationStatus status,
+    int exit_code) {
+  if (status == base::TERMINATION_STATUS_NORMAL_TERMINATION)
+    return;
+
+  FAIL() << "Abnormal renderer termination";
+  FinishAsynchronousTest();
+  render_process_exited_ = true;
+}
+
 void HeadlessAsyncDevTooledBrowserTest::RunTest() {
   browser_context_ = browser()->CreateBrowserContextBuilder().Build();
 
   web_contents_ = browser_context_->CreateWebContentsBuilder().Build();
   web_contents_->AddObserver(this);
 
   RunAsynchronousTest();
 
-  web_contents_->GetDevToolsTarget()->DetachClient(devtools_client_.get());
+  if (!render_process_exited_)
+    web_contents_->GetDevToolsTarget()->DetachClient(devtools_client_.get());
   web_contents_->RemoveObserver(this);
   web_contents_->Close();
   web_contents_ = nullptr;
   browser_context_->Close();
   browser_context_ = nullptr;
 }
 
 }  // namespace headless