From d17d3d9e06cdc30417a2d53486d2cf324ebf9f4f Mon Sep 17 00:00:00 2001
From: notshivansh <shivansh@akto.io>
Date: Thu, 2 Jan 2025 13:36:30 +0530
Subject: [PATCH] add clean up attribute

---
 .../akto/dto/testing/TestingRunConfig.java    | 22 ++++++++++++++++++-
 1 file changed, 21 insertions(+), 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
index baff43a709..615704e77d 100644
--- 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;
+    }
+
 }