From 849daad79921791a1ee3c13105c9b345bf61b043 Mon Sep 17 00:00:00 2001
From: "opensearch-trigger-bot[bot]"
 <98922864+opensearch-trigger-bot[bot]@users.noreply.github.com>
Date: Tue, 29 Oct 2024 13:20:05 -0500
Subject: [PATCH] add configurable iterations + warmup iterations to the pmc
 workload (#452) (#468)

(cherry picked from commit 9fcad568a9cbbd1d9a3b2842c1fc29c6e0c4dd9a)

Signed-off-by: Michael Oviedo <mikeovi@amazon.com>
Signed-off-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com>
Co-authored-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com>
---
 pmc/test_procedures/default.json | 24 ++++++++++++------------
 1 file changed, 12 insertions(+), 12 deletions(-)

diff --git a/pmc/test_procedures/default.json b/pmc/test_procedures/default.json
index e5f0723b..e49e54c5 100644
--- a/pmc/test_procedures/default.json
+++ b/pmc/test_procedures/default.json
@@ -70,43 +70,43 @@
         },
         {
           "operation": "default",
-          "warmup-iterations": 500,
-          "iterations": 200,
+          "warmup-iterations": {{ default_warmup_iterations or warmup_iterations | default(500) | tojson }},
+          "iterations": {{ default_iterations or iterations | default(200) | tojson }},
           "target-throughput": {{ default_target_throughput or target_throughput | default(20) | tojson }},
           "clients": {{ default_search_clients or search_clients | default(1) }}
         },
         {
           "operation": "term",
-          "warmup-iterations": 500,
-          "iterations": 200,
+          "warmup-iterations": {{ term_warmup_iterations or warmup_iterations | default(500) | tojson }},
+          "iterations": {{ term_iterations or iterations | default(200) | tojson }},
           "target-throughput": {{ term_target_throughput or target_throughput | default(20) | tojson }},
           "clients": {{ term_search_clients or search_clients | default(1) }}
         },
         {
           "operation": "phrase",
-          "warmup-iterations": 500,
-          "iterations": 200,
+          "warmup-iterations": {{ phrase_warmup_iterations or warmup_iterations | default(500) | tojson }},
+          "iterations": {{ phrase_iterations or iterations | default(200) | tojson }},
           "target-throughput": {{ phrase_target_throughput or target_throughput | default(20) | tojson }},
           "clients": {{ phrase_search_clients or search_clients | default(1) }}
         },
         {
           "operation": "articles_monthly_agg_uncached",
-          "warmup-iterations": 500,
-          "iterations": 200,
+          "warmup-iterations": {{ articles_monthly_agg_uncached_warmup_iterations or warmup_iterations | default(500) | tojson }},
+          "iterations": {{ articles_monthly_agg_uncached_iterations or iterations | default(200) | tojson }},
           "target-throughput": {{ articles_monthly_agg_uncached_target_throughput or target_throughput | default(20) | tojson }},
           "clients": {{ articles_monthly_agg_uncached_search_clients or search_clients | default(1) }}
         },
         {
           "operation": "articles_monthly_agg_cached",
-          "warmup-iterations": 500,
-          "iterations": 200,
+          "warmup-iterations": {{ articles_monthly_agg_cached_warmup_iterations or warmup_iterations | default(500) | tojson }},
+          "iterations": {{ articles_monthly_agg_cached_iterations or iterations | default(200) | tojson }},
           "target-throughput": {{ articles_monthly_agg_cached_target_throughput or target_throughput | default(20) | tojson }},
           "clients": {{ articles_monthly_agg_cached_search_clients or search_clients | default(1) }}
         },
         {
           "operation": "scroll",
-          "warmup-iterations": 50,
-          "iterations": 100,
+          "warmup-iterations": {{ scroll_warmup_iterations or warmup_iterations | default(50) | tojson }},
+          "iterations": {{ scroll_iterations or iterations | default(100) | tojson }},
           "target-throughput": {{ scroll_target_throughput or target_throughput | default(0.5) | tojson }},
           "clients": {{ scroll_search_clients or search_clients | default(1) }}
         }