From cbd1d22aef2dfc2f9d50e46eacbdbb421446f731 Mon Sep 17 00:00:00 2001
From: anirudhwarrier <12178754+anirudhwarrier@users.noreply.github.com>
Date: Thu, 23 May 2024 20:05:21 +0530
Subject: [PATCH 1/2] bump load test DB size to 20G

---
 integration-tests/load/automationv2_1/automationv2_1_test.go | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/integration-tests/load/automationv2_1/automationv2_1_test.go b/integration-tests/load/automationv2_1/automationv2_1_test.go
index bd1fcfa1559..506f717b578 100644
--- a/integration-tests/load/automationv2_1/automationv2_1_test.go
+++ b/integration-tests/load/automationv2_1/automationv2_1_test.go
@@ -101,7 +101,7 @@ Password = '%s'`
 			},
 		},
 		"stateful": true,
-		"capacity": "10Gi",
+		"capacity": "20Gi",
 	}
 
 	recNodeSpec = map[string]interface{}{

From 6c220a98c62b53d5502c8b7771e8c07727b4a46d Mon Sep 17 00:00:00 2001
From: anirudhwarrier <12178754+anirudhwarrier@users.noreply.github.com>
Date: Fri, 24 May 2024 10:01:36 +0530
Subject: [PATCH 2/2] add RemoveNamespace test config

---
 .../load/automationv2_1/automationv2_1_test.go             | 2 +-
 integration-tests/testconfig/automation/automation.toml    | 1 +
 integration-tests/testconfig/automation/config.go          | 7 +++++++
 3 files changed, 9 insertions(+), 1 deletion(-)

diff --git a/integration-tests/load/automationv2_1/automationv2_1_test.go b/integration-tests/load/automationv2_1/automationv2_1_test.go
index 506f717b578..c560bc25700 100644
--- a/integration-tests/load/automationv2_1/automationv2_1_test.go
+++ b/integration-tests/load/automationv2_1/automationv2_1_test.go
@@ -798,7 +798,7 @@ Test Duration: %s`
 			if err != nil {
 				l.Error().Err(err).Msg("Error increasing TTL of namespace")
 			}
-		} else if chainClient.Cfg.IsSimulatedNetwork() {
+		} else if chainClient.Cfg.IsSimulatedNetwork() || *loadedTestConfig.Automation.General.RemoveNamespace {
 			err := testEnvironment.Client.RemoveNamespace(testEnvironment.Cfg.Namespace)
 			if err != nil {
 				l.Error().Err(err).Msg("Error removing namespace")
diff --git a/integration-tests/testconfig/automation/automation.toml b/integration-tests/testconfig/automation/automation.toml
index 86eb279de39..0cfc2e87f42 100644
--- a/integration-tests/testconfig/automation/automation.toml
+++ b/integration-tests/testconfig/automation/automation.toml
@@ -109,6 +109,7 @@ ephemeral_addresses_number = 100
 
 [Load.Common]
 chainlink_node_funding = 100
+remove_namespace = true
 
 [Load.Automation]
 [Load.Automation.General]
diff --git a/integration-tests/testconfig/automation/config.go b/integration-tests/testconfig/automation/config.go
index 103f963d881..bde989f544b 100644
--- a/integration-tests/testconfig/automation/config.go
+++ b/integration-tests/testconfig/automation/config.go
@@ -48,6 +48,7 @@ type General struct {
 	SpecType              *string `toml:"spec_type"`
 	ChainlinkNodeLogLevel *string `toml:"chainlink_node_log_level"`
 	UsePrometheus         *bool   `toml:"use_prometheus"`
+	RemoveNamespace       *bool   `toml:"remove_namespace"`
 }
 
 func (c *General) Validate() error {
@@ -66,6 +67,12 @@ func (c *General) Validate() error {
 	if c.ChainlinkNodeLogLevel == nil {
 		return errors.New("chainlink_node_log_level must be set")
 	}
+	if c.UsePrometheus == nil {
+		return errors.New("use_prometheus must be set")
+	}
+	if c.RemoveNamespace == nil {
+		return errors.New("remove_namespace must be set")
+	}
 
 	return nil
 }