components/gcm_driver/gcm_delayed_task_controller_unittest.cc

// 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 "components/gcm_driver/gcm_delayed_task_controller.h"

#include <memory>

#include "base/bind.h"
#include "testing/gtest/include/gtest/gtest.h"

namespace gcm {

class GCMDelayedTaskControllerTest : public testing::Test {
 public:
  GCMDelayedTaskControllerTest();
  ~GCMDelayedTaskControllerTest() override;

  void TestTask();

  GCMDelayedTaskController* controller() { return controller_.get(); }

  int number_of_triggered_tasks() const { return number_of_triggered_tasks_; }

 private:
  std::unique_ptr<GCMDelayedTaskController> controller_;
  int number_of_triggered_tasks_;
};

GCMDelayedTaskControllerTest::GCMDelayedTaskControllerTest()
    : controller_(new GCMDelayedTaskController), number_of_triggered_tasks_(0) {
}

GCMDelayedTaskControllerTest::~GCMDelayedTaskControllerTest() {
}

void GCMDelayedTaskControllerTest::TestTask() {
  ++number_of_triggered_tasks_;
}

// Tests that a newly created controller forced tasks to be delayed, while
// calling SetReady allows tasks to execute.
TEST_F(GCMDelayedTaskControllerTest, SetReadyWithNoTasks) {
  EXPECT_FALSE(controller()->CanRunTaskWithoutDelay());
  EXPECT_EQ(0, number_of_triggered_tasks());

  controller()->SetReady();
  EXPECT_TRUE(controller()->CanRunTaskWithoutDelay());
  EXPECT_EQ(0, number_of_triggered_tasks());
}

// Tests that tasks are triggered when controlles is set to ready.
TEST_F(GCMDelayedTaskControllerTest, PendingTasksTriggeredWhenSetReady) {
  controller()->AddTask(base::Bind(&GCMDelayedTaskControllerTest::TestTask,
                                   base::Unretained(this)));
  controller()->AddTask(base::Bind(&GCMDelayedTaskControllerTest::TestTask,
                                   base::Unretained(this)));

  controller()->SetReady();
  EXPECT_EQ(2, number_of_triggered_tasks());
}

}  // namespace gcm