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

   1 #include "common.h"
   2 #include <cds/misc/RigtorpMPMCQueue.h>
   3 #include <cds_test/stress_test.h>
   4 #include <ctime>
   5 #include <iostream>
   6
   7 using namespace std;
   8
   9 namespace {
  10
  11 class rigtorpMPMCQueueTest : public cds_test::stress_fixture {
  12 protected:
  13   static size_t s_nRigtorpMPMCQueuePassCount;
  14   static size_t s_nRigtorpMPMCQueueEnqueueStride;
  15   static size_t s_nRigtorpMPMCQueueCapacity;
  16
  17   static void SetUpTestCase() {
  18     cds_test::config const &cfg = get_config("SequentialMisc");
  19     GetConfigExpected(RigtorpMPMCQueuePassCount, 10000);
  20     GetConfigExpected(RigtorpMPMCQueueEnqueueStride, 1024);
  21     GetConfigExpected(RigtorpMPMCQueueCapacity, 2048);
  22     if (s_nRigtorpMPMCQueueCapacity <= s_nRigtorpMPMCQueueEnqueueStride) {
  23       s_nRigtorpMPMCQueueCapacity = 2 * s_nRigtorpMPMCQueueEnqueueStride;
  24     }
  25   }
  26
  27   void test() {
  28     std::unique_ptr<rigtorp::MPMCQueue<size_t>> q(
  29         new rigtorp::MPMCQueue<size_t>(s_nRigtorpMPMCQueueCapacity));
  30     size_t nNo = 0;
  31     size_t pop_sum = 0;
  32
  33     while (nNo < s_nRigtorpMPMCQueuePassCount) {
  34       size_t curr_push_count = std::min(s_nRigtorpMPMCQueuePassCount - nNo,
  35                                         s_nRigtorpMPMCQueueEnqueueStride);
  36       for (size_t i = 0; i < curr_push_count; i++) {
  37         q->push(nNo);
  38         ++nNo;
  39       }
  40
  41       for (size_t i = 0; i < curr_push_count; i++) {
  42         size_t res;
  43         q->pop(res);
  44         pop_sum += res;
  45       }
  46     }
  47
  48     size_t supposed_sum =
  49         s_nRigtorpMPMCQueuePassCount * (s_nRigtorpMPMCQueuePassCount - 1) / 2;
  50     EXPECT_EQ(pop_sum, supposed_sum);
  51   }
  52 };
  53
  54 size_t rigtorpMPMCQueueTest::s_nRigtorpMPMCQueuePassCount;
  55 size_t rigtorpMPMCQueueTest::s_nRigtorpMPMCQueueEnqueueStride;
  56 size_t rigtorpMPMCQueueTest::s_nRigtorpMPMCQueueCapacity;
  57
  58 TEST_F(rigtorpMPMCQueueTest, PushPop) {
  59   test();
  60 }
  61
  62 } // namespace