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
   8 #include "base/files/scoped_temp_dir.h"
   9 #include "base/logging.h"
  10 #include "base/message_loop/message_loop.h"
  11 #include "content/browser/browser_thread_impl.h"
  12 #include "content/browser/service_worker/service_worker_context_core.h"
  13 #include "testing/gtest/include/gtest/gtest.h"
  14 #include "url/gurl.h"
  15
  16 namespace content {
  17
  18 class ServiceWorkerRegistrationTest : public testing::Test {
  19  public:
  20   ServiceWorkerRegistrationTest()
  21       : io_thread_(BrowserThread::IO, &message_loop_) {}
  22
  23   virtual void SetUp() OVERRIDE {
  24     context_.reset(new ServiceWorkerContextCore(
  25         base::FilePath(),
  26         base::MessageLoopProxy::current(),
  27         NULL,
  28         NULL,
  29         scoped_ptr<ServiceWorkerProcessManager>()));
  30     context_ptr_ = context_->AsWeakPtr();
  31   }
  32
  33  protected:
  34   scoped_ptr<ServiceWorkerContextCore> context_;
  35   base::WeakPtr<ServiceWorkerContextCore> context_ptr_;
  36   base::MessageLoopForIO message_loop_;
  37   BrowserThreadImpl io_thread_;
  38 };
  39
  40 // Make sure that activation does not leak
  41 TEST_F(ServiceWorkerRegistrationTest, ActivatePending) {
  42   int64 registration_id = -1L;
  43   scoped_refptr<ServiceWorkerRegistration> registration =
  44       new ServiceWorkerRegistration(
  45           GURL("http://www.example.com/*"),
  46           GURL("http://www.example.com/service_worker.js"),
  47           registration_id,
  48           context_ptr_);
  49
  50   const int64 version_1_id = 1L;
  51   const int64 version_2_id = 2L;
  52   scoped_refptr<ServiceWorkerVersion> version_1 =
  53       new ServiceWorkerVersion(registration, version_1_id, context_ptr_);
  54   version_1->SetStatus(ServiceWorkerVersion::ACTIVE);
  55   registration->set_active_version(version_1);
  56
  57   scoped_refptr<ServiceWorkerVersion> version_2 =
  58       new ServiceWorkerVersion(registration, version_2_id, context_ptr_);
  59   registration->set_pending_version(version_2);
  60
  61   registration->ActivatePendingVersion();
  62   DCHECK_EQ(version_2, registration->active_version());
  63   DCHECK(version_1->HasOneRef());
  64   version_1 = NULL;
  65
  66   DCHECK(!version_2->HasOneRef());
  67 }
  68
  69 }  // namespace content