From 8d4d52d3961263a72a071ded1fc418b41bba22a8 Mon Sep 17 00:00:00 2001
From: Craig Tiller <ctiller@google.com>
Date: Sat, 7 Oct 2017 17:28:58 -0700
Subject: Multithread & shard stats test, make it exhaustive

---
 test/core/debug/stats_test.cc | 74 ++++++++++++++++++++++++++-----------------
 1 file changed, 45 insertions(+), 29 deletions(-)

(limited to 'test/core/debug')

diff --git a/test/core/debug/stats_test.cc b/test/core/debug/stats_test.cc
index c85ab3598a..35f11eae37 100644
--- a/test/core/debug/stats_test.cc
+++ b/test/core/debug/stats_test.cc
@@ -20,7 +20,10 @@ extern "C" {
 #include "src/core/lib/debug/stats.h"
 }
 
+#include <thread>
+
 #include <grpc/grpc.h>
+#include <grpc/support/cpu.h>
 #include <grpc/support/log.h>
 #include <gtest/gtest.h>
 
@@ -79,38 +82,51 @@ static int FindExpectedBucket(int i, int j) {
          grpc_stats_histo_bucket_boundaries[i] - 1;
 }
 
-TEST(StatsTest, IncHistogram) {
-  for (int i = 0; i < GRPC_STATS_HISTOGRAM_COUNT; i++) {
-    std::vector<int> test_values;
-    for (int j = -1000;
-         j <
-         grpc_stats_histo_bucket_boundaries[i]
-                                           [grpc_stats_histo_buckets[i] - 1] +
-             1000;
-         j++) {
-      test_values.push_back(j);
-    }
-    std::random_shuffle(test_values.begin(), test_values.end());
-    if (test_values.size() > 10000) {
-      test_values.resize(10000);
-    }
-    for (auto j : test_values) {
-      Snapshot snapshot;
-
-      int expected_bucket = FindExpectedBucket(i, j);
-
-      grpc_exec_ctx exec_ctx = GRPC_EXEC_CTX_INIT;
-      grpc_stats_inc_histogram[i](&exec_ctx, j);
-      grpc_exec_ctx_finish(&exec_ctx);
-
-      auto delta = snapshot.delta();
-
-      EXPECT_EQ(delta.histograms[grpc_stats_histo_start[i] + expected_bucket],
-                1);
-    }
+class HistogramTest : public ::testing::TestWithParam<int> {};
+
+TEST_P(HistogramTest, IncHistogram) {
+  const int kHistogram = GetParam();
+  const int kThreads = std::max(1, (int)gpr_cpu_num_cores());
+  std::vector<std::thread> threads;
+  for (int thread = 0; thread < kThreads; thread++) {
+    threads.emplace_back([kHistogram, kThreads, thread]() {
+      std::vector<int> test_values;
+      for (int j = -1000 + thread;
+           j < grpc_stats_histo_bucket_boundaries
+                       [kHistogram][grpc_stats_histo_buckets[kHistogram] - 1] +
+                   1000;
+           j += kThreads) {
+        test_values.push_back(j);
+      }
+      std::random_shuffle(test_values.begin(), test_values.end());
+      for (auto j : test_values) {
+        Snapshot snapshot;
+
+        int expected_bucket = FindExpectedBucket(kHistogram, j);
+
+        grpc_exec_ctx exec_ctx = GRPC_EXEC_CTX_INIT;
+        grpc_stats_inc_histogram[kHistogram](&exec_ctx, j);
+        grpc_exec_ctx_finish(&exec_ctx);
+
+        auto delta = snapshot.delta();
+
+        EXPECT_GE(delta.histograms[grpc_stats_histo_start[kHistogram] +
+                                   expected_bucket],
+                  1)
+            << "\nhistogram:" << kHistogram
+            << "\nexpected_bucket:" << expected_bucket << "\nthread:" << thread
+            << "\nj:" << j;
+      }
+    });
+  }
+  for (auto& t : threads) {
+    t.join();
   }
 }
 
+INSTANTIATE_TEST_CASE_P(HistogramTestCases, HistogramTest,
+                        ::testing::Range<int>(0, GRPC_STATS_HISTOGRAM_COUNT));
+
 }  // namespace testing
 }  // namespace grpc
 
-- 
cgit v1.2.3