add clean up attribute

akto-api-security · Jan 2, 2025 · d17d3d9 · d17d3d9
1 parent beaabda
commit d17d3d9
Showing 1 changed file with 21 additions and 1 deletion.
diff --git a/libs/dao/src/main/java/com/akto/dto/testing/TestingRunConfig.java b/libs/dao/src/main/java/com/akto/dto/testing/TestingRunConfig.java
@@ -25,16 +25,26 @@ public class TestingRunConfig {
     private String overriddenTestAppUrl;
 
     private List<TestConfigsAdvancedSettings> configsAdvancedSettings;
+    private boolean cleanUp;
+
     public TestingRunConfig() {}
+
+    public TestingRunConfig(int id, Map<Integer, List<ApiInfo.ApiInfoKey>> collectionWiseApiInfoKey,
+            List<String> testSubCategoryList, ObjectId authMechanismId, String overriddenTestAppUrl,
+            String testRoleId) {
+        this(id, collectionWiseApiInfoKey, testSubCategoryList, authMechanismId, overriddenTestAppUrl, testRoleId, false);
+    }
+
     public TestingRunConfig(int id, Map<Integer, List<ApiInfo.ApiInfoKey>> collectionWiseApiInfoKey,
                             List<String> testSubCategoryList,
-                            ObjectId authMechanismId, String overriddenTestAppUrl, String testRoleId) {
+                            ObjectId authMechanismId, String overriddenTestAppUrl, String testRoleId, boolean cleanUp) {
         this.id = id;
         this.collectionWiseApiInfoKey = collectionWiseApiInfoKey;
         this.testSubCategoryList = testSubCategoryList;
         this.authMechanismId = authMechanismId;
         this.overriddenTestAppUrl = overriddenTestAppUrl;
         this.testRoleId = testRoleId;
+        this.cleanUp = cleanUp;
     }
 
     public List<String> getTestSubCategoryList() {
@@ -101,6 +111,8 @@ public void rebaseOn(TestingRunConfig that) {
         if(this.testRoleId == null) {
             this.testRoleId = that.testRoleId;
         }
+
+        this.cleanUp = that.cleanUp;
     }
 
     public String getTestRoleId() {
@@ -128,4 +140,12 @@ public void setConfigsAdvancedSettings(List<TestConfigsAdvancedSettings> configs
         this.configsAdvancedSettings = configsAdvancedSettings;
     }
 
+    public boolean getCleanUp() {
+        return this.cleanUp;
+    }
+
+    public void setCleanUp(boolean cleanUp) {
+        this.cleanUp = cleanUp;
+    }
+
 }