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 // Tests of CancellationFlag class. 6 7 #include "base/synchronization/cancellation_flag.h" 8 9 #include "base/bind.h" 10 #include "base/logging.h" 11 #include "base/message_loop/message_loop.h" 12 #include "base/synchronization/spin_wait.h" 13 #include "base/threading/thread.h" 14 #include "base/time/time.h" 15 #include "testing/gtest/include/gtest/gtest.h" 16 #include "testing/platform_test.h" 17 18 namespace base { 19 20 namespace { 21 22 //------------------------------------------------------------------------------ 23 // Define our test class. 24 //------------------------------------------------------------------------------ 25 CancelHelper(CancellationFlag * flag)26void CancelHelper(CancellationFlag* flag) { 27 #if GTEST_HAS_DEATH_TEST 28 ASSERT_DEBUG_DEATH(flag->Set(), ""); 29 #endif 30 } 31 TEST(CancellationFlagTest,SimpleSingleThreadedTest)32TEST(CancellationFlagTest, SimpleSingleThreadedTest) { 33 CancellationFlag flag; 34 ASSERT_FALSE(flag.IsSet()); 35 flag.Set(); 36 ASSERT_TRUE(flag.IsSet()); 37 } 38 TEST(CancellationFlagTest,DoubleSetTest)39TEST(CancellationFlagTest, DoubleSetTest) { 40 CancellationFlag flag; 41 ASSERT_FALSE(flag.IsSet()); 42 flag.Set(); 43 ASSERT_TRUE(flag.IsSet()); 44 flag.Set(); 45 ASSERT_TRUE(flag.IsSet()); 46 } 47 TEST(CancellationFlagTest,SetOnDifferentThreadDeathTest)48TEST(CancellationFlagTest, SetOnDifferentThreadDeathTest) { 49 // Checks that Set() can't be called from any other thread. 50 // CancellationFlag should die on a DCHECK if Set() is called from 51 // other thread. 52 ::testing::FLAGS_gtest_death_test_style = "threadsafe"; 53 Thread t("CancellationFlagTest.SetOnDifferentThreadDeathTest"); 54 ASSERT_TRUE(t.Start()); 55 ASSERT_TRUE(t.message_loop()); 56 ASSERT_TRUE(t.IsRunning()); 57 58 CancellationFlag flag; 59 t.message_loop()->PostTask(FROM_HERE, base::Bind(&CancelHelper, &flag)); 60 } 61 62 } // namespace 63 64 } // namespace base 65