Google Git
Sign in
chromium / chromium / src / lkgr / . / base / barrier_closure_unittest.cc
blob: 3c21338a5b9fffaec701c180afb9fdc24f38bf64 [file] [log] [blame]
Avi Drissmane4622aa2022-09-08 20:36:06[diff] [blame]1// Copyright 2013 The Chromium Authors
[email protected]285d06fc2013-08-24 15:00:33[diff] [blame]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/barrier_closure.h"
6
Avi Drissman63e1f992023-01-13 18:54:43[diff] [blame]7#include "base/functional/bind.h"
8#include "base/functional/callback.h"
9#include "base/functional/callback_helpers.h"
Keishi Hattori0e45c022021-11-27 09:25:52[diff] [blame]10#include "base/memory/raw_ptr.h"
cfredrice22247e2021-07-22 21:08:57[diff] [blame]11#include "base/test/bind.h"
cfredrice7af0642021-07-27 21:06:51[diff] [blame]12#include "base/test/gtest_util.h"
[email protected]285d06fc2013-08-24 15:00:33[diff] [blame]13#include "testing/gtest/include/gtest/gtest.h"
14
15namespace {
16
[email protected]285d06fc2013-08-24 15:00:33[diff] [blame]17TEST(BarrierClosureTest, RunImmediatelyForZeroClosures) {
18 int count = 0;
cfredrice22247e2021-07-22 21:08:57[diff] [blame]19 base::RepeatingClosure barrier_closure = base::BarrierClosure(
Sorin Jianud01ad7f2024-10-09 03:45:45[diff] [blame]20 0, base::BindLambdaForTesting([&count] { ++count; }));
[email protected]285d06fc2013-08-24 15:00:33[diff] [blame]21 EXPECT_EQ(1, count);
cfredrice7af0642021-07-27 21:06:51[diff] [blame]22}
23
24TEST(BarrierClosureTest, ChecksIfCalledForZeroClosures) {
25 base::RepeatingClosure barrier_closure =
26 base::BarrierClosure(0, base::DoNothing());
cfredrice22247e2021-07-22 21:08:57[diff] [blame]27 EXPECT_FALSE(barrier_closure.is_null());
cfredrice7af0642021-07-27 21:06:51[diff] [blame]28
Peter Boström54119652024-11-14 00:16:38[diff] [blame]29 EXPECT_NOTREACHED_DEATH(barrier_closure.Run());
[email protected]285d06fc2013-08-24 15:00:33[diff] [blame]30}
31
32TEST(BarrierClosureTest, RunAfterNumClosures) {
33 int count = 0;
cfredrice22247e2021-07-22 21:08:57[diff] [blame]34 base::RepeatingClosure barrier_closure = base::BarrierClosure(
Sorin Jianud01ad7f2024-10-09 03:45:45[diff] [blame]35 2, base::BindLambdaForTesting([&count] { ++count; }));
[email protected]285d06fc2013-08-24 15:00:33[diff] [blame]36 EXPECT_EQ(0, count);
37
dzhioev0a3f4592015-05-07 01:05:35[diff] [blame]38 barrier_closure.Run();
[email protected]285d06fc2013-08-24 15:00:33[diff] [blame]39 EXPECT_EQ(0, count);
40
dzhioev0a3f4592015-05-07 01:05:35[diff] [blame]41 barrier_closure.Run();
[email protected]285d06fc2013-08-24 15:00:33[diff] [blame]42 EXPECT_EQ(1, count);
43}
44
dzhioev0a3f4592015-05-07 01:05:35[diff] [blame]45class DestructionIndicator {
46 public:
47 // Sets |*destructed| to true in destructor.
Peter Kasting811504a72025-01-09 03:18:50[diff] [blame]48 explicit DestructionIndicator(bool* destructed) : destructed_(destructed) {
dzhioev0a3f4592015-05-07 01:05:35[diff] [blame]49 *destructed_ = false;
50 }
51
52 ~DestructionIndicator() { *destructed_ = true; }
53
54 void DoNothing() {}
55
56 private:
Keishi Hattori0e45c022021-11-27 09:25:52[diff] [blame]57 raw_ptr<bool> destructed_;
dzhioev0a3f4592015-05-07 01:05:35[diff] [blame]58};
59
60TEST(BarrierClosureTest, ReleasesDoneClosureWhenDone) {
61 bool done_destructed = false;
kylechar01598d72019-05-21 18:35:31[diff] [blame]62 base::RepeatingClosure barrier_closure = base::BarrierClosure(
tzik0c100dc2017-06-26 06:13:17[diff] [blame]63 1,
64 base::BindOnce(&DestructionIndicator::DoNothing,
65 base::Owned(new DestructionIndicator(&done_destructed))));
dzhioev0a3f4592015-05-07 01:05:35[diff] [blame]66 EXPECT_FALSE(done_destructed);
67 barrier_closure.Run();
68 EXPECT_TRUE(done_destructed);
69}
70
dzhioev0a3f4592015-05-07 01:05:35[diff] [blame]71// Tests a case when |done_closure| resets a |barrier_closure|.
kylechar01598d72019-05-21 18:35:31[diff] [blame]72// |barrier_closure| is a RepeatingClosure holding the |done_closure|.
73// |done_closure| holds a pointer back to the |barrier_closure|. When
74// |barrier_closure| is Run() it calls ResetBarrierClosure() which erases the
75// |barrier_closure| while still inside of its Run(). The Run() implementation
76// (in base::BarrierClosure) must not try use itself after executing
77// ResetBarrierClosure() or this test would crash inside Run().
dzhioev0a3f4592015-05-07 01:05:35[diff] [blame]78TEST(BarrierClosureTest, KeepingClosureAliveUntilDone) {
kylechar01598d72019-05-21 18:35:31[diff] [blame]79 base::RepeatingClosure barrier_closure;
cfredrice22247e2021-07-22 21:08:57[diff] [blame]80 barrier_closure =
Sorin Jianud01ad7f2024-10-09 03:45:45[diff] [blame]81 base::BarrierClosure(1, base::BindLambdaForTesting([&barrier_closure] {
cfredrice22247e2021-07-22 21:08:57[diff] [blame]82 barrier_closure = base::RepeatingClosure();
83 }));
dzhioev0a3f4592015-05-07 01:05:35[diff] [blame]84 barrier_closure.Run();
cfredrice22247e2021-07-22 21:08:57[diff] [blame]85 EXPECT_TRUE(barrier_closure.is_null());
dzhioev0a3f4592015-05-07 01:05:35[diff] [blame]86}
87
[email protected]285d06fc2013-08-24 15:00:33[diff] [blame]88} // namespace