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