test/stress/sequential/sequential-misc/barrier_driver.cpp

   1 #include "common.h"
   2 #include <atomic>
   3 #include <cds/gc/dhp.h>
   4 #include <cds/gc/hp.h>
   5 #include <cds/misc/barrier.h>
   6 #include <cds_test/stress_test.h>
   7 #include <iostream>
   8 #include <memory>
   9 #include <vector>
  10
  11 using namespace std;
  12
  13 namespace {
  14
  15 typedef cds_others::SpinBarrier Barrier;
  16
  17 static size_t s_nBarrierThreadCount = 2;
  18 static size_t s_nBarrierPassCount = 1;
  19
  20 class BarrierTest : public cds_test::stress_fixture {
  21 protected:
  22   static size_t count;
  23
  24   static void SetUpTestCase() {
  25     cds_test::config const &cfg = get_config("SequentialMisc");
  26     GetConfig(BarrierPassCount);
  27     GetConfig(BarrierThreadCount);
  28   }
  29 };
  30
  31 size_t BarrierTest::count;
  32
  33 TEST_F(BarrierTest, Wait) {
  34   std::unique_ptr<Barrier> barrier(new Barrier(1));
  35   for (size_t i = 0; i < s_nBarrierPassCount; i++) {
  36     if (barrier->wait()) {
  37       count++;
  38     }
  39   }
  40   EXPECT_EQ(count, s_nBarrierPassCount);
  41 }
  42
  43 } // namespace