From c35801438978b611f35092c81cfaa65fb671edcf Mon Sep 17 00:00:00 2001
From: Bojan Serafimov <bojan.serafimov7@gmail.com>
Date: Mon, 4 Dec 2023 15:15:25 -0500
Subject: [PATCH] configurable n_iterations

---
 pageserver/src/walingest.rs | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/pageserver/src/walingest.rs b/pageserver/src/walingest.rs
index d884cc5af7..b62f997994 100644
--- a/pageserver/src/walingest.rs
+++ b/pageserver/src/walingest.rs
@@ -2126,7 +2126,12 @@ mod tests {
             buffer
         };
 
-        let n_iterations = 1000;
+        // Allow number of iterations to be configured via env var, which is
+        // useful when using this test for benchmarking.
+        let n_iterations: usize =
+            std::env::var("NUM_TEST_ITERATIONS")
+            .map(|s| s.parse().unwrap())
+            .unwrap_or(1);
         let profiler = crate::profiling::init_profiler();
 
         for iteration in 0..n_iterations {