Add Page.stopLoading to devtools and --timeout switch to headless_shell

headless/app/headless_shell.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 <iostream>
 #include <memory>
 #include <string>
 
 #include "base/base64.h"
 #include "base/bind.h"
@@ skipping 145 matching lines @@
           << "Expected an integer value for --virtual-time-budget=";
       devtools_client_->GetEmulation()->GetExperimental()->SetVirtualTimePolicy(
           emulation::SetVirtualTimePolicyParams::Builder()
               .SetPolicy(emulation::VirtualTimePolicy::
                              PAUSE_IF_NETWORK_FETCHES_PENDING)
               .SetBudget(budget_ms)
               .Build());
     } else {
       PollReadyState();
     }
+
+    if (base::CommandLine::ForCurrentProcess()->HasSwitch(switches::kTimeout)) {
+      std::string timeout_ms_ascii =
+          base::CommandLine::ForCurrentProcess()->GetSwitchValueASCII(
+              switches::kTimeout);
+      int timeout_ms;
+      CHECK(base::StringToInt(timeout_ms_ascii, &timeout_ms))
+          << "Expected an integer value for --timeout=";
+      browser_->BrowserMainThread()->PostDelayedTask(
+          FROM_HERE,
+          base::Bind(&HeadlessShell::FetchTimeout, base::Unretained(this)),

[Sami, 2016/11/23 18:11:32]

Should we use a weak pointer here? Nothing seems to prevent this from firing
after the shell is gone.

[alex clarke (OOO till 29th), 2016/11/24 12:05:04]

Done.

+          base::TimeDelta::FromMilliseconds(timeout_ms));
+    }
+
     // TODO(skyostil): Implement more features to demonstrate the devtools API.
   }
 
+  void FetchTimeout() {
+    LOG(INFO) << "Timeout.";
+    web_contents_->Stop();
+  }
+
   void OnTargetCrashed(const inspector::TargetCrashedParams& params) override {
     LOG(ERROR) << "Abnormal renderer termination.";
     // NB this never gets called if remote debugging is enabled.
     Shutdown();
   }
 
   void PollReadyState() {
     // We need to check the current location in addition to the ready state to
     // be sure the expected page is ready.
     devtools_client_->GetRuntime()->Evaluate(
@@ skipping 263 matching lines @@
     }
     builder.SetWindowSize(parsed_window_size);
   }
 
   return HeadlessBrowserMain(
       builder.Build(),
       base::Bind(&HeadlessShell::OnStart, base::Unretained(&shell)));
 }
 
 }  // namespace headless