Reorgs added benchmarks (put them in misc folder)
[libcds.git] / test / stress / sequential / sequential-misc / rigtorp_mpmc_driver.cpp
diff --git a/test/stress/sequential/sequential-misc/rigtorp_mpmc_driver.cpp b/test/stress/sequential/sequential-misc/rigtorp_mpmc_driver.cpp
new file mode 100644 (file)
index 0000000..743a64a
--- /dev/null
@@ -0,0 +1,61 @@
+#include "common.h"
+#include <cds/misc/RigtorpMPMCQueue.h>
+#include <cds_test/stress_test.h>
+#include <ctime>
+#include <iostream>
+
+using namespace std;
+
+namespace {
+
+class rigtorpMPMCQueueTest : public cds_test::stress_fixture {
+protected:
+  static size_t s_nRigtorpMPMCQueuePassCount;
+  static size_t s_nRigtorpMPMCQueueEnqueueStride;
+  static size_t s_nRigtorpMPMCQueueCapacity;
+
+  static void SetUpTestCase() {
+    cds_test::config const &cfg = get_config("Misc");
+    GetConfig(RigtorpMPMCQueuePassCount);
+    GetConfig(RigtorpMPMCQueueEnqueueStride);
+    GetConfig(RigtorpMPMCQueueCapacity);
+  }
+
+  void test() {
+    rigtorp::MPMCQueue<size_t> q(s_nRigtorpMPMCQueueCapacity);
+    size_t nNo = 0;
+    size_t pop_sum = 0;
+
+    while (nNo < s_nRigtorpMPMCQueuePassCount) {
+      size_t curr_push_count =
+          std::min(s_nRigtorpMPMCQueuePassCount - nNo, s_nRigtorpMPMCQueueEnqueueStride);
+      for (size_t i = 0; i < curr_push_count; i++) {
+        q.push(nNo);
+        ++nNo;
+      }
+
+      for (size_t i = 0; i < curr_push_count; i++) {
+        size_t res;
+        q.pop(res);
+        pop_sum += res;
+      }
+    }
+
+    size_t supposed_sum =
+        s_nRigtorpMPMCQueuePassCount * (s_nRigtorpMPMCQueuePassCount - 1) / 2;
+    if (pop_sum != supposed_sum) {
+      std::cout << "Sequential rigtorpMPMC queue pop sum: " << pop_sum
+                << " != " << supposed_sum << "\n";
+    }
+  }
+};
+
+size_t rigtorpMPMCQueueTest::s_nRigtorpMPMCQueuePassCount;
+size_t rigtorpMPMCQueueTest::s_nRigtorpMPMCQueueEnqueueStride;
+size_t rigtorpMPMCQueueTest::s_nRigtorpMPMCQueueCapacity;
+
+TEST_F(rigtorpMPMCQueueTest, PushPop) {
+  test();
+}
+
+} // namespace