content/browser/service_worker/service_worker_cache_scheduler_unittest.cc

   1 // Copyright 2015 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_cache_scheduler.h"
   6
   7 #include "base/bind.h"
   8 #include "base/callback.h"
   9 #include "base/run_loop.h"
  10 #include "testing/gtest/include/gtest/gtest.h"
  11
  12 namespace content {
  13
  14 namespace {
  15
  16 class JobStats {
  17  public:
  18   JobStats(ServiceWorkerCacheScheduler* scheduler)
  19       : scheduler_(scheduler), callback_count_(0) {}
  20
  21   virtual void Run() = 0;
  22
  23   int callback_count() const { return callback_count_; }
  24
  25  protected:
  26   ServiceWorkerCacheScheduler* scheduler_;
  27   int callback_count_;
  28 };
  29
  30 class SyncJob : public JobStats {
  31  public:
  32   SyncJob(ServiceWorkerCacheScheduler* scheduler) : JobStats(scheduler) {}
  33
  34   void Run() override {
  35     callback_count_++;
  36     scheduler_->CompleteOperationAndRunNext();
  37   }
  38 };
  39
  40 class AsyncJob : public JobStats {
  41  public:
  42   AsyncJob(ServiceWorkerCacheScheduler* scheduler) : JobStats(scheduler) {}
  43
  44   void Run() override { callback_count_++; }
  45   void Done() { scheduler_->CompleteOperationAndRunNext(); }
  46 };
  47
  48 }  // namespace
  49
  50 class ServiceWorkerCacheSchedulerTest : public testing::Test {
  51  protected:
  52   ServiceWorkerCacheSchedulerTest()
  53       : async_job_(AsyncJob(&scheduler_)), sync_job_(SyncJob(&scheduler_)) {}
  54
  55   ServiceWorkerCacheScheduler scheduler_;
  56   AsyncJob async_job_;
  57   SyncJob sync_job_;
  58 };
  59
  60 TEST_F(ServiceWorkerCacheSchedulerTest, ScheduleOne) {
  61   scheduler_.ScheduleOperation(
  62       base::Bind(&JobStats::Run, base::Unretained(&sync_job_)));
  63   EXPECT_EQ(1, sync_job_.callback_count());
  64 }
  65
  66 TEST_F(ServiceWorkerCacheSchedulerTest, ScheduleTwo) {
  67   scheduler_.ScheduleOperation(
  68       base::Bind(&JobStats::Run, base::Unretained(&sync_job_)));
  69   scheduler_.ScheduleOperation(
  70       base::Bind(&JobStats::Run, base::Unretained(&sync_job_)));
  71   EXPECT_EQ(2, sync_job_.callback_count());
  72 }
  73
  74 TEST_F(ServiceWorkerCacheSchedulerTest, Block) {
  75   scheduler_.ScheduleOperation(
  76       base::Bind(&JobStats::Run, base::Unretained(&async_job_)));
  77   EXPECT_EQ(1, async_job_.callback_count());
  78   scheduler_.ScheduleOperation(
  79       base::Bind(&JobStats::Run, base::Unretained(&sync_job_)));
  80   EXPECT_EQ(0, sync_job_.callback_count());
  81   async_job_.Done();
  82   EXPECT_EQ(1, sync_job_.callback_count());
  83 }
  84
  85 }  // namespace content