base/tracked_objects_unittest.cc

   1 // Copyright (c) 2006-2008 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 // Test of classes in the tracked_objects.h classes.
   6
   7 #include "base/tracked_objects.h"
   8
   9 #include "base/message_loop.h"
  10 #include "testing/gtest/include/gtest/gtest.h"
  11
  12 namespace tracked_objects {
  13
  14 class TrackedObjectsTest : public testing::Test {
  15  public:
  16   MessageLoop message_loop_;
  17 };
  18
  19 TEST_F(TrackedObjectsTest, MinimalStartupShutdown) {
  20   // Minimal test doesn't even create any tasks.
  21   if (!ThreadData::StartTracking(true))
  22     return;
  23
  24   EXPECT_FALSE(ThreadData::first());  // No activity even on this thread.
  25   ThreadData* data = ThreadData::current();
  26   EXPECT_TRUE(ThreadData::first());  // Now class was constructed.
  27   EXPECT_TRUE(data);
  28   EXPECT_TRUE(!data->next());
  29   EXPECT_EQ(data, ThreadData::current());
  30   ThreadData::BirthMap birth_map;
  31   data->SnapshotBirthMap(&birth_map);
  32   EXPECT_EQ(0u, birth_map.size());
  33   ThreadData::DeathMap death_map;
  34   data->SnapshotDeathMap(&death_map);
  35   EXPECT_EQ(0u, death_map.size());
  36   ThreadData::ShutdownSingleThreadedCleanup();
  37
  38   // Do it again, just to be sure we reset state completely.
  39   ThreadData::StartTracking(true);
  40   EXPECT_FALSE(ThreadData::first());  // No activity even on this thread.
  41   data = ThreadData::current();
  42   EXPECT_TRUE(ThreadData::first());  // Now class was constructed.
  43   EXPECT_TRUE(data);
  44   EXPECT_TRUE(!data->next());
  45   EXPECT_EQ(data, ThreadData::current());
  46   birth_map.clear();
  47   data->SnapshotBirthMap(&birth_map);
  48   EXPECT_EQ(0u, birth_map.size());
  49   death_map.clear();
  50   data->SnapshotDeathMap(&death_map);
  51   EXPECT_EQ(0u, death_map.size());
  52   ThreadData::ShutdownSingleThreadedCleanup();
  53 }
  54
  55 class NoopTracked : public tracked_objects::Tracked {
  56 };
  57
  58 TEST_F(TrackedObjectsTest, TinyStartupShutdown) {
  59   if (!ThreadData::StartTracking(true))
  60     return;
  61
  62   // Instigate tracking on a single tracked object, or our thread.
  63   NoopTracked tracked;
  64
  65   const ThreadData* data = ThreadData::first();
  66   EXPECT_TRUE(data);
  67   EXPECT_TRUE(!data->next());
  68   EXPECT_EQ(data, ThreadData::current());
  69   ThreadData::BirthMap birth_map;
  70   data->SnapshotBirthMap(&birth_map);
  71   EXPECT_EQ(1u, birth_map.size());                         // 1 birth location.
  72   EXPECT_EQ(1, birth_map.begin()->second->birth_count());  // 1 birth.
  73   ThreadData::DeathMap death_map;
  74   data->SnapshotDeathMap(&death_map);
  75   EXPECT_EQ(0u, death_map.size());                         // No deaths.
  76
  77
  78   // Now instigate a birth, and a death.
  79   delete new NoopTracked;
  80
  81   birth_map.clear();
  82   data->SnapshotBirthMap(&birth_map);
  83   EXPECT_EQ(1u, birth_map.size());                         // 1 birth location.
  84   EXPECT_EQ(2, birth_map.begin()->second->birth_count());  // 2 births.
  85   death_map.clear();
  86   data->SnapshotDeathMap(&death_map);
  87   EXPECT_EQ(1u, death_map.size());                         // 1 location.
  88   EXPECT_EQ(1, death_map.begin()->second.count());         // 1 death.
  89
  90   // The births were at the same location as the one known death.
  91   EXPECT_EQ(birth_map.begin()->second, death_map.begin()->first);
  92
  93   ThreadData::ShutdownSingleThreadedCleanup();
  94 }
  95
  96 }  // namespace tracked_objects