OLD | NEW |
| (Empty) |
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 | |
27 void CancelHelper(CancellationFlag* flag) { | |
28 #if GTEST_HAS_DEATH_TEST | |
29 ASSERT_DEBUG_DEATH(flag->Set(), ""); | |
30 #endif | |
31 } | |
32 | |
33 TEST(CancellationFlagTest, SimpleSingleThreadedTest) { | |
34 CancellationFlag flag; | |
35 ASSERT_FALSE(flag.IsSet()); | |
36 flag.Set(); | |
37 ASSERT_TRUE(flag.IsSet()); | |
38 } | |
39 | |
40 TEST(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 | |
49 TEST(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 | |
OLD | NEW |