Index: content/browser/service_worker/embedded_worker_test_helper.h |
diff --git a/content/browser/service_worker/embedded_worker_test_helper.h b/content/browser/service_worker/embedded_worker_test_helper.h |
index 5ac893d3318d32d7c99662ae9a448d773333c69c..6030f0431c3c248e0d55985e2670c8dd0a2c875f 100644 |
--- a/content/browser/service_worker/embedded_worker_test_helper.h |
+++ b/content/browser/service_worker/embedded_worker_test_helper.h |
@@ -73,15 +73,17 @@ class EmbeddedWorkerTestHelper : public IPC::Sender, |
static void Bind(const base::WeakPtr<EmbeddedWorkerTestHelper>& helper, |
mojom::EmbeddedWorkerInstanceClientRequest request); |
- private: |
+ protected: |
// Implementation of mojo interfaces |
void StartWorker(const EmbeddedWorkerStartParams& params) override; |
+ void StopWorker(const StopWorkerCallback& callback) override; |
base::WeakPtr<EmbeddedWorkerTestHelper> helper_; |
mojo::Binding<mojom::EmbeddedWorkerInstanceClient> binding_; |
base::Optional<int> embedded_worker_id_; |
+ private: |
DISALLOW_COPY_AND_ASSIGN(MockEmbeddedWorkerInstanceClient); |
}; |
@@ -100,6 +102,11 @@ class EmbeddedWorkerTestHelper : public IPC::Sender, |
// IPC::Listener implementation. |
bool OnMessageReceived(const IPC::Message& msg) override; |
+ // Register a mojo endpoint object derived from |
+ // MockEmbeddedWorkerInstanceClient. |
+ void RegisterMockInstanceClient( |
+ std::unique_ptr<MockEmbeddedWorkerInstanceClient> client); |
+ |
// IPC sink for EmbeddedWorker messages. |
IPC::TestSink* ipc_sink() { return &sink_; } |
// Inner IPC sink for script context messages sent via EmbeddedWorker. |
@@ -130,6 +137,10 @@ class EmbeddedWorkerTestHelper : public IPC::Sender, |
TestBrowserContext* browser_context() { return browser_context_.get(); } |
+ base::WeakPtr<EmbeddedWorkerTestHelper> AsWeakPtr() { |
+ return weak_factory_.GetWeakPtr(); |
+ } |
+ |
protected: |
// Called when StartWorker, StopWorker and SendMessageToWorker message |
// is sent to the embedded worker. Override if necessary. By default |