| // Copyright 2014 The Chromium Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| #include <stdint.h> |
| #include <tuple> |
| |
| #include "base/bind.h" |
| #include "base/bind_helpers.h" |
| #include "base/files/scoped_temp_dir.h" |
| #include "base/logging.h" |
| #include "base/optional.h" |
| #include "base/run_loop.h" |
| #include "base/stl_util.h" |
| #include "base/test/test_simple_task_runner.h" |
| #include "base/time/time.h" |
| #include "content/browser/service_worker/embedded_worker_status.h" |
| #include "content/browser/service_worker/embedded_worker_test_helper.h" |
| #include "content/browser/service_worker/fake_embedded_worker_instance_client.h" |
| #include "content/browser/service_worker/service_worker_context_core.h" |
| #include "content/browser/service_worker/service_worker_context_wrapper.h" |
| #include "content/browser/service_worker/service_worker_disk_cache.h" |
| #include "content/browser/service_worker/service_worker_job_coordinator.h" |
| #include "content/browser/service_worker/service_worker_object_host.h" |
| #include "content/browser/service_worker/service_worker_registration.h" |
| #include "content/browser/service_worker/service_worker_registration_object_host.h" |
| #include "content/browser/service_worker/service_worker_registration_status.h" |
| #include "content/browser/service_worker/service_worker_test_utils.h" |
| #include "content/browser/service_worker/test_service_worker_observer.h" |
| #include "content/common/service_worker/service_worker_utils.h" |
| #include "content/public/test/test_browser_context.h" |
| #include "content/public/test/test_browser_thread_bundle.h" |
| #include "ipc/ipc_test_sink.h" |
| #include "net/base/io_buffer.h" |
| #include "net/base/net_errors.h" |
| #include "net/base/test_completion_callback.h" |
| #include "net/http/http_response_headers.h" |
| #include "testing/gtest/include/gtest/gtest.h" |
| #include "third_party/blink/public/mojom/service_worker/service_worker.mojom.h" |
| #include "third_party/blink/public/mojom/service_worker/service_worker_event_status.mojom.h" |
| #include "third_party/blink/public/mojom/service_worker/service_worker_object.mojom.h" |
| #include "third_party/blink/public/mojom/service_worker/service_worker_registration.mojom.h" |
| |
| using net::IOBuffer; |
| using net::TestCompletionCallback; |
| using net::WrappedIOBuffer; |
| |
| // Unit tests for testing all job registration tasks. |
| namespace content { |
| |
| namespace { |
| |
| void SaveRegistrationCallback( |
| blink::ServiceWorkerStatusCode expected_status, |
| bool* called, |
| scoped_refptr<ServiceWorkerRegistration>* registration_out, |
| blink::ServiceWorkerStatusCode status, |
| const std::string& status_message, |
| ServiceWorkerRegistration* registration) { |
| EXPECT_EQ(expected_status, status); |
| *called = true; |
| *registration_out = registration; |
| } |
| |
| void SaveFoundRegistrationCallback( |
| blink::ServiceWorkerStatusCode expected_status, |
| bool* called, |
| scoped_refptr<ServiceWorkerRegistration>* registration, |
| blink::ServiceWorkerStatusCode status, |
| scoped_refptr<ServiceWorkerRegistration> result) { |
| EXPECT_EQ(expected_status, status); |
| *called = true; |
| *registration = std::move(result); |
| } |
| |
| // Creates a callback which both keeps track of if it's been called, |
| // as well as the resulting registration. Whent the callback is fired, |
| // it ensures that the resulting status matches the expectation. |
| // 'called' is useful for making sure a sychronous callback is or |
| // isn't called. |
| ServiceWorkerRegisterJob::RegistrationCallback SaveRegistration( |
| blink::ServiceWorkerStatusCode expected_status, |
| bool* called, |
| scoped_refptr<ServiceWorkerRegistration>* registration) { |
| *called = false; |
| return base::BindOnce(&SaveRegistrationCallback, expected_status, called, |
| registration); |
| } |
| |
| ServiceWorkerStorage::FindRegistrationCallback SaveFoundRegistration( |
| blink::ServiceWorkerStatusCode expected_status, |
| bool* called, |
| scoped_refptr<ServiceWorkerRegistration>* registration) { |
| *called = false; |
| return base::BindOnce(&SaveFoundRegistrationCallback, expected_status, called, |
| registration); |
| } |
| |
| void SaveUnregistrationCallback(blink::ServiceWorkerStatusCode expected_status, |
| bool* called, |
| int64_t registration_id, |
| blink::ServiceWorkerStatusCode status) { |
| EXPECT_EQ(expected_status, status); |
| *called = true; |
| } |
| |
| ServiceWorkerUnregisterJob::UnregistrationCallback SaveUnregistration( |
| blink::ServiceWorkerStatusCode expected_status, |
| bool* called) { |
| *called = false; |
| return base::BindOnce(&SaveUnregistrationCallback, expected_status, called); |
| } |
| |
| } // namespace |
| |
| class ServiceWorkerJobTest : public testing::Test { |
| public: |
| ServiceWorkerJobTest() |
| : browser_thread_bundle_(TestBrowserThreadBundle::IO_MAINLOOP) {} |
| |
| void SetUp() override { |
| helper_.reset(new EmbeddedWorkerTestHelper(base::FilePath())); |
| } |
| |
| void TearDown() override { helper_.reset(); } |
| |
| ServiceWorkerContextCore* context() const { return helper_->context(); } |
| |
| ServiceWorkerJobCoordinator* job_coordinator() const { |
| return context()->job_coordinator(); |
| } |
| ServiceWorkerStorage* storage() const { return context()->storage(); } |
| |
| protected: |
| scoped_refptr<ServiceWorkerRegistration> RunRegisterJob( |
| const GURL& script_url, |
| const blink::mojom::ServiceWorkerRegistrationOptions& options, |
| blink::ServiceWorkerStatusCode expected_status = |
| blink::ServiceWorkerStatusCode::kOk); |
| void RunUnregisterJob(const GURL& scope, |
| blink::ServiceWorkerStatusCode expected_status = |
| blink::ServiceWorkerStatusCode::kOk); |
| scoped_refptr<ServiceWorkerRegistration> FindRegistrationForScope( |
| const GURL& scope, |
| blink::ServiceWorkerStatusCode expected_status = |
| blink::ServiceWorkerStatusCode::kOk); |
| ServiceWorkerProviderHost* CreateControllee(); |
| |
| TestBrowserThreadBundle browser_thread_bundle_; |
| std::unique_ptr<EmbeddedWorkerTestHelper> helper_; |
| std::vector<ServiceWorkerRemoteProviderEndpoint> remote_endpoints_; |
| }; |
| |
| scoped_refptr<ServiceWorkerRegistration> ServiceWorkerJobTest::RunRegisterJob( |
| const GURL& script_url, |
| const blink::mojom::ServiceWorkerRegistrationOptions& options, |
| blink::ServiceWorkerStatusCode expected_status) { |
| scoped_refptr<ServiceWorkerRegistration> registration; |
| bool called; |
| job_coordinator()->Register( |
| script_url, options, |
| SaveRegistration(expected_status, &called, ®istration)); |
| EXPECT_FALSE(called); |
| base::RunLoop().RunUntilIdle(); |
| EXPECT_TRUE(called); |
| return registration; |
| } |
| |
| void ServiceWorkerJobTest::RunUnregisterJob( |
| const GURL& scope, |
| blink::ServiceWorkerStatusCode expected_status) { |
| bool called; |
| job_coordinator()->Unregister(scope, |
| SaveUnregistration(expected_status, &called)); |
| EXPECT_FALSE(called); |
| base::RunLoop().RunUntilIdle(); |
| EXPECT_TRUE(called); |
| } |
| |
| scoped_refptr<ServiceWorkerRegistration> |
| ServiceWorkerJobTest::FindRegistrationForScope( |
| const GURL& scope, |
| blink::ServiceWorkerStatusCode expected_status) { |
| bool called; |
| scoped_refptr<ServiceWorkerRegistration> registration; |
| storage()->FindRegistrationForScope( |
| scope, SaveFoundRegistration(expected_status, &called, ®istration)); |
| |
| EXPECT_FALSE(called); |
| base::RunLoop().RunUntilIdle(); |
| EXPECT_TRUE(called); |
| return registration; |
| } |
| |
| ServiceWorkerProviderHost* ServiceWorkerJobTest::CreateControllee() { |
| remote_endpoints_.emplace_back(); |
| base::WeakPtr<ServiceWorkerProviderHost> host = CreateProviderHostForWindow( |
| 33 /* dummy render process id */, true /* is_parent_frame_secure */, |
| helper_->context()->AsWeakPtr(), &remote_endpoints_.back()); |
| auto* host_ptr = host.get(); |
| return host_ptr; |
| } |
| |
| TEST_F(ServiceWorkerJobTest, SameDocumentSameRegistration) { |
| blink::mojom::ServiceWorkerRegistrationOptions options; |
| options.scope = GURL("https://www.example.com/"); |
| scoped_refptr<ServiceWorkerRegistration> original_registration = |
| RunRegisterJob(GURL("https://www.example.com/service_worker.js"), |
| options); |
| bool called; |
| scoped_refptr<ServiceWorkerRegistration> registration1; |
| storage()->FindRegistrationForDocument( |
| GURL("https://www.example.com/"), |
| SaveFoundRegistration(blink::ServiceWorkerStatusCode::kOk, &called, |
| ®istration1)); |
| scoped_refptr<ServiceWorkerRegistration> registration2; |
| storage()->FindRegistrationForDocument( |
| GURL("https://www.example.com/"), |
| SaveFoundRegistration(blink::ServiceWorkerStatusCode::kOk, &called, |
| ®istration2)); |
| base::RunLoop().RunUntilIdle(); |
| EXPECT_TRUE(called); |
| ASSERT_TRUE(registration1.get()); |
| ASSERT_EQ(registration1, original_registration); |
| ASSERT_EQ(registration1, registration2); |
| } |
| |
| TEST_F(ServiceWorkerJobTest, SameMatchSameRegistration) { |
| bool called; |
| blink::mojom::ServiceWorkerRegistrationOptions options; |
| options.scope = GURL("https://www.example.com/"); |
| scoped_refptr<ServiceWorkerRegistration> original_registration = |
| RunRegisterJob(GURL("https://www.example.com/service_worker.js"), |
| options); |
| ASSERT_NE(static_cast<ServiceWorkerRegistration*>(nullptr), |
| original_registration.get()); |
| |
| scoped_refptr<ServiceWorkerRegistration> registration1; |
| storage()->FindRegistrationForDocument( |
| GURL("https://www.example.com/one"), |
| SaveFoundRegistration(blink::ServiceWorkerStatusCode::kOk, &called, |
| ®istration1)); |
| base::RunLoop().RunUntilIdle(); |
| EXPECT_TRUE(called); |
| |
| scoped_refptr<ServiceWorkerRegistration> registration2; |
| storage()->FindRegistrationForDocument( |
| GURL("https://www.example.com/two"), |
| SaveFoundRegistration(blink::ServiceWorkerStatusCode::kOk, &called, |
| ®istration2)); |
| base::RunLoop().RunUntilIdle(); |
| EXPECT_TRUE(called); |
| ASSERT_EQ(registration1, original_registration); |
| ASSERT_EQ(registration1, registration2); |
| } |
| |
| TEST_F(ServiceWorkerJobTest, DifferentMatchDifferentRegistration) { |
| bool called1; |
| blink::mojom::ServiceWorkerRegistrationOptions options1; |
| options1.scope = GURL("https://www.example.com/one/"); |
| scoped_refptr<ServiceWorkerRegistration> original_registration1; |
| job_coordinator()->Register( |
| GURL("https://www.example.com/service_worker.js"), options1, |
| SaveRegistration(blink::ServiceWorkerStatusCode::kOk, &called1, |
| &original_registration1)); |
| |
| bool called2; |
| blink::mojom::ServiceWorkerRegistrationOptions options2; |
| options2.scope = GURL("https://www.example.com/two/"); |
| scoped_refptr<ServiceWorkerRegistration> original_registration2; |
| job_coordinator()->Register( |
| GURL("https://www.example.com/service_worker.js"), options2, |
| SaveRegistration(blink::ServiceWorkerStatusCode::kOk, &called2, |
| &original_registration2)); |
| |
| EXPECT_FALSE(called1); |
| EXPECT_FALSE(called2); |
| base::RunLoop().RunUntilIdle(); |
| EXPECT_TRUE(called2); |
| EXPECT_TRUE(called1); |
| |
| scoped_refptr<ServiceWorkerRegistration> registration1; |
| storage()->FindRegistrationForDocument( |
| GURL("https://www.example.com/one/"), |
| SaveFoundRegistration(blink::ServiceWorkerStatusCode::kOk, &called1, |
| ®istration1)); |
| scoped_refptr<ServiceWorkerRegistration> registration2; |
| storage()->FindRegistrationForDocument( |
| GURL("https://www.example.com/two/"), |
| SaveFoundRegistration(blink::ServiceWorkerStatusCode::kOk, &called2, |
| ®istration2)); |
| |
| base::RunLoop().RunUntilIdle(); |
| EXPECT_TRUE(called2); |
| EXPECT_TRUE(called1); |
| ASSERT_NE(registration1, registration2); |
| } |
| |
| class RecordInstallActivateWorker : public FakeServiceWorker { |
| public: |
| RecordInstallActivateWorker(EmbeddedWorkerTestHelper* helper) |
| : FakeServiceWorker(helper) {} |
| ~RecordInstallActivateWorker() override = default; |
| |
| const std::vector<ServiceWorkerMetrics::EventType>& events() const { |
| return events_; |
| } |
| |
| void DispatchInstallEvent( |
| blink::mojom::ServiceWorker::DispatchInstallEventCallback callback) |
| override { |
| events_.emplace_back(ServiceWorkerMetrics::EventType::INSTALL); |
| FakeServiceWorker::DispatchInstallEvent(std::move(callback)); |
| } |
| |
| void DispatchActivateEvent( |
| blink::mojom::ServiceWorker::DispatchActivateEventCallback callback) |
| override { |
| events_.emplace_back(ServiceWorkerMetrics::EventType::ACTIVATE); |
| FakeServiceWorker::DispatchActivateEvent(std::move(callback)); |
| } |
| |
| private: |
| std::vector<ServiceWorkerMetrics::EventType> events_; |
| }; |
| |
| // Make sure basic registration is working. |
| TEST_F(ServiceWorkerJobTest, Register) { |
| blink::mojom::ServiceWorkerRegistrationOptions options; |
| options.scope = GURL("https://www.example.com/"); |
| auto* worker = |
| helper_->AddNewPendingServiceWorker<RecordInstallActivateWorker>( |
| helper_.get()); |
| scoped_refptr<ServiceWorkerRegistration> registration = RunRegisterJob( |
| GURL("https://www.example.com/service_worker.js"), options); |
| |
| EXPECT_TRUE(registration); |
| ASSERT_EQ(2u, worker->events().size()); |
| EXPECT_EQ(ServiceWorkerMetrics::EventType::INSTALL, worker->events()[0]); |
| EXPECT_EQ(ServiceWorkerMetrics::EventType::ACTIVATE, worker->events()[1]); |
| } |
| |
| // Make sure registrations are cleaned up when they are unregistered. |
| TEST_F(ServiceWorkerJobTest, Unregister) { |
| blink::mojom::ServiceWorkerRegistrationOptions options; |
| options.scope = GURL("https://www.example.com/"); |
| scoped_refptr<ServiceWorkerRegistration> registration = RunRegisterJob( |
| GURL("https://www.example.com/service_worker.js"), options); |
| scoped_refptr<ServiceWorkerVersion> version = registration->active_version(); |
| |
| ServiceWorkerProviderHost* provider_host = |
| registration->active_version()->provider_host(); |
| ASSERT_NE(nullptr, provider_host); |
| // One ServiceWorkerRegistrationObjectHost should have been created for the |
| // new registration. |
| EXPECT_EQ(1UL, provider_host->registration_object_hosts_.size()); |
| // One ServiceWorkerObjectHost should have been created for the new service |
| // worker. |
| EXPECT_EQ(1UL, provider_host->service_worker_object_hosts_.size()); |
| |
| RunUnregisterJob(options.scope); |
| |
| // The service worker registration object host and service worker object host |
| // have been destroyed together with |provider_host| by the above |
| // unregistration. Then |registration| and |version| should be the last one |
| // reference to the corresponding instance. |
| EXPECT_TRUE(registration->HasOneRef()); |
| EXPECT_TRUE(version->HasOneRef()); |
| EXPECT_EQ(EmbeddedWorkerStatus::STOPPED, version->running_status()); |
| EXPECT_EQ(ServiceWorkerVersion::REDUNDANT, version->status()); |
| |
| registration = FindRegistrationForScope( |
| options.scope, blink::ServiceWorkerStatusCode::kErrorNotFound); |
| |
| EXPECT_FALSE(registration); |
| } |
| |
| TEST_F(ServiceWorkerJobTest, Unregister_NothingRegistered) { |
| GURL scope("https://www.example.com/"); |
| |
| RunUnregisterJob(scope, blink::ServiceWorkerStatusCode::kErrorNotFound); |
| } |
| |
| // Make sure registering a new script creates a new version and shares an |
| // existing registration. |
| TEST_F(ServiceWorkerJobTest, RegisterNewScript) { |
| blink::mojom::ServiceWorkerRegistrationOptions options; |
| options.scope = GURL("https://www.example.com/"); |
| |
| scoped_refptr<ServiceWorkerRegistration> old_registration = RunRegisterJob( |
| GURL("https://www.example.com/service_worker.js"), options); |
| |
| scoped_refptr<ServiceWorkerRegistration> old_registration_by_scope = |
| FindRegistrationForScope(options.scope); |
| |
| ASSERT_EQ(old_registration, old_registration_by_scope); |
| old_registration_by_scope = nullptr; |
| |
| scoped_refptr<ServiceWorkerRegistration> new_registration = RunRegisterJob( |
| GURL("https://www.example.com/service_worker_new.js"), options); |
| |
| ASSERT_EQ(old_registration, new_registration); |
| |
| scoped_refptr<ServiceWorkerRegistration> new_registration_by_scope = |
| FindRegistrationForScope(options.scope); |
| |
| ASSERT_EQ(new_registration, new_registration_by_scope); |
| } |
| |
| // Make sure that when registering a duplicate scope+script_url |
| // combination, that the same registration is used. |
| TEST_F(ServiceWorkerJobTest, RegisterDuplicateScript) { |
| GURL script_url("https://www.example.com/service_worker.js"); |
| blink::mojom::ServiceWorkerRegistrationOptions options; |
| options.scope = GURL("https://www.example.com/"); |
| |
| scoped_refptr<ServiceWorkerRegistration> old_registration = |
| RunRegisterJob(script_url, options); |
| |
| // During the above registration, a service worker registration object host |
| // for ServiceWorkerGlobalScope#registration has been created/added into |
| // |provider_host|. |
| ServiceWorkerProviderHost* provider_host = |
| old_registration->active_version()->provider_host(); |
| ASSERT_NE(nullptr, provider_host); |
| |
| // Clear all service worker object hosts. |
| provider_host->service_worker_object_hosts_.clear(); |
| // Ensure that the registration's object host doesn't have the reference. |
| EXPECT_EQ(1UL, provider_host->registration_object_hosts_.size()); |
| provider_host->registration_object_hosts_.clear(); |
| EXPECT_EQ(0UL, provider_host->registration_object_hosts_.size()); |
| ASSERT_TRUE(old_registration->HasOneRef()); |
| |
| scoped_refptr<ServiceWorkerRegistration> old_registration_by_scope = |
| FindRegistrationForScope(options.scope); |
| |
| ASSERT_TRUE(old_registration_by_scope.get()); |
| |
| scoped_refptr<ServiceWorkerRegistration> new_registration = |
| RunRegisterJob(script_url, options); |
| |
| ASSERT_EQ(old_registration, new_registration); |
| |
| ASSERT_FALSE(old_registration->HasOneRef()); |
| |
| scoped_refptr<ServiceWorkerRegistration> new_registration_by_scope = |
| FindRegistrationForScope(options.scope); |
| |
| EXPECT_EQ(new_registration_by_scope, old_registration); |
| } |
| |
| // Make sure that the same registration is used and the update_via_cache value |
| // is updated when registering a duplicate scope+script_url with a different |
| // update_via_cache value. |
| TEST_F(ServiceWorkerJobTest, RegisterWithDifferentUpdateViaCache) { |
| GURL script_url("https://www.example.com/service_worker.js"); |
| blink::mojom::ServiceWorkerRegistrationOptions options; |
| options.scope = GURL("https://www.example.com/"); |
| |
| scoped_refptr<ServiceWorkerRegistration> old_registration = |
| RunRegisterJob(script_url, options); |
| |
| EXPECT_EQ(blink::mojom::ServiceWorkerUpdateViaCache::kImports, |
| old_registration->update_via_cache()); |
| |
| // During the above registration, a service worker registration object host |
| // for ServiceWorkerGlobalScope#registration has been created/added into |
| // |provider_host|. |
| ServiceWorkerProviderHost* provider_host = |
| old_registration->active_version()->provider_host(); |
| ASSERT_NE(nullptr, provider_host); |
| |
| // Clear all service worker object hosts. |
| provider_host->service_worker_object_hosts_.clear(); |
| // Ensure that the registration's object host doesn't have the reference. |
| EXPECT_EQ(1UL, provider_host->registration_object_hosts_.size()); |
| provider_host->registration_object_hosts_.clear(); |
| EXPECT_EQ(0UL, provider_host->registration_object_hosts_.size()); |
| ASSERT_TRUE(old_registration->HasOneRef()); |
| |
| scoped_refptr<ServiceWorkerRegistration> old_registration_by_scope = |
| FindRegistrationForScope(options.scope); |
| |
| ASSERT_TRUE(old_registration_by_scope.get()); |
| |
| options.update_via_cache = blink::mojom::ServiceWorkerUpdateViaCache::kNone; |
| scoped_refptr<ServiceWorkerRegistration> new_registration = |
| RunRegisterJob(script_url, options); |
| |
| // Ensure that the registration object is not copied. |
| ASSERT_EQ(old_registration, new_registration); |
| EXPECT_EQ(blink::mojom::ServiceWorkerUpdateViaCache::kNone, |
| new_registration->update_via_cache()); |
| |
| ASSERT_FALSE(old_registration->HasOneRef()); |
| |
| scoped_refptr<ServiceWorkerRegistration> new_registration_by_scope = |
| FindRegistrationForScope(options.scope); |
| |
| EXPECT_EQ(new_registration_by_scope, old_registration); |
| } |
| |
| // An instance client that breaks the Mojo connection upon receiving the |
| // Start() message. |
| class FailStartInstanceClient : public FakeEmbeddedWorkerInstanceClient { |
| public: |
| FailStartInstanceClient(EmbeddedWorkerTestHelper* helper) |
| : FakeEmbeddedWorkerInstanceClient(helper) {} |
| |
| void StartWorker(blink::mojom::EmbeddedWorkerStartParamsPtr params) override { |
| // Don't save the Mojo ptrs. The connection breaks. |
| } |
| }; |
| |
| TEST_F(ServiceWorkerJobTest, Register_FailToStartWorker) { |
| helper_->AddPendingInstanceClient( |
| std::make_unique<FailStartInstanceClient>(helper_.get())); |
| |
| blink::mojom::ServiceWorkerRegistrationOptions options; |
| options.scope = GURL("https://www.example.com/"); |
| scoped_refptr<ServiceWorkerRegistration> registration = |
| RunRegisterJob(GURL("https://www.example.com/service_worker.js"), options, |
| blink::ServiceWorkerStatusCode::kErrorStartWorkerFailed); |
| |
| ASSERT_EQ(scoped_refptr<ServiceWorkerRegistration>(nullptr), registration); |
| } |
| |
| // Register and then unregister the scope, in parallel. Job coordinator should |
| // process jobs until the last job. |
| TEST_F(ServiceWorkerJobTest, ParallelRegUnreg) { |
| GURL script_url("https://www.example.com/service_worker.js"); |
| blink::mojom::ServiceWorkerRegistrationOptions options; |
| options.scope = GURL("https://www.example.com/"); |
| |
| bool registration_called = false; |
| scoped_refptr<ServiceWorkerRegistration> registration; |
| job_coordinator()->Register( |
| script_url, options, |
| SaveRegistration(blink::ServiceWorkerStatusCode::kOk, |
| ®istration_called, ®istration)); |
| |
| bool unregistration_called = false; |
| job_coordinator()->Unregister( |
| options.scope, SaveUnregistration(blink::ServiceWorkerStatusCode::kOk, |
| &unregistration_called)); |
| |
| ASSERT_FALSE(registration_called); |
| ASSERT_FALSE(unregistration_called); |
| base::RunLoop().RunUntilIdle(); |
| ASSERT_TRUE(registration_called); |
| ASSERT_TRUE(unregistration_called); |
| |
| registration = FindRegistrationForScope( |
| options.scope, blink::ServiceWorkerStatusCode::kErrorNotFound); |
| |
| ASSERT_EQ(scoped_refptr<ServiceWorkerRegistration>(), registration); |
| } |
| |
| // Register conflicting scripts for the same scope. The most recent |
| // registration should win, and the old registration should have been |
| // shutdown. |
| TEST_F(ServiceWorkerJobTest, ParallelRegNewScript) { |
| GURL scope("https://www.example.com/"); |
| |
| GURL script_url1("https://www.example.com/service_worker1.js"); |
| bool registration1_called = false; |
| scoped_refptr<ServiceWorkerRegistration> registration1; |
| job_coordinator()->Register( |
| script_url1, |
| blink::mojom::ServiceWorkerRegistrationOptions( |
| scope, blink::mojom::ScriptType::kClassic, |
| blink::mojom::ServiceWorkerUpdateViaCache::kNone), |
| SaveRegistration(blink::ServiceWorkerStatusCode::kOk, |
| ®istration1_called, ®istration1)); |
| |
| GURL script_url2("https://www.example.com/service_worker2.js"); |
| bool registration2_called = false; |
| scoped_refptr<ServiceWorkerRegistration> registration2; |
| job_coordinator()->Register( |
| script_url2, |
| blink::mojom::ServiceWorkerRegistrationOptions( |
| scope, blink::mojom::ScriptType::kClassic, |
| blink::mojom::ServiceWorkerUpdateViaCache::kAll), |
| SaveRegistration(blink::ServiceWorkerStatusCode::kOk, |
| ®istration2_called, ®istration2)); |
| |
| ASSERT_FALSE(registration1_called); |
| ASSERT_FALSE(registration2_called); |
| base::RunLoop().RunUntilIdle(); |
| ASSERT_TRUE(registration1_called); |
| ASSERT_TRUE(registration2_called); |
| |
| scoped_refptr<ServiceWorkerRegistration> registration = |
| FindRegistrationForScope(scope); |
| |
| ASSERT_EQ(registration2, registration); |
| } |
| |
| // Register the exact same scope + script. Requests should be |
| // coalesced such that both callers get the exact same registration |
| // object. |
| TEST_F(ServiceWorkerJobTest, ParallelRegSameScript) { |
| blink::mojom::ServiceWorkerRegistrationOptions options; |
| options.scope = GURL("https://www.example.com/"); |
| |
| GURL script_url("https://www.example.com/service_worker1.js"); |
| bool registration1_called = false; |
| scoped_refptr<ServiceWorkerRegistration> registration1; |
| job_coordinator()->Register( |
| script_url, options, |
| SaveRegistration(blink::ServiceWorkerStatusCode::kOk, |
| ®istration1_called, ®istration1)); |
| |
| bool registration2_called = false; |
| scoped_refptr<ServiceWorkerRegistration> registration2; |
| job_coordinator()->Register( |
| script_url, options, |
| SaveRegistration(blink::ServiceWorkerStatusCode::kOk, |
| ®istration2_called, ®istration2)); |
| |
| ASSERT_FALSE(registration1_called); |
| ASSERT_FALSE(registration2_called); |
| base::RunLoop().RunUntilIdle(); |
| ASSERT_TRUE(registration1_called); |
| ASSERT_TRUE(registration2_called); |
| |
| ASSERT_EQ(registration1, registration2); |
| |
| scoped_refptr<ServiceWorkerRegistration> registration = |
| FindRegistrationForScope(options.scope); |
| |
| ASSERT_EQ(registration, registration1); |
| } |
| |
| // Call simulataneous unregister calls. |
| TEST_F(ServiceWorkerJobTest, ParallelUnreg) { |
| GURL scope("https://www.example.com/"); |
| |
| GURL script_url("https://www.example.com/service_worker.js"); |
| bool unregistration1_called = false; |
| job_coordinator()->Unregister( |
| scope, SaveUnregistration(blink::ServiceWorkerStatusCode::kErrorNotFound, |
| &unregistration1_called)); |
| |
| bool unregistration2_called = false; |
| job_coordinator()->Unregister( |
| scope, SaveUnregistration(blink::ServiceWorkerStatusCode::kErrorNotFound, |
| &unregistration2_called)); |
| |
| ASSERT_FALSE(unregistration1_called); |
| ASSERT_FALSE(unregistration2_called); |
| base::RunLoop().RunUntilIdle(); |
| ASSERT_TRUE(unregistration1_called); |
| ASSERT_TRUE(unregistration2_called); |
| |
| // There isn't really a way to test that they are being coalesced, |
| // but we can make sure they can exist simultaneously without |
| // crashing. |
| scoped_refptr<ServiceWorkerRegistration> registration = |
| FindRegistrationForScope(scope, |
| blink::ServiceWorkerStatusCode::kErrorNotFound); |
| |
| ASSERT_EQ(scoped_refptr<ServiceWorkerRegistration>(), registration); |
| } |
| |
| TEST_F(ServiceWorkerJobTest, AbortAll_Register) { |
| GURL script_url1("https://www1.example.com/service_worker.js"); |
| GURL script_url2("https://www2.example.com/service_worker.js"); |
| |
| blink::mojom::ServiceWorkerRegistrationOptions options1; |
| options1.scope = GURL("https://www1.example.com/"); |
| blink::mojom::ServiceWorkerRegistrationOptions options2; |
| options2.scope = GURL("https://www2.example.com/"); |
| |
| bool registration_called1 = false; |
| scoped_refptr<ServiceWorkerRegistration> registration1; |
| job_coordinator()->Register( |
| script_url1, options1, |
| SaveRegistration(blink::ServiceWorkerStatusCode::kErrorAbort, |
| ®istration_called1, ®istration1)); |
| |
| bool registration_called2 = false; |
| scoped_refptr<ServiceWorkerRegistration> registration2; |
| job_coordinator()->Register( |
| script_url2, options2, |
| SaveRegistration(blink::ServiceWorkerStatusCode::kErrorAbort, |
| ®istration_called2, ®istration2)); |
| |
| ASSERT_FALSE(registration_called1); |
| ASSERT_FALSE(registration_called2); |
| job_coordinator()->AbortAll(); |
| |
| base::RunLoop().RunUntilIdle(); |
| ASSERT_TRUE(registration_called1); |
| ASSERT_TRUE(registration_called2); |
| |
| bool find_called1 = false; |
| storage()->FindRegistrationForScope( |
| options1.scope, |
| SaveFoundRegistration(blink::ServiceWorkerStatusCode::kErrorNotFound, |
| &find_called1, ®istration1)); |
| |
| bool find_called2 = false; |
| storage()->FindRegistrationForScope( |
| options2.scope, |
| SaveFoundRegistration(blink::ServiceWorkerStatusCode::kErrorNotFound, |
| &find_called2, ®istration2)); |
| |
| base::RunLoop().RunUntilIdle(); |
| ASSERT_TRUE(find_called1); |
| ASSERT_TRUE(find_called2); |
| EXPECT_EQ(scoped_refptr<ServiceWorkerRegistration>(), registration1); |
| EXPECT_EQ(scoped_refptr<ServiceWorkerRegistration>(), registration2); |
| } |
| |
| TEST_F(ServiceWorkerJobTest, AbortAll_Unregister) { |
| GURL scope1("https://www1.example.com/"); |
| GURL scope2("https://www2.example.com/"); |
| |
| bool unregistration_called1 = false; |
| scoped_refptr<ServiceWorkerRegistration> registration1; |
| job_coordinator()->Unregister( |
| scope1, SaveUnregistration(blink::ServiceWorkerStatusCode::kErrorAbort, |
| &unregistration_called1)); |
| |
| bool unregistration_called2 = false; |
| job_coordinator()->Unregister( |
| scope2, SaveUnregistration(blink::ServiceWorkerStatusCode::kErrorAbort, |
| &unregistration_called2)); |
| |
| ASSERT_FALSE(unregistration_called1); |
| ASSERT_FALSE(unregistration_called2); |
| job_coordinator()->AbortAll(); |
| |
| base::RunLoop().RunUntilIdle(); |
| ASSERT_TRUE(unregistration_called1); |
| ASSERT_TRUE(unregistration_called2); |
| } |
| |
| TEST_F(ServiceWorkerJobTest, AbortAll_RegUnreg) { |
| GURL script_url("https://www.example.com/service_worker.js"); |
| blink::mojom::ServiceWorkerRegistrationOptions options; |
| options.scope = GURL("https://www.example.com/"); |
| |
| bool registration_called = false; |
| scoped_refptr<ServiceWorkerRegistration> registration; |
| job_coordinator()->Register( |
| script_url, options, |
| SaveRegistration(blink::ServiceWorkerStatusCode::kErrorAbort, |
| ®istration_called, ®istration)); |
| |
| bool unregistration_called = false; |
| job_coordinator()->Unregister( |
| options.scope, |
| SaveUnregistration(blink::ServiceWorkerStatusCode::kErrorAbort, |
| &unregistration_called)); |
| |
| ASSERT_FALSE(registration_called); |
| ASSERT_FALSE(unregistration_called); |
| job_coordinator()->AbortAll(); |
| |
| base::RunLoop().RunUntilIdle(); |
| ASSERT_TRUE(registration_called); |
| ASSERT_TRUE(unregistration_called); |
| |
| registration = FindRegistrationForScope( |
| options.scope, blink::ServiceWorkerStatusCode::kErrorNotFound); |
| |
| EXPECT_EQ(scoped_refptr<ServiceWorkerRegistration>(), registration); |
| } |
| |
| TEST_F(ServiceWorkerJobTest, AbortScope) { |
| GURL script_url("https://www.example.com/service_worker.js"); |
| blink::mojom::ServiceWorkerRegistrationOptions options1; |
| options1.scope = GURL("https://www.example.com/1"); |
| blink::mojom::ServiceWorkerRegistrationOptions options2; |
| options2.scope = GURL("https://www.example.com/2"); |
| |
| bool registration1_called = false; |
| scoped_refptr<ServiceWorkerRegistration> registration1; |
| job_coordinator()->Register( |
| script_url, options1, |
| SaveRegistration(blink::ServiceWorkerStatusCode::kErrorAbort, |
| ®istration1_called, ®istration1)); |
| |
| bool registration2_called = false; |
| scoped_refptr<ServiceWorkerRegistration> registration2; |
| job_coordinator()->Register( |
| script_url, options2, |
| SaveRegistration(blink::ServiceWorkerStatusCode::kOk, |
| ®istration2_called, ®istration2)); |
| |
| ASSERT_FALSE(registration1_called); |
| ASSERT_FALSE(registration2_called); |
| job_coordinator()->Abort(options1.scope); |
| |
| base::RunLoop().RunUntilIdle(); |
| ASSERT_TRUE(registration1_called); |
| ASSERT_TRUE(registration2_called); |
| |
| registration1 = FindRegistrationForScope( |
| options1.scope, blink::ServiceWorkerStatusCode::kErrorNotFound); |
| EXPECT_EQ(nullptr, registration1); |
| |
| registration2 = FindRegistrationForScope(options2.scope, |
| blink::ServiceWorkerStatusCode::kOk); |
| EXPECT_NE(nullptr, registration2); |
| } |
| |
| // Tests that the waiting worker enters the 'redundant' state upon |
| // unregistration. |
| TEST_F(ServiceWorkerJobTest, UnregisterWaitingSetsRedundant) { |
| GURL script_url("https://www.example.com/service_worker.js"); |
| blink::mojom::ServiceWorkerRegistrationOptions options; |
| options.scope = GURL("https://www.example.com/"); |
| scoped_refptr<ServiceWorkerRegistration> registration = |
| RunRegisterJob(script_url, options); |
| ASSERT_TRUE(registration.get()); |
| |
| // Manually create the waiting worker since there is no way to become a |
| // waiting worker until Update is implemented. |
| scoped_refptr<ServiceWorkerVersion> version = new ServiceWorkerVersion( |
| registration.get(), script_url, blink::mojom::ScriptType::kClassic, 1L, |
| helper_->context()->AsWeakPtr()); |
| base::Optional<blink::ServiceWorkerStatusCode> status; |
| base::RunLoop run_loop; |
| version->StartWorker( |
| ServiceWorkerMetrics::EventType::UNKNOWN, |
| ReceiveServiceWorkerStatus(&status, run_loop.QuitClosure())); |
| run_loop.Run(); |
| ASSERT_EQ(blink::ServiceWorkerStatusCode::kOk, status.value()); |
| |
| version->set_fetch_handler_existence( |
| ServiceWorkerVersion::FetchHandlerExistence::EXISTS); |
| version->SetStatus(ServiceWorkerVersion::INSTALLED); |
| registration->SetWaitingVersion(version); |
| EXPECT_EQ(EmbeddedWorkerStatus::RUNNING, version->running_status()); |
| EXPECT_EQ(ServiceWorkerVersion::INSTALLED, version->status()); |
| |
| RunUnregisterJob(GURL("https://www.example.com/")); |
| |
| // The version should be stopped since there is no controllee after |
| // unregistration. |
| EXPECT_EQ(EmbeddedWorkerStatus::STOPPED, version->running_status()); |
| EXPECT_EQ(ServiceWorkerVersion::REDUNDANT, version->status()); |
| } |
| |
| // Tests that the active worker enters the 'redundant' state upon |
| // unregistration. |
| TEST_F(ServiceWorkerJobTest, UnregisterActiveSetsRedundant) { |
| blink::mojom::ServiceWorkerRegistrationOptions options; |
| options.scope = GURL("https://www.example.com/"); |
| scoped_refptr<ServiceWorkerRegistration> registration = RunRegisterJob( |
| GURL("https://www.example.com/service_worker.js"), options); |
| ASSERT_TRUE(registration.get()); |
| |
| scoped_refptr<ServiceWorkerVersion> version = registration->active_version(); |
| EXPECT_EQ(EmbeddedWorkerStatus::RUNNING, version->running_status()); |
| EXPECT_EQ(ServiceWorkerVersion::ACTIVATED, version->status()); |
| |
| RunUnregisterJob(GURL("https://www.example.com/")); |
| |
| // The version should be stopped since there is no controllee after |
| // unregistration. |
| EXPECT_EQ(EmbeddedWorkerStatus::STOPPED, version->running_status()); |
| EXPECT_EQ(ServiceWorkerVersion::REDUNDANT, version->status()); |
| } |
| |
| // Tests that the active worker enters the 'redundant' state upon |
| // unregistration. |
| TEST_F(ServiceWorkerJobTest, |
| UnregisterActiveSetsRedundant_WaitForNoControllee) { |
| blink::mojom::ServiceWorkerRegistrationOptions options; |
| options.scope = GURL("https://www.example.com/"); |
| scoped_refptr<ServiceWorkerRegistration> registration = RunRegisterJob( |
| GURL("https://www.example.com/service_worker.js"), options); |
| ASSERT_TRUE(registration.get()); |
| |
| ServiceWorkerProviderHost* host = CreateControllee(); |
| registration->active_version()->AddControllee(host); |
| |
| scoped_refptr<ServiceWorkerVersion> version = registration->active_version(); |
| EXPECT_EQ(EmbeddedWorkerStatus::RUNNING, version->running_status()); |
| EXPECT_EQ(ServiceWorkerVersion::ACTIVATED, version->status()); |
| |
| RunUnregisterJob(GURL("https://www.example.com/")); |
| |
| // The version should be running since there is still a controllee. |
| EXPECT_EQ(EmbeddedWorkerStatus::RUNNING, version->running_status()); |
| EXPECT_EQ(ServiceWorkerVersion::ACTIVATED, version->status()); |
| |
| registration->active_version()->RemoveControllee(host->client_uuid()); |
| base::RunLoop().RunUntilIdle(); |
| |
| // The version should be stopped since there is no controllee. |
| EXPECT_EQ(EmbeddedWorkerStatus::STOPPED, version->running_status()); |
| EXPECT_EQ(ServiceWorkerVersion::REDUNDANT, version->status()); |
| } |
| |
| namespace { // Helpers for the update job tests. |
| |
| const GURL kNoChangeOrigin("https://nochange/"); |
| const GURL kNewVersionOrigin("https://newversion/"); |
| const char kScope[] = "scope/"; |
| const char kScript[] = "script.js"; |
| |
| void RunNestedUntilIdle() { |
| base::RunLoop(base::RunLoop::Type::kNestableTasksAllowed).RunUntilIdle(); |
| } |
| |
| void OnIOComplete(int* rv_out, int rv) { |
| *rv_out = rv; |
| } |
| |
| void WriteResponse(ServiceWorkerStorage* storage, |
| int64_t id, |
| const std::string& headers, |
| IOBuffer* body, |
| int length) { |
| std::unique_ptr<ServiceWorkerResponseWriter> writer = |
| storage->CreateResponseWriter(id); |
| |
| std::unique_ptr<net::HttpResponseInfo> info = |
| std::make_unique<net::HttpResponseInfo>(); |
| info->request_time = base::Time::Now(); |
| info->response_time = base::Time::Now(); |
| info->was_cached = false; |
| info->headers = new net::HttpResponseHeaders(headers); |
| scoped_refptr<HttpResponseInfoIOBuffer> info_buffer = |
| base::MakeRefCounted<HttpResponseInfoIOBuffer>(std::move(info)); |
| |
| int rv = -1234; |
| writer->WriteInfo(info_buffer.get(), base::BindOnce(&OnIOComplete, &rv)); |
| RunNestedUntilIdle(); |
| EXPECT_LT(0, rv); |
| |
| rv = -1234; |
| writer->WriteData(body, length, base::BindOnce(&OnIOComplete, &rv)); |
| RunNestedUntilIdle(); |
| EXPECT_EQ(length, rv); |
| } |
| |
| void WriteStringResponse(ServiceWorkerStorage* storage, |
| int64_t id, |
| const std::string& body) { |
| scoped_refptr<IOBuffer> body_buffer = |
| base::MakeRefCounted<WrappedIOBuffer>(body.data()); |
| const char kHttpHeaders[] = "HTTP/1.0 200 HONKYDORY\0\0"; |
| std::string headers(kHttpHeaders, base::size(kHttpHeaders)); |
| WriteResponse(storage, id, headers, body_buffer.get(), body.length()); |
| } |
| |
| class UpdateJobTestHelper : public EmbeddedWorkerTestHelper, |
| public ServiceWorkerRegistration::Listener, |
| public ServiceWorkerContextCoreObserver { |
| public: |
| struct AttributeChangeLogEntry { |
| int64_t registration_id; |
| blink::mojom::ChangedServiceWorkerObjectsMaskPtr mask; |
| ServiceWorkerRegistrationInfo info; |
| }; |
| |
| struct StateChangeLogEntry { |
| int64_t version_id; |
| ServiceWorkerVersion::Status status; |
| }; |
| |
| UpdateJobTestHelper() |
| : EmbeddedWorkerTestHelper(base::FilePath()), weak_factory_(this) { |
| context_wrapper()->AddObserver(this); |
| } |
| ~UpdateJobTestHelper() override { |
| context_wrapper()->RemoveObserver(this); |
| if (observed_registration_.get()) |
| observed_registration_->RemoveListener(this); |
| } |
| |
| class UpdateJobEmbeddedWorkerInstanceClient |
| : public FakeEmbeddedWorkerInstanceClient { |
| public: |
| UpdateJobEmbeddedWorkerInstanceClient(UpdateJobTestHelper* helper) |
| : FakeEmbeddedWorkerInstanceClient(helper) {} |
| ~UpdateJobEmbeddedWorkerInstanceClient() override = default; |
| |
| void set_force_start_worker_failure(bool force_start_worker_failure) { |
| force_start_worker_failure_ = force_start_worker_failure; |
| } |
| |
| void ResumeAfterDownload() override { |
| if (force_start_worker_failure_) { |
| host()->OnScriptEvaluationStart(); |
| host()->OnStarted( |
| blink::mojom::ServiceWorkerStartStatus::kAbruptCompletion, |
| helper()->GetNextThreadId(), |
| blink::mojom::EmbeddedWorkerStartTiming::New()); |
| return; |
| } |
| FakeEmbeddedWorkerInstanceClient::ResumeAfterDownload(); |
| } |
| |
| private: |
| bool force_start_worker_failure_ = false; |
| }; |
| |
| ServiceWorkerStorage* storage() { return context()->storage(); } |
| ServiceWorkerJobCoordinator* job_coordinator() { |
| return context()->job_coordinator(); |
| } |
| |
| scoped_refptr<ServiceWorkerRegistration> SetupInitialRegistration( |
| const GURL& test_origin) { |
| blink::mojom::ServiceWorkerRegistrationOptions options; |
| options.scope = test_origin.Resolve(kScope); |
| scoped_refptr<ServiceWorkerRegistration> registration; |
| bool called = false; |
| |
| auto client = std::make_unique<UpdateJobEmbeddedWorkerInstanceClient>(this); |
| initial_embedded_worker_instance_client_ = client.get(); |
| AddPendingInstanceClient(std::move(client)); |
| |
| job_coordinator()->Register( |
| test_origin.Resolve(kScript), options, |
| SaveRegistration(blink::ServiceWorkerStatusCode::kOk, &called, |
| ®istration)); |
| base::RunLoop().RunUntilIdle(); |
| EXPECT_TRUE(called); |
| EXPECT_TRUE(registration.get()); |
| EXPECT_TRUE(registration->active_version()); |
| EXPECT_FALSE(registration->installing_version()); |
| EXPECT_FALSE(registration->waiting_version()); |
| observed_registration_ = registration; |
| return registration; |
| } |
| |
| // EmbeddedWorkerTestHelper overrides: |
| void PopulateScriptCacheMap(int64_t version_id, |
| base::OnceClosure callback) override { |
| const std::string kMockScriptBody = "mock_script"; |
| const uint64_t kMockScriptSize = 19284; |
| ServiceWorkerVersion* version = context()->GetLiveVersion(version_id); |
| ASSERT_TRUE(version); |
| ServiceWorkerRegistration* registration = |
| context()->GetLiveRegistration(version->registration_id()); |
| ASSERT_TRUE(registration); |
| GURL script = version->script_url(); |
| bool is_update = registration->active_version() && |
| version != registration->active_version(); |
| |
| // Simulate network access. |
| base::TimeDelta time_since_last_check = |
| base::Time::Now() - registration->last_update_check(); |
| if (!is_update || script.GetOrigin() != kNoChangeOrigin || |
| time_since_last_check > kServiceWorkerScriptMaxCacheAge) { |
| version->embedded_worker()->OnNetworkAccessedForScriptLoad(); |
| } |
| |
| int64_t resource_id = storage()->NewResourceId(); |
| version->script_cache_map()->NotifyStartedCaching(script, resource_id); |
| if (!is_update) { |
| // Spoof caching the script for the initial version. |
| WriteStringResponse(storage(), resource_id, kMockScriptBody); |
| version->script_cache_map()->NotifyFinishedCaching( |
| script, kMockScriptSize, net::OK, std::string()); |
| } else if (script.GetOrigin() == kNoChangeOrigin) { |
| // Simulate fetching the updated script and finding it's identical to |
| // the incumbent. |
| version->script_cache_map()->NotifyFinishedCaching( |
| script, kMockScriptSize, net::ERR_FILE_EXISTS, std::string()); |
| } else { |
| // Spoof caching the script for the new version. |
| WriteStringResponse(storage(), resource_id, "mock_different_script"); |
| version->script_cache_map()->NotifyFinishedCaching( |
| script, kMockScriptSize, net::OK, std::string()); |
| } |
| version->SetMainScriptHttpResponseInfo(CreateHttpResponseInfo()); |
| std::move(callback).Run(); |
| } |
| |
| // ServiceWorkerContextCoreObserver overrides |
| void OnVersionStateChanged(int64_t version_id, |
| const GURL& scope, |
| ServiceWorkerVersion::Status status) override { |
| StateChangeLogEntry entry; |
| entry.version_id = version_id; |
| entry.status = status; |
| state_change_log_.push_back(std::move(entry)); |
| } |
| |
| // ServiceWorkerRegistration::Listener overrides |
| void OnVersionAttributesChanged( |
| ServiceWorkerRegistration* registration, |
| blink::mojom::ChangedServiceWorkerObjectsMaskPtr changed_mask, |
| const ServiceWorkerRegistrationInfo& info) override { |
| AttributeChangeLogEntry entry; |
| entry.registration_id = registration->id(); |
| entry.mask = std::move(changed_mask); |
| entry.info = info; |
| attribute_change_log_.push_back(std::move(entry)); |
| } |
| |
| void OnRegistrationFailed(ServiceWorkerRegistration* registration) override { |
| registration_failed_ = true; |
| } |
| |
| void OnUpdateFound(ServiceWorkerRegistration* registration) override { |
| update_found_ = true; |
| } |
| |
| UpdateJobEmbeddedWorkerInstanceClient* |
| initial_embedded_worker_instance_client_ = nullptr; |
| scoped_refptr<ServiceWorkerRegistration> observed_registration_; |
| std::vector<AttributeChangeLogEntry> attribute_change_log_; |
| std::vector<StateChangeLogEntry> state_change_log_; |
| bool update_found_ = false; |
| bool registration_failed_ = false; |
| bool force_start_worker_failure_ = false; |
| base::Optional<bool> will_be_terminated_; |
| |
| base::WeakPtrFactory<UpdateJobTestHelper> weak_factory_; |
| }; |
| |
| void RequestTermination( |
| blink::mojom::EmbeddedWorkerInstanceHostAssociatedPtr* host) { |
| // We can't wait for the callback since StopWorker() arrives before it which |
| // severs the Mojo connection. |
| (*host)->RequestTermination(base::DoNothing()); |
| } |
| |
| } // namespace |
| |
| TEST_F(ServiceWorkerJobTest, Update_NoChange) { |
| UpdateJobTestHelper* update_helper = new UpdateJobTestHelper; |
| helper_.reset(update_helper); |
| scoped_refptr<ServiceWorkerRegistration> registration = |
| update_helper->SetupInitialRegistration(kNoChangeOrigin); |
| ASSERT_TRUE(registration.get()); |
| ASSERT_EQ(4u, update_helper->state_change_log_.size()); |
| EXPECT_EQ(ServiceWorkerVersion::INSTALLING, |
| update_helper->state_change_log_[0].status); |
| EXPECT_EQ(ServiceWorkerVersion::INSTALLED, |
| update_helper->state_change_log_[1].status); |
| EXPECT_EQ(ServiceWorkerVersion::ACTIVATING, |
| update_helper->state_change_log_[2].status); |
| EXPECT_EQ(ServiceWorkerVersion::ACTIVATED, |
| update_helper->state_change_log_[3].status); |
| update_helper->state_change_log_.clear(); |
| |
| // Run the update job. |
| registration->AddListener(update_helper); |
| scoped_refptr<ServiceWorkerVersion> first_version = |
| registration->active_version(); |
| first_version->StartUpdate(); |
| base::RunLoop().RunUntilIdle(); |
| |
| // Verify results. |
| ASSERT_TRUE(registration->active_version()); |
| EXPECT_EQ(first_version.get(), registration->active_version()); |
| EXPECT_FALSE(registration->installing_version()); |
| EXPECT_FALSE(registration->waiting_version()); |
| EXPECT_TRUE(update_helper->attribute_change_log_.empty()); |
| ASSERT_EQ(1u, update_helper->state_change_log_.size()); |
| EXPECT_NE(registration->active_version()->version_id(), |
| update_helper->state_change_log_[0].version_id); |
| EXPECT_EQ(ServiceWorkerVersion::REDUNDANT, |
| update_helper->state_change_log_[0].status); |
| EXPECT_FALSE(update_helper->update_found_); |
| } |
| |
| TEST_F(ServiceWorkerJobTest, Update_BumpLastUpdateCheckTime) { |
| const base::Time kToday = base::Time::Now(); |
| const base::Time kYesterday = |
| kToday - base::TimeDelta::FromDays(1) - base::TimeDelta::FromHours(1); |
| UpdateJobTestHelper* update_helper = new UpdateJobTestHelper; |
| helper_.reset(update_helper); |
| scoped_refptr<ServiceWorkerRegistration> registration = |
| update_helper->SetupInitialRegistration(kNoChangeOrigin); |
| ASSERT_TRUE(registration.get()); |
| |
| registration->AddListener(update_helper); |
| |
| // Run an update where the script did not change and the network was not |
| // accessed. The check time should not be updated. |
| registration->set_last_update_check(kToday); |
| registration->active_version()->StartUpdate(); |
| base::RunLoop().RunUntilIdle(); |
| EXPECT_EQ(kToday, registration->last_update_check()); |
| EXPECT_FALSE(update_helper->update_found_); |
| |
| // Run an update where the script did not change and the network was accessed. |
| // The check time should be updated. |
| registration->set_last_update_check(kYesterday); |
| registration->active_version()->StartUpdate(); |
| base::RunLoop().RunUntilIdle(); |
| EXPECT_LT(kYesterday, registration->last_update_check()); |
| EXPECT_FALSE(update_helper->update_found_); |
| registration->RemoveListener(update_helper); |
| |
| registration = update_helper->SetupInitialRegistration(kNewVersionOrigin); |
| ASSERT_TRUE(registration.get()); |
| |
| registration->AddListener(update_helper); |
| |
| // Run an update where the script changed. The check time should be updated. |
| registration->set_last_update_check(kYesterday); |
| registration->active_version()->StartUpdate(); |
| base::RunLoop().RunUntilIdle(); |
| EXPECT_LT(kYesterday, registration->last_update_check()); |
| |
| // Run an update to a worker that loads successfully but fails to start up |
| // (script evaluation failure). The check time should be updated. |
| auto* embedded_worker_instance_client = |
| update_helper->AddNewPendingInstanceClient< |
| UpdateJobTestHelper::UpdateJobEmbeddedWorkerInstanceClient>( |
| update_helper); |
| embedded_worker_instance_client->set_force_start_worker_failure(true); |
| registration->set_last_update_check(kYesterday); |
| registration->active_version()->StartUpdate(); |
| base::RunLoop().RunUntilIdle(); |
| EXPECT_LT(kYesterday, registration->last_update_check()); |
| } |
| |
| TEST_F(ServiceWorkerJobTest, Update_NewVersion) { |
| UpdateJobTestHelper* update_helper = new UpdateJobTestHelper; |
| helper_.reset(update_helper); |
| scoped_refptr<ServiceWorkerRegistration> registration = |
| update_helper->SetupInitialRegistration(kNewVersionOrigin); |
| ASSERT_TRUE(registration.get()); |
| update_helper->state_change_log_.clear(); |
| auto runner = base::MakeRefCounted<base::TestSimpleTaskRunner>(); |
| registration->SetTaskRunnerForTest(runner); |
| |
| // Run the update job. |
| registration->AddListener(update_helper); |
| scoped_refptr<ServiceWorkerVersion> first_version = |
| registration->active_version(); |
| first_version->StartUpdate(); |
| base::RunLoop().RunUntilIdle(); |
| |
| // The worker is updated after RequestTermination() is called from the |
| // renderer. Until then, the active version stays active. |
| EXPECT_EQ(first_version.get(), registration->active_version()); |
| // The new worker is installed but not yet to be activated. |
| scoped_refptr<ServiceWorkerVersion> new_version = |
| registration->waiting_version(); |
| EXPECT_EQ(2u, update_helper->attribute_change_log_.size()); |
| UpdateJobTestHelper::UpdateJobEmbeddedWorkerInstanceClient* client = |
| update_helper->initial_embedded_worker_instance_client_; |
| RequestTermination(&client->host()); |
| |
| TestServiceWorkerObserver observer(helper_->context_wrapper()); |
| observer.RunUntilActivated(new_version.get(), runner); |
| |
| // Pump the loop again. This ensures |update_helper| observes all |
| // the status changes, since RunUntilActivated() only ensured |
| // ServiceWorkerJobTest did. |
| base::RunLoop().RunUntilIdle(); |
| |
| // Verify results. |
| ASSERT_TRUE(registration->active_version()); |
| EXPECT_NE(first_version.get(), registration->active_version()); |
| EXPECT_FALSE(registration->installing_version()); |
| EXPECT_FALSE(registration->waiting_version()); |
| ASSERT_EQ(3u, update_helper->attribute_change_log_.size()); |
| |
| { |
| const UpdateJobTestHelper::AttributeChangeLogEntry& entry = |
| update_helper->attribute_change_log_[0]; |
| EXPECT_TRUE(entry.mask->installing); |
| EXPECT_FALSE(entry.mask->waiting); |
| EXPECT_FALSE(entry.mask->active); |
| EXPECT_NE(entry.info.installing_version.version_id, |
| blink::mojom::kInvalidServiceWorkerVersionId); |
| EXPECT_EQ(entry.info.waiting_version.version_id, |
| blink::mojom::kInvalidServiceWorkerVersionId); |
| EXPECT_NE(entry.info.active_version.version_id, |
| blink::mojom::kInvalidServiceWorkerVersionId); |
| } |
| |
| { |
| const UpdateJobTestHelper::AttributeChangeLogEntry& entry = |
| update_helper->attribute_change_log_[1]; |
| EXPECT_TRUE(entry.mask->installing); |
| EXPECT_TRUE(entry.mask->waiting); |
| EXPECT_FALSE(entry.mask->active); |
| EXPECT_EQ(entry.info.installing_version.version_id, |
| blink::mojom::kInvalidServiceWorkerVersionId); |
| EXPECT_NE(entry.info.waiting_version.version_id, |
| blink::mojom::kInvalidServiceWorkerVersionId); |
| EXPECT_NE(entry.info.active_version.version_id, |
| blink::mojom::kInvalidServiceWorkerVersionId); |
| } |
| |
| { |
| const UpdateJobTestHelper::AttributeChangeLogEntry& entry = |
| update_helper->attribute_change_log_[2]; |
| EXPECT_FALSE(entry.mask->installing); |
| EXPECT_TRUE(entry.mask->waiting); |
| EXPECT_TRUE(entry.mask->active); |
| EXPECT_EQ(entry.info.installing_version.version_id, |
| blink::mojom::kInvalidServiceWorkerVersionId); |
| EXPECT_EQ(entry.info.waiting_version.version_id, |
| blink::mojom::kInvalidServiceWorkerVersionId); |
| EXPECT_NE(entry.info.active_version.version_id, |
| blink::mojom::kInvalidServiceWorkerVersionId); |
| } |
| |
| // expected version state transitions: |
| // new.installing, new.installed, |
| // old.redundant, |
| // new.activating, new.activated |
| ASSERT_EQ(5u, update_helper->state_change_log_.size()); |
| |
| EXPECT_EQ(registration->active_version()->version_id(), |
| update_helper->state_change_log_[0].version_id); |
| EXPECT_EQ(ServiceWorkerVersion::INSTALLING, |
| update_helper->state_change_log_[0].status); |
| |
| EXPECT_EQ(registration->active_version()->version_id(), |
| update_helper->state_change_log_[1].version_id); |
| EXPECT_EQ(ServiceWorkerVersion::INSTALLED, |
| update_helper->state_change_log_[1].status); |
| |
| EXPECT_EQ(first_version->version_id(), |
| update_helper->state_change_log_[2].version_id); |
| EXPECT_EQ(ServiceWorkerVersion::REDUNDANT, |
| update_helper->state_change_log_[2].status); |
| |
| EXPECT_EQ(registration->active_version()->version_id(), |
| update_helper->state_change_log_[3].version_id); |
| EXPECT_EQ(ServiceWorkerVersion::ACTIVATING, |
| update_helper->state_change_log_[3].status); |
| |
| EXPECT_EQ(registration->active_version()->version_id(), |
| update_helper->state_change_log_[4].version_id); |
| EXPECT_EQ(ServiceWorkerVersion::ACTIVATED, |
| update_helper->state_change_log_[4].status); |
| |
| EXPECT_TRUE(update_helper->update_found_); |
| } |
| |
| // Test that the update job uses the script URL of the newest worker when the |
| // job starts, rather than when it is scheduled. |
| TEST_F(ServiceWorkerJobTest, Update_ScriptUrlChanged) { |
| // Create a registration with an active version. |
| blink::mojom::ServiceWorkerRegistrationOptions options; |
| options.scope = GURL("https://www.example.com/one/"); |
| auto* initial_client = |
| helper_->AddNewPendingInstanceClient<FakeEmbeddedWorkerInstanceClient>( |
| helper_.get()); |
| scoped_refptr<ServiceWorkerRegistration> registration = RunRegisterJob( |
| GURL("https://www.example.com/service_worker.js"), options); |
| auto runner = base::MakeRefCounted<base::TestSimpleTaskRunner>(); |
| registration->SetTaskRunnerForTest(runner); |
| |
| // Queue an Update. When this runs, it will use the waiting version's script. |
| job_coordinator()->Update(registration.get(), false); |
| |
| // Add a waiting version with a new script. |
| GURL new_script("https://www.example.com/new_worker.js"); |
| scoped_refptr<ServiceWorkerVersion> version = new ServiceWorkerVersion( |
| registration.get(), new_script, blink::mojom::ScriptType::kClassic, |
| 2L /* dummy version id */, helper_->context()->AsWeakPtr()); |
| registration->SetWaitingVersion(version); |
| |
| // Run the update job. |
| base::RunLoop().RunUntilIdle(); |
| |
| // The worker is activated after RequestTermination() is called from the |
| // renderer. Until then, the active version stays active. |
| // Still waiting, but the waiting version isn't |version| since another |
| // ServiceWorkerVersion is created during the update job and the job wipes |
| // out the older waiting version. |
| ServiceWorkerVersion* waiting_version = registration->waiting_version(); |
| EXPECT_TRUE(registration->active_version()); |
| EXPECT_TRUE(waiting_version); |
| EXPECT_NE(version.get(), waiting_version); |
| |
| RequestTermination(&initial_client->host()); |
| TestServiceWorkerObserver observer(helper_->context_wrapper()); |
| observer.RunUntilActivated(waiting_version, runner); |
| |
| // The update job should have created a new version with the new script, |
| // and promoted it to the active version. |
| EXPECT_EQ(new_script, registration->active_version()->script_url()); |
| EXPECT_EQ(nullptr, registration->waiting_version()); |
| EXPECT_EQ(nullptr, registration->installing_version()); |
| } |
| |
| // Test that update succeeds if the incumbent worker was evicted |
| // during the update job (this can happen on disk cache failure). |
| TEST_F(ServiceWorkerJobTest, Update_EvictedIncumbent) { |
| UpdateJobTestHelper* update_helper = new UpdateJobTestHelper; |
| helper_.reset(update_helper); |
| scoped_refptr<ServiceWorkerRegistration> registration = |
| update_helper->SetupInitialRegistration(kNewVersionOrigin); |
| ASSERT_TRUE(registration.get()); |
| update_helper->state_change_log_.clear(); |
| |
| registration->AddListener(update_helper); |
| scoped_refptr<ServiceWorkerVersion> first_version = |
| registration->active_version(); |
| auto* instance_client = helper_->AddNewPendingInstanceClient< |
| DelayedFakeEmbeddedWorkerInstanceClient>(helper_.get()); |
| |
| // Start the update job and make it block on the worker starting. |
| // Evict the incumbent during that time. |
| first_version->StartUpdate(); |
| instance_client->RunUntilStartWorker(); |
| registration->DeleteVersion(first_version); |
| |
| // Finish the update job. |
| instance_client->UnblockStartWorker(); |
| base::RunLoop().RunUntilIdle(); |
| |
| // Verify results. |
| ASSERT_TRUE(registration->active_version()); |
| EXPECT_NE(first_version.get(), registration->active_version()); |
| EXPECT_FALSE(registration->installing_version()); |
| EXPECT_FALSE(registration->waiting_version()); |
| EXPECT_EQ(ServiceWorkerVersion::REDUNDANT, first_version->status()); |
| EXPECT_EQ(ServiceWorkerVersion::ACTIVATED, |
| registration->active_version()->status()); |
| ASSERT_EQ(4u, update_helper->attribute_change_log_.size()); |
| EXPECT_TRUE(update_helper->update_found_); |
| EXPECT_TRUE(update_helper->registration_failed_); |
| EXPECT_FALSE(registration->is_uninstalled()); |
| } |
| |
| TEST_F(ServiceWorkerJobTest, Update_UninstallingRegistration) { |
| blink::mojom::ServiceWorkerRegistrationOptions options; |
| options.scope = GURL("https://www.example.com/one/"); |
| bool called; |
| scoped_refptr<ServiceWorkerRegistration> registration = RunRegisterJob( |
| GURL("https://www.example.com/service_worker.js"), options); |
| |
| // Add a controllee and queue an unregister to force the uninstalling state. |
| ServiceWorkerProviderHost* host = CreateControllee(); |
| ServiceWorkerVersion* active_version = registration->active_version(); |
| active_version->AddControllee(host); |
| job_coordinator()->Unregister( |
| GURL("https://www.example.com/one/"), |
| SaveUnregistration(blink::ServiceWorkerStatusCode::kOk, &called)); |
| |
| // Update should abort after it starts and sees uninstalling. |
| job_coordinator()->Update(registration.get(), false); |
| |
| EXPECT_FALSE(called); |
| base::RunLoop().RunUntilIdle(); |
| EXPECT_TRUE(called); |
| |
| // Verify the registration was not modified by the Update. |
| EXPECT_TRUE(registration->is_uninstalling()); |
| EXPECT_EQ(active_version, registration->active_version()); |
| EXPECT_EQ(nullptr, registration->waiting_version()); |
| EXPECT_EQ(nullptr, registration->installing_version()); |
| } |
| |
| TEST_F(ServiceWorkerJobTest, RegisterWhileUninstalling) { |
| GURL script1("https://www.example.com/service_worker.js"); |
| GURL script2("https://www.example.com/service_worker.js?new"); |
| blink::mojom::ServiceWorkerRegistrationOptions options; |
| options.scope = GURL("https://www.example.com/one/"); |
| |
| auto* initial_client = |
| helper_->AddNewPendingInstanceClient<FakeEmbeddedWorkerInstanceClient>( |
| helper_.get()); |
| scoped_refptr<ServiceWorkerRegistration> registration = |
| RunRegisterJob(script1, options); |
| auto runner = base::MakeRefCounted<base::TestSimpleTaskRunner>(); |
| registration->SetTaskRunnerForTest(runner); |
| |
| // Add a controllee and queue an unregister to force the uninstalling state. |
| ServiceWorkerProviderHost* host = CreateControllee(); |
| scoped_refptr<ServiceWorkerVersion> old_version = |
| registration->active_version(); |
| old_version->AddControllee(host); |
| RunUnregisterJob(options.scope); |
| |
| // Register another script. |
| EXPECT_EQ(registration, RunRegisterJob(script2, options)); |
| |
| EXPECT_FALSE(registration->is_uninstalling()); |
| EXPECT_EQ(old_version, registration->active_version()); |
| scoped_refptr<ServiceWorkerVersion> new_version = |
| registration->waiting_version(); |
| |
| // Verify the new version is installed but not activated yet. |
| EXPECT_EQ(nullptr, registration->installing_version()); |
| EXPECT_TRUE(new_version); |
| EXPECT_EQ(EmbeddedWorkerStatus::RUNNING, new_version->running_status()); |
| EXPECT_EQ(ServiceWorkerVersion::INSTALLED, new_version->status()); |
| |
| // Make the old version eligible for eviction. |
| old_version->RemoveControllee(host->client_uuid()); |
| RequestTermination(&initial_client->host()); |
| |
| // Wait for activated. |
| TestServiceWorkerObserver observer(helper_->context_wrapper()); |
| observer.RunUntilActivated(new_version.get(), runner); |
| |
| // Verify state after the new version is activated. |
| EXPECT_FALSE(registration->is_uninstalling()); |
| EXPECT_FALSE(registration->is_uninstalled()); |
| EXPECT_EQ(nullptr, registration->installing_version()); |
| EXPECT_EQ(nullptr, registration->waiting_version()); |
| EXPECT_EQ(new_version, registration->active_version()); |
| EXPECT_EQ(ServiceWorkerVersion::ACTIVATED, new_version->status()); |
| } |
| |
| TEST_F(ServiceWorkerJobTest, RegisterAndUnregisterWhileUninstalling) { |
| GURL script1("https://www.example.com/service_worker.js"); |
| GURL script2("https://www.example.com/service_worker.js?new"); |
| blink::mojom::ServiceWorkerRegistrationOptions options; |
| options.scope = GURL("https://www.example.com/one/"); |
| |
| scoped_refptr<ServiceWorkerRegistration> registration = |
| RunRegisterJob(script1, options); |
| |
| // Add a controllee and queue an unregister to force the uninstalling state. |
| ServiceWorkerProviderHost* host = CreateControllee(); |
| scoped_refptr<ServiceWorkerVersion> old_version = |
| registration->active_version(); |
| old_version->AddControllee(host); |
| RunUnregisterJob(options.scope); |
| |
| EXPECT_EQ(registration, RunRegisterJob(script2, options)); |
| |
| EXPECT_EQ(registration, FindRegistrationForScope(options.scope)); |
| scoped_refptr<ServiceWorkerVersion> new_version = |
| registration->waiting_version(); |
| ASSERT_TRUE(new_version); |
| |
| // Unregister the registration (but it's still live). |
| RunUnregisterJob(options.scope); |
| // Now it's not found in the storage. |
| RunUnregisterJob(options.scope, |
| blink::ServiceWorkerStatusCode::kErrorNotFound); |
| |
| FindRegistrationForScope(options.scope, |
| blink::ServiceWorkerStatusCode::kErrorNotFound); |
| EXPECT_TRUE(registration->is_uninstalling()); |
| EXPECT_EQ(old_version, registration->active_version()); |
| |
| EXPECT_EQ(EmbeddedWorkerStatus::RUNNING, old_version->running_status()); |
| EXPECT_EQ(ServiceWorkerVersion::ACTIVATED, old_version->status()); |
| EXPECT_EQ(EmbeddedWorkerStatus::RUNNING, new_version->running_status()); |
| EXPECT_EQ(ServiceWorkerVersion::INSTALLED, new_version->status()); |
| |
| old_version->RemoveControllee(host->client_uuid()); |
| base::RunLoop().RunUntilIdle(); |
| |
| EXPECT_FALSE(registration->is_uninstalling()); |
| EXPECT_TRUE(registration->is_uninstalled()); |
| |
| EXPECT_EQ(EmbeddedWorkerStatus::STOPPED, old_version->running_status()); |
| EXPECT_EQ(ServiceWorkerVersion::REDUNDANT, old_version->status()); |
| EXPECT_EQ(EmbeddedWorkerStatus::STOPPED, new_version->running_status()); |
| EXPECT_EQ(ServiceWorkerVersion::REDUNDANT, new_version->status()); |
| } |
| |
| TEST_F(ServiceWorkerJobTest, RegisterSameScriptMultipleTimesWhileUninstalling) { |
| GURL script1("https://www.example.com/service_worker.js"); |
| GURL script2("https://www.example.com/service_worker.js?new"); |
| blink::mojom::ServiceWorkerRegistrationOptions options; |
| options.scope = GURL("https://www.example.com/one/"); |
| |
| auto* initial_client = |
| helper_->AddNewPendingInstanceClient<FakeEmbeddedWorkerInstanceClient>( |
| helper_.get()); |
| scoped_refptr<ServiceWorkerRegistration> registration = |
| RunRegisterJob(script1, options); |
| auto runner = base::MakeRefCounted<base::TestSimpleTaskRunner>(); |
| registration->SetTaskRunnerForTest(runner); |
| |
| // Add a controllee and queue an unregister to force the uninstalling state. |
| ServiceWorkerProviderHost* host = CreateControllee(); |
| scoped_refptr<ServiceWorkerVersion> old_version = |
| registration->active_version(); |
| old_version->AddControllee(host); |
| RunUnregisterJob(options.scope); |
| |
| EXPECT_EQ(registration, RunRegisterJob(script2, options)); |
| |
| scoped_refptr<ServiceWorkerVersion> new_version = |
| registration->waiting_version(); |
| ASSERT_TRUE(new_version); |
| |
| RunUnregisterJob(options.scope); |
| |
| EXPECT_TRUE(registration->is_uninstalling()); |
| |
| EXPECT_EQ(registration, RunRegisterJob(script2, options)); |
| |
| EXPECT_FALSE(registration->is_uninstalling()); |
| EXPECT_EQ(new_version, registration->waiting_version()); |
| |
| old_version->RemoveControllee(host->client_uuid()); |
| RequestTermination(&initial_client->host()); |
| |
| // Wait for activated. |
| TestServiceWorkerObserver observer(helper_->context_wrapper()); |
| observer.RunUntilActivated(new_version.get(), runner); |
| |
| // Verify state after the new version is activated. |
| EXPECT_FALSE(registration->is_uninstalling()); |
| EXPECT_FALSE(registration->is_uninstalled()); |
| EXPECT_EQ(nullptr, registration->installing_version()); |
| EXPECT_EQ(nullptr, registration->waiting_version()); |
| EXPECT_EQ(new_version, registration->active_version()); |
| EXPECT_EQ(ServiceWorkerVersion::ACTIVATED, new_version->status()); |
| } |
| |
| TEST_F(ServiceWorkerJobTest, RegisterMultipleTimesWhileUninstalling) { |
| GURL script1("https://www.example.com/service_worker.js?first"); |
| GURL script2("https://www.example.com/service_worker.js?second"); |
| GURL script3("https://www.example.com/service_worker.js?third"); |
| blink::mojom::ServiceWorkerRegistrationOptions options; |
| options.scope = GURL("https://www.example.com/one/"); |
| |
| auto* initial_client = |
| helper_->AddNewPendingInstanceClient<FakeEmbeddedWorkerInstanceClient>( |
| helper_.get()); |
| scoped_refptr<ServiceWorkerRegistration> registration = |
| RunRegisterJob(script1, options); |
| auto runner = base::MakeRefCounted<base::TestSimpleTaskRunner>(); |
| registration->SetTaskRunnerForTest(runner); |
| |
| // Add a controllee and queue an unregister to force the uninstalling state. |
| ServiceWorkerProviderHost* host = CreateControllee(); |
| scoped_refptr<ServiceWorkerVersion> first_version = |
| registration->active_version(); |
| first_version->AddControllee(host); |
| RunUnregisterJob(options.scope); |
| |
| EXPECT_EQ(registration, RunRegisterJob(script2, options)); |
| |
| scoped_refptr<ServiceWorkerVersion> second_version = |
| registration->waiting_version(); |
| ASSERT_TRUE(second_version); |
| |
| RunUnregisterJob(options.scope); |
| |
| EXPECT_TRUE(registration->is_uninstalling()); |
| |
| EXPECT_EQ(registration, RunRegisterJob(script3, options)); |
| |
| scoped_refptr<ServiceWorkerVersion> third_version = |
| registration->waiting_version(); |
| ASSERT_TRUE(third_version); |
| |
| EXPECT_FALSE(registration->is_uninstalling()); |
| EXPECT_EQ(ServiceWorkerVersion::REDUNDANT, second_version->status()); |
| |
| first_version->RemoveControllee(host->client_uuid()); |
| RequestTermination(&initial_client->host()); |
| |
| // Wait for activated. |
| TestServiceWorkerObserver observer(helper_->context_wrapper()); |
| observer.RunUntilActivated(third_version.get(), runner); |
| |
| // Verify the new version is activated. |
| EXPECT_FALSE(registration->is_uninstalling()); |
| EXPECT_FALSE(registration->is_uninstalled()); |
| EXPECT_EQ(nullptr, registration->installing_version()); |
| EXPECT_EQ(nullptr, registration->waiting_version()); |
| EXPECT_EQ(third_version, registration->active_version()); |
| EXPECT_EQ(ServiceWorkerVersion::ACTIVATED, third_version->status()); |
| } |
| |
| // A fake service worker for toggling whether a fetch event handler exists. |
| class FetchHandlerWorker : public FakeServiceWorker { |
| public: |
| FetchHandlerWorker(EmbeddedWorkerTestHelper* helper) |
| : FakeServiceWorker(helper) {} |
| ~FetchHandlerWorker() override = default; |
| |
| void set_has_fetch_handler(bool has_fetch_handler) { |
| has_fetch_handler_ = has_fetch_handler; |
| } |
| |
| void DispatchInstallEvent( |
| blink::mojom::ServiceWorker::DispatchInstallEventCallback callback) |
| override { |
| std::move(callback).Run(blink::mojom::ServiceWorkerEventStatus::COMPLETED, |
| has_fetch_handler_); |
| } |
| |
| private: |
| bool has_fetch_handler_ = false; |
| }; |
| |
| TEST_F(ServiceWorkerJobTest, HasFetchHandler) { |
| GURL script("https://www.example.com/service_worker.js"); |
| blink::mojom::ServiceWorkerRegistrationOptions options; |
| options.scope = GURL("https://www.example.com/"); |
| scoped_refptr<ServiceWorkerRegistration> registration; |
| |
| auto* fetch_handler_worker = |
| helper_->AddNewPendingServiceWorker<FetchHandlerWorker>(helper_.get()); |
| fetch_handler_worker->set_has_fetch_handler(true); |
| RunRegisterJob(script, options); |
| registration = FindRegistrationForScope(options.scope); |
| EXPECT_EQ(ServiceWorkerVersion::FetchHandlerExistence::EXISTS, |
| registration->active_version()->fetch_handler_existence()); |
| RunUnregisterJob(options.scope); |
| |
| auto* no_fetch_handler_worker = |
| helper_->AddNewPendingServiceWorker<FetchHandlerWorker>(helper_.get()); |
| no_fetch_handler_worker->set_has_fetch_handler(false); |
| RunRegisterJob(script, options); |
| registration = FindRegistrationForScope(options.scope); |
| EXPECT_EQ(ServiceWorkerVersion::FetchHandlerExistence::DOES_NOT_EXIST, |
| registration->active_version()->fetch_handler_existence()); |
| RunUnregisterJob(options.scope); |
| } |
| |
| class CheckPauseAfterDownloadEmbeddedWorkerInstanceClient |
| : public FakeEmbeddedWorkerInstanceClient { |
| public: |
| explicit CheckPauseAfterDownloadEmbeddedWorkerInstanceClient( |
| EmbeddedWorkerTestHelper* helper) |
| : FakeEmbeddedWorkerInstanceClient(helper) {} |
| |
| int num_of_startworker() const { return num_of_startworker_; } |
| void set_next_pause_after_download(bool expectation) { |
| next_pause_after_download_ = expectation; |
| } |
| |
| protected: |
| void StartWorker(blink::mojom::EmbeddedWorkerStartParamsPtr params) override { |
| ASSERT_TRUE(next_pause_after_download_.has_value()); |
| EXPECT_EQ(next_pause_after_download_.value(), params->pause_after_download); |
| num_of_startworker_++; |
| FakeEmbeddedWorkerInstanceClient::StartWorker(std::move(params)); |
| } |
| |
| private: |
| base::Optional<bool> next_pause_after_download_; |
| int num_of_startworker_ = 0; |
| DISALLOW_COPY_AND_ASSIGN(CheckPauseAfterDownloadEmbeddedWorkerInstanceClient); |
| }; |
| |
| TEST_F(ServiceWorkerJobTest, Update_PauseAfterDownload) { |
| UpdateJobTestHelper* update_helper = new UpdateJobTestHelper; |
| helper_.reset(update_helper); |
| |
| std::vector<CheckPauseAfterDownloadEmbeddedWorkerInstanceClient*> clients; |
| clients.push_back( |
| helper_->AddNewPendingInstanceClient< |
| CheckPauseAfterDownloadEmbeddedWorkerInstanceClient>(helper_.get())); |
| clients.push_back( |
| helper_->AddNewPendingInstanceClient< |
| CheckPauseAfterDownloadEmbeddedWorkerInstanceClient>(helper_.get())); |
| |
| // The initial version should not pause after download. |
| clients[0]->set_next_pause_after_download(false); |
| scoped_refptr<ServiceWorkerRegistration> registration = |
| update_helper->SetupInitialRegistration(kNewVersionOrigin); |
| ASSERT_EQ(1, clients[0]->num_of_startworker()); |
| |
| // The updated version should pause after download. |
| clients[1]->set_next_pause_after_download(true); |
| registration->AddListener(update_helper); |
| registration->active_version()->StartUpdate(); |
| base::RunLoop().RunUntilIdle(); |
| ASSERT_EQ(1, clients[1]->num_of_startworker()); |
| } |
| |
| // Test that activation doesn't complete if it's triggered by removing a |
| // controllee and starting the worker failed due to shutdown. |
| TEST_F(ServiceWorkerJobTest, ActivateCancelsOnShutdown) { |
| UpdateJobTestHelper* update_helper = new UpdateJobTestHelper; |
| helper_.reset(update_helper); |
| GURL script("https://www.example.com/service_worker.js"); |
| blink::mojom::ServiceWorkerRegistrationOptions options; |
| options.scope = GURL("https://www.example.com/"); |
| |
| auto* initial_client = |
| helper_->AddNewPendingInstanceClient<FakeEmbeddedWorkerInstanceClient>( |
| helper_.get()); |
| scoped_refptr<ServiceWorkerRegistration> registration = |
| RunRegisterJob(script, options); |
| auto runner = base::MakeRefCounted<base::TestSimpleTaskRunner>(); |
| registration->SetTaskRunnerForTest(runner); |
| |
| // Add a controllee. |
| ServiceWorkerProviderHost* host = CreateControllee(); |
| scoped_refptr<ServiceWorkerVersion> first_version = |
| registration->active_version(); |
| first_version->AddControllee(host); |
| |
| // Update. The new version should be waiting. |
| registration->AddListener(update_helper); |
| first_version->StartUpdate(); |
| base::RunLoop().RunUntilIdle(); |
| scoped_refptr<ServiceWorkerVersion> new_version = |
| registration->waiting_version(); |
| ASSERT_TRUE(new_version); |
| EXPECT_EQ(ServiceWorkerVersion::INSTALLED, new_version->status()); |
| |
| // Stop the worker so that it must start again when activation is attempted. |
| // (This is not strictly necessary to exercise the codepath, but it makes it |
| // easy to cause a failure with set_force_start_worker_failure after |
| // shutdown is simulated. Otherwise our test helper often fails on |
| // DCHECK(context)). |
| new_version->StopWorker(base::DoNothing()); |
| |
| // Remove the controllee. The new version should be activating, and delayed |
| // until the runner runs again. |
| first_version->RemoveControllee(host->client_uuid()); |
| base::RunLoop().RunUntilIdle(); |
| |
| // Activating the new version won't happen until |
| // RequestTermination() is called. |
| EXPECT_EQ(first_version.get(), registration->active_version()); |
| RequestTermination(&initial_client->host()); |
| |
| TestServiceWorkerObserver observer(helper_->context_wrapper()); |
| observer.RunUntilStatusChange(new_version.get(), |
| ServiceWorkerVersion::ACTIVATING); |
| EXPECT_EQ(new_version.get(), registration->active_version()); |
| EXPECT_EQ(ServiceWorkerVersion::ACTIVATING, new_version->status()); |
| |
| // Shutdown. |
| update_helper->context()->wrapper()->Shutdown(); |
| auto* embedded_worker_instance_client = |
| update_helper->AddNewPendingInstanceClient< |
| UpdateJobTestHelper::UpdateJobEmbeddedWorkerInstanceClient>( |
| update_helper); |
| embedded_worker_instance_client->set_force_start_worker_failure(true); |
| |
| // Allow the activation to continue. It will fail, and the worker |
| // should not be promoted to ACTIVATED because failure occur |
| // during shutdown. |
| runner->RunPendingTasks(); |
| base::RunLoop().RunUntilIdle(); |
| EXPECT_EQ(new_version.get(), registration->active_version()); |
| EXPECT_EQ(ServiceWorkerVersion::ACTIVATING, new_version->status()); |
| registration->RemoveListener(update_helper); |
| // Dispatch Mojo messages for those Mojo interfaces bound on |runner| to avoid |
| // possible memory leak. |
| runner->RunUntilIdle(); |
| } |
| |
| // Test that clients are alerted of new registrations if they are |
| // in-scope, so that Clients.claim() or ServiceWorkerContainer.ready work |
| // correctly. |
| TEST_F(ServiceWorkerJobTest, AddRegistrationToMatchingProviderHosts) { |
| GURL scope("https://www.example.com/scope/"); |
| GURL in_scope("https://www.example.com/scope/page"); |
| GURL out_scope("https://www.example.com/page"); |
| |
| // Make an in-scope client. |
| ServiceWorkerProviderHost* client = CreateControllee(); |
| client->UpdateUrls(in_scope, in_scope); |
| |
| // Make an in-scope reserved client. |
| auto provider_info = blink::mojom::ServiceWorkerProviderInfoForWindow::New(); |
| base::WeakPtr<ServiceWorkerProviderHost> reserved_client = |
| ServiceWorkerProviderHost::PreCreateNavigationHost( |
| helper_->context()->AsWeakPtr(), true /* are_ancestors_secure */, |
| {} /* web_contents_getter */, &provider_info); |
| reserved_client->UpdateUrls(in_scope, in_scope); |
| |
| // Make an out-scope client. |
| ServiceWorkerProviderHost* out_scope_client = CreateControllee(); |
| out_scope_client->UpdateUrls(out_scope, out_scope); |
| |
| // Make a new registration. |
| GURL script("https://www.example.com/service_worker.js"); |
| blink::mojom::ServiceWorkerRegistrationOptions options; |
| options.scope = scope; |
| scoped_refptr<ServiceWorkerRegistration> registration = |
| RunRegisterJob(script, options); |
| |
| EXPECT_EQ(registration.get(), client->MatchRegistration()); |
| EXPECT_EQ(registration.get(), reserved_client->MatchRegistration()); |
| EXPECT_NE(registration.get(), out_scope_client->MatchRegistration()); |
| } |
| |
| } // namespace content |