base/at_exit_unittest.cc

   1 // Copyright 2011 The Chromium Authors
   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/functional/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
  15 void IncrementTestCounter1(void* unused) {
  16   ++g_test_counter_1;
  17 }
  18
  19 void IncrementTestCounter2(void* unused) {
  20   ++g_test_counter_2;
  21 }
  22
  23 void ZeroTestCounters() {
  24   g_test_counter_1 = 0;
  25   g_test_counter_2 = 0;
  26 }
  27
  28 void ExpectCounter1IsZero(void* unused) {
  29   EXPECT_EQ(0, g_test_counter_1);
  30 }
  31
  32 void ExpectParamIsNull(void* param) {
  33   EXPECT_EQ(nullptr, param);
  34 }
  35
  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
  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
  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
  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
  82 TEST_F(AtExitTest, Task) {
  83   ZeroTestCounters();
  84   base::AtExitManager::RegisterTask(
  85       base::BindOnce(&ExpectParamIsCounter, &g_test_counter_1));
  86   base::AtExitManager::ProcessCallbacksNow();
  87 }