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 19namespace base { 20 21namespace { 22 23//------------------------------------------------------------------------------ 24// Define our test class. 25//------------------------------------------------------------------------------ 26 27void CancelHelper(CancellationFlag* flag) { 28#if GTEST_HAS_DEATH_TEST 29 ASSERT_DEBUG_DEATH(flag->Set(), ""); 30#endif 31} 32 33TEST(CancellationFlagTest, SimpleSingleThreadedTest) { 34 CancellationFlag flag; 35 ASSERT_FALSE(flag.IsSet()); 36 flag.Set(); 37 ASSERT_TRUE(flag.IsSet()); 38} 39 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 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