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/at_exit.h"
6 #include "base/bind.h"
7
8 #include "testing/gtest/include/gtest/gtest.h"
9
10 namespace {
11
12 int g_test_counter_1 = 0;
13 int g_test_counter_2 = 0;
14
IncrementTestCounter1(void * unused)15 void IncrementTestCounter1(void* unused) {
16 ++g_test_counter_1;
17 }
18
IncrementTestCounter2(void * unused)19 void IncrementTestCounter2(void* unused) {
20 ++g_test_counter_2;
21 }
22
ZeroTestCounters()23 void ZeroTestCounters() {
24 g_test_counter_1 = 0;
25 g_test_counter_2 = 0;
26 }
27
ExpectCounter1IsZero(void * unused)28 void ExpectCounter1IsZero(void* unused) {
29 EXPECT_EQ(0, g_test_counter_1);
30 }
31
ExpectParamIsNull(void * param)32 void ExpectParamIsNull(void* param) {
33 EXPECT_EQ(nullptr, param);
34 }
35
ExpectParamIsCounter(void * param)36 void ExpectParamIsCounter(void* param) {
37 EXPECT_EQ(&g_test_counter_1, param);
38 }
39
40 } // namespace
41
42 class AtExitTest : public testing::Test {
43 private:
44 // Don't test the global AtExitManager, because asking it to process its
45 // AtExit callbacks can ruin the global state that other tests may depend on.
46 base::ShadowingAtExitManager exit_manager_;
47 };
48
TEST_F(AtExitTest,Basic)49 TEST_F(AtExitTest, Basic) {
50 ZeroTestCounters();
51 base::AtExitManager::RegisterCallback(&IncrementTestCounter1, nullptr);
52 base::AtExitManager::RegisterCallback(&IncrementTestCounter2, nullptr);
53 base::AtExitManager::RegisterCallback(&IncrementTestCounter1, nullptr);
54
55 EXPECT_EQ(0, g_test_counter_1);
56 EXPECT_EQ(0, g_test_counter_2);
57 base::AtExitManager::ProcessCallbacksNow();
58 EXPECT_EQ(2, g_test_counter_1);
59 EXPECT_EQ(1, g_test_counter_2);
60 }
61
TEST_F(AtExitTest,LIFOOrder)62 TEST_F(AtExitTest, LIFOOrder) {
63 ZeroTestCounters();
64 base::AtExitManager::RegisterCallback(&IncrementTestCounter1, nullptr);
65 base::AtExitManager::RegisterCallback(&ExpectCounter1IsZero, nullptr);
66 base::AtExitManager::RegisterCallback(&IncrementTestCounter2, nullptr);
67
68 EXPECT_EQ(0, g_test_counter_1);
69 EXPECT_EQ(0, g_test_counter_2);
70 base::AtExitManager::ProcessCallbacksNow();
71 EXPECT_EQ(1, g_test_counter_1);
72 EXPECT_EQ(1, g_test_counter_2);
73 }
74
TEST_F(AtExitTest,Param)75 TEST_F(AtExitTest, Param) {
76 base::AtExitManager::RegisterCallback(&ExpectParamIsNull, nullptr);
77 base::AtExitManager::RegisterCallback(&ExpectParamIsCounter,
78 &g_test_counter_1);
79 base::AtExitManager::ProcessCallbacksNow();
80 }
81
TEST_F(AtExitTest,Task)82 TEST_F(AtExitTest, Task) {
83 ZeroTestCounters();
84 base::AtExitManager::RegisterTask(
85 base::BindOnce(&ExpectParamIsCounter, &g_test_counter_1));
86 base::AtExitManager::ProcessCallbacksNow();
87 }
88