Home | History | Annotate | Download | only in base 
      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 
     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(static_cast<void*>(NULL), 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, NULL);
     52   base::AtExitManager::RegisterCallback(&IncrementTestCounter2, NULL);
     53   base::AtExitManager::RegisterCallback(&IncrementTestCounter1, NULL);
     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, NULL);
     65   base::AtExitManager::RegisterCallback(&ExpectCounter1IsZero, NULL);
     66   base::AtExitManager::RegisterCallback(&IncrementTestCounter2, NULL);
     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, NULL);
     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(base::Bind(&ExpectParamIsCounter,
     85                                                &g_test_counter_1));
     86   base::AtExitManager::ProcessCallbacksNow();
     87 }
     88