Index: content/browser/service_worker/service_worker_url_request_job_unittest.cc |
diff --git a/content/browser/service_worker/service_worker_url_request_job_unittest.cc b/content/browser/service_worker/service_worker_url_request_job_unittest.cc |
index 690154c23ea88576ed89108ba36da479309351b7..8672247c5a6052a6ee72e09bacb913d36b52917e 100644 |
--- a/content/browser/service_worker/service_worker_url_request_job_unittest.cc |
+++ b/content/browser/service_worker/service_worker_url_request_job_unittest.cc |
@@ -34,6 +34,7 @@ |
#include "content/browser/streams/stream_registry.h" |
#include "content/common/resource_request_body_impl.h" |
#include "content/common/service_worker/service_worker_messages.h" |
+#include "content/common/service_worker/service_worker_status_code.h" |
#include "content/public/browser/blob_handle.h" |
#include "content/public/common/request_context_frame_type.h" |
#include "content/public/common/request_context_type.h" |
@@ -364,8 +365,8 @@ class ProviderDeleteHelper : public EmbeddedWorkerTestHelper { |
protected: |
void OnFetchEvent(int embedded_worker_id, |
int response_id, |
- int event_finish_id, |
- const ServiceWorkerFetchRequest& request) override { |
+ const ServiceWorkerFetchRequest& request, |
+ const FetchCallback& callback) override { |
context()->RemoveProviderHost(mock_render_process_id(), kProviderID); |
SimulateSend(new ServiceWorkerHostMsg_FetchEventResponse( |
embedded_worker_id, response_id, |
@@ -378,9 +379,7 @@ class ProviderDeleteHelper : public EmbeddedWorkerTestHelper { |
std::string() /* response_cache_storage_cache_name */, |
ServiceWorkerHeaderList() /* cors_exposed_header_names */), |
base::Time::Now())); |
- SimulateSend(new ServiceWorkerHostMsg_FetchEventFinished( |
- embedded_worker_id, event_finish_id, |
- blink::WebServiceWorkerEventResultCompleted, base::Time::Now())); |
+ callback.Run(SERVICE_WORKER_OK, base::Time::Now()); |
} |
private: |
@@ -449,8 +448,8 @@ class BlobResponder : public EmbeddedWorkerTestHelper { |
protected: |
void OnFetchEvent(int embedded_worker_id, |
int response_id, |
- int event_finish_id, |
- const ServiceWorkerFetchRequest& request) override { |
+ const ServiceWorkerFetchRequest& request, |
+ const FetchCallback& callback) override { |
SimulateSend(new ServiceWorkerHostMsg_FetchEventResponse( |
embedded_worker_id, response_id, |
SERVICE_WORKER_FETCH_EVENT_RESULT_RESPONSE, |
@@ -462,9 +461,7 @@ class BlobResponder : public EmbeddedWorkerTestHelper { |
std::string() /* response_cache_storage_cache_name */, |
ServiceWorkerHeaderList() /* cors_exposed_header_names */), |
base::Time::Now())); |
- SimulateSend(new ServiceWorkerHostMsg_FetchEventFinished( |
- embedded_worker_id, event_finish_id, |
- blink::WebServiceWorkerEventResultCompleted, base::Time::Now())); |
+ callback.Run(SERVICE_WORKER_OK, base::Time::Now()); |
} |
std::string blob_uuid_; |
@@ -533,8 +530,8 @@ class StreamResponder : public EmbeddedWorkerTestHelper { |
protected: |
void OnFetchEvent(int embedded_worker_id, |
int response_id, |
- int event_finish_id, |
- const ServiceWorkerFetchRequest& request) override { |
+ const ServiceWorkerFetchRequest& request, |
+ const FetchCallback& callback) override { |
SimulateSend(new ServiceWorkerHostMsg_FetchEventResponse( |
embedded_worker_id, response_id, |
SERVICE_WORKER_FETCH_EVENT_RESULT_RESPONSE, |
@@ -546,9 +543,7 @@ class StreamResponder : public EmbeddedWorkerTestHelper { |
std::string() /* response_cache_storage_cache_name */, |
ServiceWorkerHeaderList() /* cors_exposed_header_names */), |
base::Time::Now())); |
- SimulateSend(new ServiceWorkerHostMsg_FetchEventFinished( |
- embedded_worker_id, event_finish_id, |
- blink::WebServiceWorkerEventResultCompleted, base::Time::Now())); |
+ callback.Run(SERVICE_WORKER_OK, base::Time::Now()); |
} |
const GURL stream_url_; |
@@ -848,9 +843,10 @@ class FailFetchHelper : public EmbeddedWorkerTestHelper { |
protected: |
void OnFetchEvent(int embedded_worker_id, |
int response_id, |
- int event_finish_id, |
- const ServiceWorkerFetchRequest& request) override { |
+ const ServiceWorkerFetchRequest& request, |
+ const FetchCallback& callback) override { |
SimulateWorkerStopped(embedded_worker_id); |
+ callback.Run(SERVICE_WORKER_ERROR_ABORT, base::Time::Now()); |
} |
private: |
@@ -925,18 +921,15 @@ class EarlyResponseHelper : public EmbeddedWorkerTestHelper { |
~EarlyResponseHelper() override {} |
void FinishWaitUntil() { |
- SimulateSend(new ServiceWorkerHostMsg_FetchEventFinished( |
- embedded_worker_id_, event_finish_id_, |
- blink::WebServiceWorkerEventResultCompleted, base::Time::Now())); |
+ callback_.Run(SERVICE_WORKER_OK, base::Time::Now()); |
} |
protected: |
void OnFetchEvent(int embedded_worker_id, |
int response_id, |
- int event_finish_id, |
- const ServiceWorkerFetchRequest& request) override { |
- embedded_worker_id_ = embedded_worker_id; |
- event_finish_id_ = event_finish_id; |
+ const ServiceWorkerFetchRequest& request, |
+ const FetchCallback& callback) override { |
+ callback_ = callback; |
SimulateSend(new ServiceWorkerHostMsg_FetchEventResponse( |
embedded_worker_id, response_id, |
SERVICE_WORKER_FETCH_EVENT_RESULT_RESPONSE, |
@@ -951,8 +944,7 @@ class EarlyResponseHelper : public EmbeddedWorkerTestHelper { |
} |
private: |
- int embedded_worker_id_ = 0; |
- int event_finish_id_ = 0; |
+ FetchCallback callback_; |
DISALLOW_COPY_AND_ASSIGN(EarlyResponseHelper); |
}; |
@@ -981,6 +973,7 @@ TEST_F(ServiceWorkerURLRequestJobTest, EarlyResponse) { |
EXPECT_TRUE(version_->HasWork()); |
helper->FinishWaitUntil(); |
+ base::RunLoop().RunUntilIdle(); |
EXPECT_FALSE(version_->HasWork()); |
} |
@@ -1002,25 +995,23 @@ class DelayedResponseHelper : public EmbeddedWorkerTestHelper { |
std::string() /* response_cache_storage_cache_name */, |
ServiceWorkerHeaderList() /* cors_exposed_header_names */), |
base::Time::Now())); |
- SimulateSend(new ServiceWorkerHostMsg_FetchEventFinished( |
- embedded_worker_id_, event_finish_id_, |
- blink::WebServiceWorkerEventResultCompleted, base::Time::Now())); |
+ callback_.Run(SERVICE_WORKER_OK, base::Time::Now()); |
} |
protected: |
void OnFetchEvent(int embedded_worker_id, |
int response_id, |
- int event_finish_id, |
- const ServiceWorkerFetchRequest& request) override { |
+ const ServiceWorkerFetchRequest& request, |
+ const FetchCallback& callback) override { |
embedded_worker_id_ = embedded_worker_id; |
response_id_ = response_id; |
- event_finish_id_ = event_finish_id; |
+ callback_ = callback; |
} |
private: |
int embedded_worker_id_ = 0; |
int response_id_ = 0; |
- int event_finish_id_ = 0; |
+ FetchCallback callback_; |
DISALLOW_COPY_AND_ASSIGN(DelayedResponseHelper); |
}; |