| OLD | NEW |
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "base/at_exit.h" | 5 #include "base/at_exit.h" |
| 6 #include "base/memory/singleton.h" | 6 #include "base/memory/singleton.h" |
| 7 #include "testing/gtest/include/gtest/gtest.h" | 7 #include "testing/gtest/include/gtest/gtest.h" |
| 8 | 8 |
| 9 namespace base { |
| 9 namespace { | 10 namespace { |
| 10 | 11 |
| 11 COMPILE_ASSERT(DefaultSingletonTraits<int>::kRegisterAtExit == true, a); | 12 COMPILE_ASSERT(DefaultSingletonTraits<int>::kRegisterAtExit == true, a); |
| 12 | 13 |
| 13 typedef void (*CallbackFunc)(); | 14 typedef void (*CallbackFunc)(); |
| 14 | 15 |
| 15 class IntSingleton { | 16 class IntSingleton { |
| 16 public: | 17 public: |
| 17 static IntSingleton* GetInstance() { | 18 static IntSingleton* GetInstance() { |
| 18 return Singleton<IntSingleton>::get(); | 19 return Singleton<IntSingleton>::get(); |
| (...skipping 89 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 108 } | 109 } |
| 109 }; | 110 }; |
| 110 | 111 |
| 111 template <class Type> | 112 template <class Type> |
| 112 class AlignedTestSingleton { | 113 class AlignedTestSingleton { |
| 113 public: | 114 public: |
| 114 AlignedTestSingleton() {} | 115 AlignedTestSingleton() {} |
| 115 ~AlignedTestSingleton() {} | 116 ~AlignedTestSingleton() {} |
| 116 static AlignedTestSingleton* GetInstance() { | 117 static AlignedTestSingleton* GetInstance() { |
| 117 return Singleton<AlignedTestSingleton, | 118 return Singleton<AlignedTestSingleton, |
| 118 StaticMemorySingletonTraits<AlignedTestSingleton> >::get(); | 119 StaticMemorySingletonTraits<AlignedTestSingleton>>::get(); |
| 119 } | 120 } |
| 120 | 121 |
| 121 Type type_; | 122 Type type_; |
| 122 }; | 123 }; |
| 123 | 124 |
| 124 | 125 |
| 125 void SingletonNoLeak(CallbackFunc CallOnQuit) { | 126 void SingletonNoLeak(CallbackFunc CallOnQuit) { |
| 126 CallbackSingletonWithNoLeakTrait::GetInstance()->callback_ = CallOnQuit; | 127 CallbackSingletonWithNoLeakTrait::GetInstance()->callback_ = CallOnQuit; |
| 127 } | 128 } |
| 128 | 129 |
| (...skipping 11 matching lines...) Expand all Loading... |
| 140 } | 141 } |
| 141 | 142 |
| 142 void SingletonStatic(CallbackFunc CallOnQuit) { | 143 void SingletonStatic(CallbackFunc CallOnQuit) { |
| 143 CallbackSingletonWithStaticTrait::GetInstance()->callback_ = CallOnQuit; | 144 CallbackSingletonWithStaticTrait::GetInstance()->callback_ = CallOnQuit; |
| 144 } | 145 } |
| 145 | 146 |
| 146 CallbackFunc* GetStaticSingleton() { | 147 CallbackFunc* GetStaticSingleton() { |
| 147 return &CallbackSingletonWithStaticTrait::GetInstance()->callback_; | 148 return &CallbackSingletonWithStaticTrait::GetInstance()->callback_; |
| 148 } | 149 } |
| 149 | 150 |
| 150 } // namespace | |
| 151 | 151 |
| 152 class SingletonTest : public testing::Test { | 152 class SingletonTest : public testing::Test { |
| 153 public: | 153 public: |
| 154 SingletonTest() {} | 154 SingletonTest() {} |
| 155 | 155 |
| 156 void SetUp() override { | 156 void SetUp() override { |
| 157 non_leak_called_ = false; | 157 non_leak_called_ = false; |
| 158 leaky_called_ = false; | 158 leaky_called_ = false; |
| 159 static_called_ = false; | 159 static_called_ = false; |
| 160 } | 160 } |
| (...skipping 39 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 200 bool SingletonTest::leaky_called_ = false; | 200 bool SingletonTest::leaky_called_ = false; |
| 201 bool SingletonTest::static_called_ = false; | 201 bool SingletonTest::static_called_ = false; |
| 202 | 202 |
| 203 TEST_F(SingletonTest, Basic) { | 203 TEST_F(SingletonTest, Basic) { |
| 204 int* singleton_int; | 204 int* singleton_int; |
| 205 int* singleton_int_5; | 205 int* singleton_int_5; |
| 206 CallbackFunc* leaky_singleton; | 206 CallbackFunc* leaky_singleton; |
| 207 CallbackFunc* static_singleton; | 207 CallbackFunc* static_singleton; |
| 208 | 208 |
| 209 { | 209 { |
| 210 base::ShadowingAtExitManager sem; | 210 ShadowingAtExitManager sem; |
| 211 { | 211 { |
| 212 singleton_int = SingletonInt(); | 212 singleton_int = SingletonInt(); |
| 213 } | 213 } |
| 214 // Ensure POD type initialization. | 214 // Ensure POD type initialization. |
| 215 EXPECT_EQ(*singleton_int, 0); | 215 EXPECT_EQ(*singleton_int, 0); |
| 216 *singleton_int = 1; | 216 *singleton_int = 1; |
| 217 | 217 |
| 218 EXPECT_EQ(singleton_int, SingletonInt()); | 218 EXPECT_EQ(singleton_int, SingletonInt()); |
| 219 EXPECT_EQ(*singleton_int, 1); | 219 EXPECT_EQ(*singleton_int, 1); |
| 220 | 220 |
| (...skipping 13 matching lines...) Expand all Loading... |
| 234 | 234 |
| 235 // Verify that only the expected callback has been called. | 235 // Verify that only the expected callback has been called. |
| 236 VerifiesCallbacks(); | 236 VerifiesCallbacks(); |
| 237 // Delete the leaky singleton. | 237 // Delete the leaky singleton. |
| 238 DeleteLeakySingleton(); | 238 DeleteLeakySingleton(); |
| 239 | 239 |
| 240 // The static singleton can't be acquired post-atexit. | 240 // The static singleton can't be acquired post-atexit. |
| 241 EXPECT_EQ(NULL, GetStaticSingleton()); | 241 EXPECT_EQ(NULL, GetStaticSingleton()); |
| 242 | 242 |
| 243 { | 243 { |
| 244 base::ShadowingAtExitManager sem; | 244 ShadowingAtExitManager sem; |
| 245 // Verifiy that the variables were reset. | 245 // Verifiy that the variables were reset. |
| 246 { | 246 { |
| 247 singleton_int = SingletonInt(); | 247 singleton_int = SingletonInt(); |
| 248 EXPECT_EQ(*singleton_int, 0); | 248 EXPECT_EQ(*singleton_int, 0); |
| 249 } | 249 } |
| 250 { | 250 { |
| 251 singleton_int_5 = SingletonInt5(); | 251 singleton_int_5 = SingletonInt5(); |
| 252 EXPECT_EQ(*singleton_int_5, 5); | 252 EXPECT_EQ(*singleton_int_5, 5); |
| 253 } | 253 } |
| 254 { | 254 { |
| (...skipping 23 matching lines...) Expand all Loading... |
| 278 AlignedTestSingleton<AlignedMemory<128, 128> >* align128 = | 278 AlignedTestSingleton<AlignedMemory<128, 128> >* align128 = |
| 279 AlignedTestSingleton<AlignedMemory<128, 128> >::GetInstance(); | 279 AlignedTestSingleton<AlignedMemory<128, 128> >::GetInstance(); |
| 280 AlignedTestSingleton<AlignedMemory<4096, 4096> >* align4096 = | 280 AlignedTestSingleton<AlignedMemory<4096, 4096> >* align4096 = |
| 281 AlignedTestSingleton<AlignedMemory<4096, 4096> >::GetInstance(); | 281 AlignedTestSingleton<AlignedMemory<4096, 4096> >::GetInstance(); |
| 282 | 282 |
| 283 EXPECT_ALIGNED(align4, 4); | 283 EXPECT_ALIGNED(align4, 4); |
| 284 EXPECT_ALIGNED(align32, 32); | 284 EXPECT_ALIGNED(align32, 32); |
| 285 EXPECT_ALIGNED(align128, 128); | 285 EXPECT_ALIGNED(align128, 128); |
| 286 EXPECT_ALIGNED(align4096, 4096); | 286 EXPECT_ALIGNED(align4096, 4096); |
| 287 } | 287 } |
| 288 |
| 289 } // namespace |
| 290 } // namespace base |
| OLD | NEW |