base/task_unittest.cc

   1 // Copyright (c) 2011 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 "base/memory/ref_counted.h"
   6 #include "base/task.h"
   7 #include "testing/gtest/include/gtest/gtest.h"
   8
   9 namespace {
  10
  11 class CancelInDestructor : public base::RefCounted<CancelInDestructor> {
  12  public:
  13   CancelInDestructor() : cancelable_task_(NULL) {}
  14
  15   void Start() {
  16     if (cancelable_task_) {
  17       ADD_FAILURE();
  18       return;
  19     }
  20     AddRef();
  21     cancelable_task_ = NewRunnableMethod(
  22         this, &CancelInDestructor::NeverIssuedCallback);
  23     Release();
  24   }
  25
  26   CancelableTask* cancelable_task() {
  27     return cancelable_task_;
  28   }
  29
  30  private:
  31   friend class base::RefCounted<CancelInDestructor>;
  32
  33   ~CancelInDestructor() {
  34     if (cancelable_task_)
  35       cancelable_task_->Cancel();
  36   }
  37
  38   void NeverIssuedCallback() { NOTREACHED(); }
  39
  40   CancelableTask* cancelable_task_;
  41 };
  42
  43 TEST(TaskTest, TestCancelInDestructor) {
  44   // Intentionally not using a scoped_refptr for cancel_in_destructor.
  45   CancelInDestructor* cancel_in_destructor = new CancelInDestructor();
  46   cancel_in_destructor->Start();
  47   CancelableTask* cancelable_task = cancel_in_destructor->cancelable_task();
  48   ASSERT_TRUE(cancelable_task);
  49   delete cancelable_task;
  50 }
  51
  52 class DoneTask : public Task {
  53  public:
  54   DoneTask(int* run_count, bool* was_deleted)
  55       : run_count_(run_count), was_deleted_(was_deleted) {
  56   }
  57   virtual ~DoneTask() {
  58     *was_deleted_ = true;
  59   }
  60
  61   virtual void Run() {
  62     ++(*run_count_);
  63   }
  64
  65   int* run_count_;
  66   bool* was_deleted_;
  67 };
  68
  69 TEST(TaskTest, TestScopedTaskRunnerExitScope) {
  70   int run_count = 0;
  71   bool was_deleted = false;
  72   {
  73     base::ScopedTaskRunner runner(new DoneTask(&run_count, &was_deleted));
  74     EXPECT_EQ(0, run_count);
  75   }
  76   EXPECT_EQ(1, run_count);
  77   EXPECT_TRUE(was_deleted);
  78 }
  79
  80 TEST(TaskTest, TestScopedTaskRunnerRelease) {
  81   int run_count = 0;
  82   bool was_deleted = false;
  83   {
  84     base::ScopedTaskRunner runner(new DoneTask(&run_count, &was_deleted));
  85     delete runner.Release();
  86     EXPECT_TRUE(was_deleted);
  87   }
  88   EXPECT_EQ(0, run_count);
  89 }
  90
  91 TEST(TaskTest, TestScopedTaskRunnerManualRun) {
  92   int run_count = 0;
  93   Task* done_task = NULL;
  94   bool was_deleted = false;
  95   {
  96     base::ScopedTaskRunner runner(new DoneTask(&run_count, &was_deleted));
  97     done_task = runner.Release();
  98     EXPECT_TRUE(NULL != done_task);
  99     EXPECT_FALSE(was_deleted);
 100     EXPECT_EQ(0, run_count);
 101     done_task->Run();
 102     EXPECT_FALSE(was_deleted);
 103     EXPECT_EQ(1, run_count);
 104   }
 105   EXPECT_EQ(1, run_count);
 106   delete done_task;
 107   EXPECT_TRUE(was_deleted);
 108 }
 109
 110 }  // namespace