From 0e374700f645357881e4b70ef3543a9a4d7be841 Mon Sep 17 00:00:00 2001
From: Nicolas Peltier <1032754+npeltier@users.noreply.github.com>
Date: Fri, 6 Dec 2024 16:29:36 +0100
Subject: [PATCH] MWPW-162805 allow stage tests (#106)

---
 k6-scripts/wcs-fetches.js | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/k6-scripts/wcs-fetches.js b/k6-scripts/wcs-fetches.js
index 2cc5dfff..7b921852 100644
--- a/k6-scripts/wcs-fetches.js
+++ b/k6-scripts/wcs-fetches.js
@@ -29,7 +29,9 @@ export default function () {
     const locale = { country: 'US', locale: 'en_US' };
     const osiIndex = (__VU - 1 + __ITER) % osis.length; // rotates per user & iteration
     const osi = osis[osiIndex];
-    const url = `${__ENV.TEST_WCS_URL}${osi}&country=${locale.country}&language=${locale.country === 'GB' ? 'EN' : 'MULT'}&locale=${locale.locale}&api_key=${api_key}&landscape=PUBLISHED`;
+    const landscape =
+        __ENV.TEST_WCS_URL?.indexOf('.stage.') > 0 ? 'ALL' : 'PUBLISHED';
+    const url = `${__ENV.TEST_WCS_URL}${osi}&country=${locale.country}&language=${locale.country === 'GB' ? 'EN' : 'MULT'}&locale=${locale.locale}&api_key=${api_key}&landscape=${landscape}`;
     const res = http.get(url);
 
     // Assertions to validate response