Index: headless/public/util/deterministic_http_protocol_handler.h |
diff --git a/headless/public/util/deterministic_http_protocol_handler.h b/headless/public/util/deterministic_http_protocol_handler.h |
index 05d5215402a8ffbeef445c2307f3d5f8354cfee8..7fc6057632742eacaacf351d626863006f68a0a2 100644 |
--- a/headless/public/util/deterministic_http_protocol_handler.h |
+++ b/headless/public/util/deterministic_http_protocol_handler.h |
@@ -18,6 +18,7 @@ class URLRequestJobFactory; |
namespace headless { |
class DeterministicDispatcher; |
+class HeadlessBrowserContext; |
// A deterministic protocol handler. Requests made to this protocol handler |
// will return in order of creation, regardless of what order the network |
@@ -37,10 +38,16 @@ class HEADLESS_EXPORT DeterministicHttpProtocolHandler |
net::URLRequest* request, |
net::NetworkDelegate* network_delegate) const override; |
+ void SetHeadlessBrowserContext( |
+ HeadlessBrowserContext* headless_browser_context) { |
+ headless_browser_context_ = headless_browser_context; |
+ } |
+ |
private: |
class NopGenericURLRequestJobDelegate; |
DeterministicDispatcher* deterministic_dispatcher_; // NOT OWNED. |
+ HeadlessBrowserContext* headless_browser_context_; // NOT OWNED. |
scoped_refptr<base::SingleThreadTaskRunner> io_task_runner_; |
std::unique_ptr<NopGenericURLRequestJobDelegate> nop_delegate_; |