content/browser/service_worker/service_worker_registration_unittest.cc

   1 // Copyright 2013 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "content/browser/service_worker/service_worker_registration.h"
   6
   7 #include "base/files/scoped_temp_dir.h"
   8 #include "base/logging.h"
   9 #include "base/message_loop/message_loop.h"
  10 #include "base/run_loop.h"
  11 #include "base/thread_task_runner_handle.h"
  12 #include "content/browser/browser_thread_impl.h"
  13 #include "content/browser/service_worker/service_worker_context_core.h"
  14 #include "content/browser/service_worker/service_worker_registration_handle.h"
  15 #include "testing/gtest/include/gtest/gtest.h"
  16 #include "url/gurl.h"
  17
  18 namespace content {
  19
  20 class ServiceWorkerRegistrationTest : public testing::Test {
  21  public:
  22   ServiceWorkerRegistrationTest()
  23       : io_thread_(BrowserThread::IO, &message_loop_) {}
  24
  25   void SetUp() override {
  26     scoped_ptr<ServiceWorkerDatabaseTaskManager> database_task_manager(
  27         new MockServiceWorkerDatabaseTaskManager(
  28             base::ThreadTaskRunnerHandle::Get()));
  29     context_.reset(
  30         new ServiceWorkerContextCore(base::FilePath(),
  31                                      base::ThreadTaskRunnerHandle::Get(),
  32                                      database_task_manager.Pass(),
  33                                      base::ThreadTaskRunnerHandle::Get(),
  34                                      NULL,
  35                                      NULL,
  36                                      NULL,
  37                                      NULL));
  38     context_ptr_ = context_->AsWeakPtr();
  39   }
  40
  41   void TearDown() override {
  42     context_.reset();
  43     base::RunLoop().RunUntilIdle();
  44   }
  45
  46   class RegistrationListener : public ServiceWorkerRegistration::Listener {
  47    public:
  48     RegistrationListener() {}
  49     ~RegistrationListener() {
  50       if (observed_registration_.get())
  51         observed_registration_->RemoveListener(this);
  52     }
  53
  54     void OnVersionAttributesChanged(
  55         ServiceWorkerRegistration* registration,
  56         ChangedVersionAttributesMask changed_mask,
  57         const ServiceWorkerRegistrationInfo& info) override {
  58       observed_registration_ = registration;
  59       observed_changed_mask_ = changed_mask;
  60       observed_info_ = info;
  61     }
  62
  63     void OnRegistrationFailed(
  64         ServiceWorkerRegistration* registration) override {
  65       NOTREACHED();
  66     }
  67
  68     void OnRegistrationFinishedUninstalling(
  69         ServiceWorkerRegistration* registration) override {
  70       NOTREACHED();
  71     }
  72
  73     void OnUpdateFound(ServiceWorkerRegistration* registration) override {
  74       NOTREACHED();
  75     }
  76
  77     void Reset() {
  78       observed_registration_ = NULL;
  79       observed_changed_mask_ = ChangedVersionAttributesMask();
  80       observed_info_ = ServiceWorkerRegistrationInfo();
  81     }
  82
  83     scoped_refptr<ServiceWorkerRegistration> observed_registration_;
  84     ChangedVersionAttributesMask observed_changed_mask_;
  85     ServiceWorkerRegistrationInfo observed_info_;
  86   };
  87
  88  protected:
  89   scoped_ptr<ServiceWorkerContextCore> context_;
  90   base::WeakPtr<ServiceWorkerContextCore> context_ptr_;
  91   base::MessageLoopForIO message_loop_;
  92   BrowserThreadImpl io_thread_;
  93 };
  94
  95 TEST_F(ServiceWorkerRegistrationTest, SetAndUnsetVersions) {
  96   const GURL kScope("http://www.example.not/");
  97   const GURL kScript("http://www.example.not/service_worker.js");
  98   int64 kRegistrationId = 1L;
  99   scoped_refptr<ServiceWorkerRegistration> registration =
 100       new ServiceWorkerRegistration(
 101           kScope,
 102           kRegistrationId,
 103           context_ptr_);
 104
 105   const int64 version_1_id = 1L;
 106   const int64 version_2_id = 2L;
 107   scoped_refptr<ServiceWorkerVersion> version_1 = new ServiceWorkerVersion(
 108       registration.get(), kScript, version_1_id, context_ptr_);
 109   scoped_refptr<ServiceWorkerVersion> version_2 = new ServiceWorkerVersion(
 110       registration.get(), kScript, version_2_id, context_ptr_);
 111
 112   RegistrationListener listener;
 113   registration->AddListener(&listener);
 114   registration->SetActiveVersion(version_1.get());
 115
 116   EXPECT_EQ(version_1.get(), registration->active_version());
 117   EXPECT_EQ(registration, listener.observed_registration_);
 118   EXPECT_EQ(ChangedVersionAttributesMask::ACTIVE_VERSION,
 119             listener.observed_changed_mask_.changed());
 120   EXPECT_EQ(kScope, listener.observed_info_.pattern);
 121   EXPECT_EQ(version_1_id, listener.observed_info_.active_version.version_id);
 122   EXPECT_EQ(kScript, listener.observed_info_.active_version.script_url);
 123   EXPECT_EQ(listener.observed_info_.installing_version.version_id,
 124             kInvalidServiceWorkerVersionId);
 125   EXPECT_EQ(listener.observed_info_.waiting_version.version_id,
 126             kInvalidServiceWorkerVersionId);
 127   listener.Reset();
 128
 129   registration->SetInstallingVersion(version_2.get());
 130
 131   EXPECT_EQ(version_2.get(), registration->installing_version());
 132   EXPECT_EQ(ChangedVersionAttributesMask::INSTALLING_VERSION,
 133             listener.observed_changed_mask_.changed());
 134   EXPECT_EQ(version_1_id, listener.observed_info_.active_version.version_id);
 135   EXPECT_EQ(version_2_id,
 136             listener.observed_info_.installing_version.version_id);
 137   EXPECT_EQ(listener.observed_info_.waiting_version.version_id,
 138             kInvalidServiceWorkerVersionId);
 139   listener.Reset();
 140
 141   registration->SetWaitingVersion(version_2.get());
 142
 143   EXPECT_EQ(version_2.get(), registration->waiting_version());
 144   EXPECT_FALSE(registration->installing_version());
 145   EXPECT_TRUE(listener.observed_changed_mask_.waiting_changed());
 146   EXPECT_TRUE(listener.observed_changed_mask_.installing_changed());
 147   EXPECT_EQ(version_1_id, listener.observed_info_.active_version.version_id);
 148   EXPECT_EQ(version_2_id, listener.observed_info_.waiting_version.version_id);
 149   EXPECT_EQ(listener.observed_info_.installing_version.version_id,
 150             kInvalidServiceWorkerVersionId);
 151   listener.Reset();
 152
 153   registration->UnsetVersion(version_2.get());
 154
 155   EXPECT_FALSE(registration->waiting_version());
 156   EXPECT_EQ(ChangedVersionAttributesMask::WAITING_VERSION,
 157             listener.observed_changed_mask_.changed());
 158   EXPECT_EQ(version_1_id, listener.observed_info_.active_version.version_id);
 159   EXPECT_EQ(listener.observed_info_.waiting_version.version_id,
 160             kInvalidServiceWorkerVersionId);
 161   EXPECT_EQ(listener.observed_info_.installing_version.version_id,
 162             kInvalidServiceWorkerVersionId);
 163 }
 164
 165 TEST_F(ServiceWorkerRegistrationTest, FailedRegistrationNoCrash) {
 166   const GURL kScope("http://www.example.not/");
 167   int64 kRegistrationId = 1L;
 168   scoped_refptr<ServiceWorkerRegistration> registration =
 169       new ServiceWorkerRegistration(
 170           kScope,
 171           kRegistrationId,
 172           context_ptr_);
 173   scoped_ptr<ServiceWorkerRegistrationHandle> handle(
 174       new ServiceWorkerRegistrationHandle(
 175           context_ptr_,
 176           base::WeakPtr<ServiceWorkerProviderHost>(),
 177           registration.get()));
 178   registration->NotifyRegistrationFailed();
 179   // Don't crash when handle gets destructed.
 180 }
 181
 182 }  // namespace content