| 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 eb2ea20844f557782deee41597054ce5c1c74735..fade1fd8d81d4106cd3f2cdeded8095098a4dc2c 100644
|
| --- a/content/browser/service_worker/embedded_worker_test_helper.h
|
| +++ b/content/browser/service_worker/embedded_worker_test_helper.h
|
| @@ -79,15 +79,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);
|
| };
|
|
|
| @@ -106,6 +108,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.
|
| @@ -136,6 +143,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
|
|
|