added mandatory keys in test

stfbk · Oct 19, 2023 · 86620b3 · 86620b3
1 parent 75eb2ca
commit 86620b3
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/tool/src/main/java/migt/Test.java b/tool/src/main/java/migt/Test.java
@@ -38,6 +38,7 @@ public class Test {
     // Infos
     String name;
     String description;
+    List<String> mandatory_keys = new ArrayList<>();
 
     /**
      * Empty constructor for tests
@@ -58,6 +59,13 @@ public Test(JSONObject test_json,
         setType(test_json.getString("type"));
 
         Iterator<String> keys = test_json.keys();
+
+        for (String k : mandatory_keys) {
+            if (!test_json.keySet().contains(k)) {
+                throw new ParsingException("Test is missing required \"" + k + "\" key");
+            }
+        }
+
         while (keys.hasNext()) {
             String key = keys.next();
 
@@ -144,6 +152,11 @@ public void init() {
         references = "";
         violated_properties = "";
         mitigations = "";
+
+        mandatory_keys.add("result");
+        mandatory_keys.add("name");
+        mandatory_keys.add("type");
+        mandatory_keys.add("sessions");
     }
 
     public String getName() {