2 * Copyright 2016 Facebook, Inc.
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
8 * http://www.apache.org/licenses/LICENSE-2.0
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
19 #include <folly/Baton.h>
20 #include <folly/test/DeterministicSchedule.h>
22 #include <gtest/gtest.h>
27 typedef DeterministicSchedule DSched;
29 template <template <typename> class Atom>
30 void run_pingpong_test(int numRounds) {
31 Baton<Atom> batons[17];
32 Baton<Atom>& a = batons[0];
33 Baton<Atom>& b = batons[16]; // to get it on a different cache line
34 auto thr = DSched::thread([&] {
35 for (int i = 0; i < numRounds; ++i) {
41 for (int i = 0; i < numRounds; ++i) {
49 template <template <typename> class Atom, typename Clock>
50 void run_basic_timed_wait_tests() {
53 // tests if early delivery works fine
54 EXPECT_TRUE(b.timed_wait(Clock::now()));
57 template <template <typename> class Atom, typename Clock>
58 void run_timed_wait_tmo_tests() {
61 auto thr = DSched::thread([&] {
62 bool rv = b.timed_wait(Clock::now() + std::chrono::milliseconds(1));
63 // main thread is guaranteed to not post until timeout occurs
69 template <template <typename> class Atom, typename Clock>
70 void run_timed_wait_regular_test() {
73 auto thr = DSched::thread([&] {
74 // To wait forever we'd like to use time_point<Clock>::max, but
75 // std::condition_variable does math to convert the timeout to
76 // system_clock without handling overflow.
77 auto farFuture = Clock::now() + std::chrono::hours(1000);
78 bool rv = b.timed_wait(farFuture);
79 if (!std::is_same<Atom<int>, DeterministicAtomic<int>>::value) {
80 // DeterministicAtomic ignores actual times, so doesn't guarantee
86 if (!std::is_same<Atom<int>, DeterministicAtomic<int>>::value) {
87 // If we are using std::atomic (or EmulatedFutexAtomic) then
88 // a sleep here guarantees to a large extent that 'thr' will
89 // execute wait before we post it, thus testing late delivery. For
90 // DeterministicAtomic, we just rely on DeterministicSchedule to do
91 // the scheduling. The test won't fail if we lose the race, we just
92 // don't get coverage.
93 std::this_thread::sleep_for(std::chrono::milliseconds(2));
100 template <template <typename> class Atom>
101 void run_try_wait_tests() {
103 EXPECT_FALSE(b.try_wait());
105 EXPECT_TRUE(b.try_wait());