Index: content/browser/service_worker/embedded_worker_test_helper.cc |
diff --git a/content/browser/service_worker/embedded_worker_test_helper.cc b/content/browser/service_worker/embedded_worker_test_helper.cc |
index eeaf4c4bfe96bdc07c72a93e854b44ca1e02d0ef..5856822cca3e374466b7358e608690538871c9ca 100644 |
--- a/content/browser/service_worker/embedded_worker_test_helper.cc |
+++ b/content/browser/service_worker/embedded_worker_test_helper.cc |
@@ -150,6 +150,7 @@ bool EmbeddedWorkerTestHelper::OnMessageToWorker( |
IPC_MESSAGE_HANDLER(ServiceWorkerMsg_ActivateEvent, OnActivateEventStub) |
IPC_MESSAGE_HANDLER(ServiceWorkerMsg_InstallEvent, OnInstallEventStub) |
IPC_MESSAGE_HANDLER(ServiceWorkerMsg_FetchEvent, OnFetchEventStub) |
+ IPC_MESSAGE_HANDLER(ServiceWorkerMsg_PushEvent, OnPushEventStub) |
IPC_MESSAGE_UNHANDLED(handled = false) |
IPC_END_MESSAGE_MAP() |
// Record all messages directed to inner script context. |
@@ -194,6 +195,14 @@ void EmbeddedWorkerTestHelper::OnFetchEvent( |
GURL()))); |
} |
+void EmbeddedWorkerTestHelper::OnPushEvent(int embedded_worker_id, |
+ int request_id, |
+ const std::string& data) { |
+ SimulateSend(new ServiceWorkerHostMsg_PushEventFinished( |
+ embedded_worker_id, request_id, |
+ blink::WebServiceWorkerEventResultCompleted)); |
+} |
+ |
void EmbeddedWorkerTestHelper::SimulatePausedAfterDownload( |
int embedded_worker_id) { |
EmbeddedWorkerInstance* worker = registry()->GetWorker(embedded_worker_id); |
@@ -348,6 +357,14 @@ void EmbeddedWorkerTestHelper::OnFetchEventStub( |
request)); |
} |
+void EmbeddedWorkerTestHelper::OnPushEventStub(int request_id, |
+ const std::string& data) { |
+ base::ThreadTaskRunnerHandle::Get()->PostTask( |
+ FROM_HERE, base::Bind(&EmbeddedWorkerTestHelper::OnPushEvent, |
+ weak_factory_.GetWeakPtr(), |
+ current_embedded_worker_id_, request_id, data)); |
+} |
+ |
EmbeddedWorkerRegistry* EmbeddedWorkerTestHelper::registry() { |
DCHECK(context()); |
return context()->embedded_worker_registry(); |