Chromium Code Reviews| Index: content/browser/service_worker/service_worker_context_unittest.cc |
| diff --git a/content/browser/service_worker/service_worker_context_unittest.cc b/content/browser/service_worker/service_worker_context_unittest.cc |
| index 0dbdf1cb09b5810ca1de36909c18f0442e2dd6b3..2d6279c9282f11554f4a56863d2a51fd073f6677 100644 |
| --- a/content/browser/service_worker/service_worker_context_unittest.cc |
| +++ b/content/browser/service_worker/service_worker_context_unittest.cc |
| @@ -85,11 +85,12 @@ class RejectInstallTestHelper : public EmbeddedWorkerTestHelper { |
| public: |
| RejectInstallTestHelper() : EmbeddedWorkerTestHelper(base::FilePath()) {} |
| - void OnInstallEvent(int embedded_worker_id, |
| - int request_id) override { |
| - SimulateSend(new ServiceWorkerHostMsg_InstallEventFinished( |
| - embedded_worker_id, request_id, |
| - blink::kWebServiceWorkerEventResultRejected, true, base::Time::Now())); |
| + void OnInstallEvent( |
| + const mojom::ServiceWorkerEventDispatcher::DispatchInstallEventCallback& |
| + callback) override { |
| + dispatched_events()->push_back(Event::Install); |
| + callback.Run(SERVICE_WORKER_ERROR_EVENT_WAITUNTIL_REJECTED, |
| + true /* has_fetch_handler */, base::Time::Now()); |
| } |
| }; |
| @@ -217,15 +218,14 @@ TEST_F(ServiceWorkerContextTest, Register) { |
| base::RunLoop().RunUntilIdle(); |
| EXPECT_TRUE(called); |
| - EXPECT_EQ(1UL, helper_->ipc_sink()->message_count()); |
| - ASSERT_EQ(1UL, helper_->dispatched_events()->size()); |
| + ASSERT_EQ(2UL, helper_->dispatched_events()->size()); |
| ASSERT_EQ(2UL, client->events().size()); |
| EXPECT_EQ(RecordableEmbeddedWorkerInstanceClient::Message::StartWorker, |
| client->events()[0]); |
| - EXPECT_TRUE(helper_->inner_ipc_sink()->GetUniqueMessageMatching( |
| - ServiceWorkerMsg_InstallEvent::ID)); |
| - EXPECT_EQ(EmbeddedWorkerTestHelper::Event::Activate, |
| + EXPECT_EQ(EmbeddedWorkerTestHelper::Event::Install, |
| helper_->dispatched_events()->at(0)); |
| + EXPECT_EQ(EmbeddedWorkerTestHelper::Event::Activate, |
| + helper_->dispatched_events()->at(1)); |
| EXPECT_EQ(RecordableEmbeddedWorkerInstanceClient::Message::StopWorker, |
| client->events()[1]); |
| @@ -273,13 +273,12 @@ TEST_F(ServiceWorkerContextTest, Register_RejectInstall) { |
| base::RunLoop().RunUntilIdle(); |
| EXPECT_TRUE(called); |
| - EXPECT_EQ(1UL, helper_->ipc_sink()->message_count()); |
| - EXPECT_EQ(0UL, helper_->dispatched_events()->size()); |
| + EXPECT_EQ(1UL, helper_->dispatched_events()->size()); |
|
shimazu
2017/04/24 05:00:13
ASSERT_EQ to check if enough elements exist before
xiaofengzhang
2017/04/25 06:01:28
Done.
|
| ASSERT_EQ(2UL, client->events().size()); |
| EXPECT_EQ(RecordableEmbeddedWorkerInstanceClient::Message::StartWorker, |
| client->events()[0]); |
| - EXPECT_TRUE(helper_->inner_ipc_sink()->GetUniqueMessageMatching( |
| - ServiceWorkerMsg_InstallEvent::ID)); |
| + EXPECT_EQ(EmbeddedWorkerTestHelper::Event::Install, |
| + helper_->dispatched_events()->at(0)); |
| EXPECT_EQ(RecordableEmbeddedWorkerInstanceClient::Message::StopWorker, |
| client->events()[1]); |
| @@ -324,15 +323,14 @@ TEST_F(ServiceWorkerContextTest, Register_RejectActivate) { |
| base::RunLoop().RunUntilIdle(); |
| EXPECT_TRUE(called); |
| - EXPECT_EQ(1UL, helper_->ipc_sink()->message_count()); |
| - ASSERT_EQ(1UL, helper_->dispatched_events()->size()); |
| + ASSERT_EQ(2UL, helper_->dispatched_events()->size()); |
| ASSERT_EQ(2UL, client->events().size()); |
| EXPECT_EQ(RecordableEmbeddedWorkerInstanceClient::Message::StartWorker, |
| client->events()[0]); |
| - EXPECT_TRUE(helper_->inner_ipc_sink()->GetUniqueMessageMatching( |
| - ServiceWorkerMsg_InstallEvent::ID)); |
| - EXPECT_EQ(EmbeddedWorkerTestHelper::Event::Activate, |
| + EXPECT_EQ(EmbeddedWorkerTestHelper::Event::Install, |
| helper_->dispatched_events()->at(0)); |
| + EXPECT_EQ(EmbeddedWorkerTestHelper::Event::Activate, |
| + helper_->dispatched_events()->at(1)); |
| EXPECT_EQ(RecordableEmbeddedWorkerInstanceClient::Message::StopWorker, |
| client->events()[1]); |