Update upgrade lines

rh-messaging-qe · Sep 27, 2024 · 20edaaf · 20edaaf
1 parent 6d4faff
commit 20edaaf
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/standalone-suite/src/test/java/io/brokerqe/claire/upgrade/UpgradeTests.java b/standalone-suite/src/test/java/io/brokerqe/claire/upgrade/UpgradeTests.java
@@ -26,14 +26,18 @@ public abstract class UpgradeTests extends AbstractSystemTests {
     protected static int messagesReceivedTotal = 0;
     protected static int upgradeCount;
 
+    private static final String VERSION = "version";
+    private static final String ARTEMIS_VERSION = "artemisVersion";
+    private static final String ARTEMIS_ZIP_URL = "artemisZipUrl";
+
     Stream<? extends Arguments> getUpgradePlanArguments() {
         LOGGER.info("[UpgradeTestPlan] {}", getEnvironment().getTestUpgradePlanContent());
         ArrayList<HashMap<String, String>> mapped = new Yaml().load(getEnvironment().getTestUpgradePlanContent());
         messagesSentInitials = 2000 * (mapped.size() - 1);
         messagesReceivePartial = messagesSentInitials / (mapped.size() - 1);
         upgradeCount = mapped.size();
         return  mapped.stream().map(line ->
-                Arguments.of(line.get("version"), line.get("artemis_version"), line.get("artemis_zip_url"))
+                Arguments.of(line.get(VERSION), line.get(ARTEMIS_VERSION), line.get(ARTEMIS_ZIP_URL))
         );
     }