Set network conditions in all available browser contexts instead of only the default.

Bug: 
Change-Id: I96727d1c6b4cd09303eba6ca88630d83405623e2
Reviewed-on: https://chromium-review.googlesource.com/722362
Commit-Queue: Iris Uy <irisu@chromium.org>
Reviewed-by: Jianzhou Feng <jzfeng@chromium.org>
Reviewed-by: David Vallet <dvallet@chromium.org>
Cr-Commit-Position: refs/heads/master@{#509651}

headless/lib/browser/headless_devtools_manager_delegate.cc

@@ -21,7 +21,6 @@
 #include "headless/grit/headless_lib_resources.h"
 #include "headless/lib/browser/headless_browser_context_impl.h"
 #include "headless/lib/browser/headless_browser_impl.h"
-#include "headless/lib/browser/headless_network_conditions.h"
 #include "headless/lib/browser/headless_web_contents_impl.h"
 #include "headless/public/devtools/domains/target.h"
 #include "printing/units.h"
@@ -686,9 +685,10 @@ HeadlessDevToolsManagerDelegate::EmulateNetworkConditions(
     int command_id,
     const base::DictionaryValue* params) {
   // Associate NetworkConditions to context
-  HeadlessBrowserContextImpl* browser_context =
-      static_cast<HeadlessBrowserContextImpl*>(
-          browser_->GetDefaultBrowserContext());
+  std::vector<HeadlessBrowserContext*> browser_contexts =
+      browser_->GetAllBrowserContexts();
+  if (browser_contexts.empty())
+    return CreateSuccessResponse(command_id, nullptr);
   bool offline = false;
   double latency = 0, download_throughput = 0, upload_throughput = 0;
   params->GetBoolean("offline", &offline);
@@ -698,7 +698,7 @@ HeadlessDevToolsManagerDelegate::EmulateNetworkConditions(
   HeadlessNetworkConditions conditions(HeadlessNetworkConditions(
       offline, std::max(latency, 0.0), std::max(download_throughput, 0.0),
       std::max(upload_throughput, 0.0)));
-  browser_context->SetNetworkConditions(conditions);
+  SetNetworkConditions(browser_contexts, conditions);
   return CreateSuccessResponse(command_id, nullptr);
 }
 
@@ -708,13 +708,26 @@ HeadlessDevToolsManagerDelegate::NetworkDisable(
     int session_id,
     int command_id,
     const base::DictionaryValue* params) {
-  HeadlessBrowserContextImpl* browser_context =
-      static_cast<HeadlessBrowserContextImpl*>(
-          browser_->GetDefaultBrowserContext());
-  browser_context->SetNetworkConditions(HeadlessNetworkConditions());
+  std::vector<HeadlessBrowserContext*> browser_contexts =
+      browser_->GetAllBrowserContexts();
+  if (browser_contexts.empty())
+    return CreateSuccessResponse(command_id, nullptr);
+  SetNetworkConditions(browser_contexts, HeadlessNetworkConditions());
   return CreateSuccessResponse(command_id, nullptr);
 }
 
+void HeadlessDevToolsManagerDelegate::SetNetworkConditions(
+    std::vector<HeadlessBrowserContext*> browser_contexts,
+    HeadlessNetworkConditions conditions) {
+  for (std::vector<HeadlessBrowserContext*>::iterator it =
+           browser_contexts.begin();
+       it != browser_contexts.end(); ++it) {
+    HeadlessBrowserContextImpl* context =
+        static_cast<HeadlessBrowserContextImpl*>(*it);
+    context->SetNetworkConditions(HeadlessNetworkConditions());
+  }
+}
+
 std::unique_ptr<base::DictionaryValue>
 HeadlessDevToolsManagerDelegate::EnableHeadlessExperimental(
     content::DevToolsAgentHost* agent_host,

headless/lib/browser/headless_devtools_manager_delegate.h

@@ -13,6 +13,8 @@
 #include "base/memory/weak_ptr.h"
 #include "base/values.h"
 #include "content/public/browser/devtools_manager_delegate.h"
+#include "headless/lib/browser/headless_network_conditions.h"
+#include "headless/public/headless_browser_context.h"
 #include "printing/features/features.h"
 
 #if BUILDFLAG(ENABLE_BASIC_PRINTING)
@@ -116,6 +118,10 @@ class HeadlessDevToolsManagerDelegate
       int command_id,
       const base::DictionaryValue* params);
 
+  void SetNetworkConditions(
+      std::vector<HeadlessBrowserContext*> browser_contexts,
+      HeadlessNetworkConditions conditions);
+
   void PrintToPDF(content::DevToolsAgentHost* agent_host,
                   int session_id,
                   int command_id,