barrier_closure_unittest.cc 925 Bytes
Newer Older
gejun's avatar
gejun committed
1 2 3 4
// Copyright 2013 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.

5
#include "butil/barrier_closure.h"
gejun's avatar
gejun committed
6

7
#include "butil/bind.h"
gejun's avatar
gejun committed
8 9 10 11 12 13 14 15
#include <gtest/gtest.h>

namespace {

void Increment(int* count) { (*count)++; }

TEST(BarrierClosureTest, RunImmediatelyForZeroClosures) {
  int count = 0;
16
  butil::Closure doneClosure(butil::Bind(&Increment, butil::Unretained(&count)));
gejun's avatar
gejun committed
17

18
  butil::Closure barrierClosure = butil::BarrierClosure(0, doneClosure);
gejun's avatar
gejun committed
19 20 21 22 23
  EXPECT_EQ(1, count);
}

TEST(BarrierClosureTest, RunAfterNumClosures) {
  int count = 0;
24
  butil::Closure doneClosure(butil::Bind(&Increment, butil::Unretained(&count)));
gejun's avatar
gejun committed
25

26
  butil::Closure barrierClosure = butil::BarrierClosure(2, doneClosure);
gejun's avatar
gejun committed
27 28 29 30 31 32 33 34 35 36
  EXPECT_EQ(0, count);

  barrierClosure.Run();
  EXPECT_EQ(0, count);

  barrierClosure.Run();
  EXPECT_EQ(1, count);
}

}  // namespace