| 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
|
|
|