From 43736c1f5fe169103cba339a2ada0ccb10857d73 Mon Sep 17 00:00:00 2001
From: Andrei Matveyeu <andreimu@axis.com>
Date: Wed, 21 Feb 2024 15:09:00 +0100
Subject: [PATCH] add terminationGracePeriodSeconds to Job template

---
 src/suite_starter/esr_yaml.py      | 2 ++
 src/suite_starter/suite_starter.py | 1 +
 2 files changed, 3 insertions(+)

diff --git a/src/suite_starter/esr_yaml.py b/src/suite_starter/esr_yaml.py
index c6a0df1..710d175 100644
--- a/src/suite_starter/esr_yaml.py
+++ b/src/suite_starter/esr_yaml.py
@@ -29,6 +29,7 @@
     "    metadata:\n"
     "      name: {job_name}\n"
     "    spec:\n"
+    "      terminationGracePeriodSeconds: {termination_grace_period}\n"
     "      volumes:\n"
     "      - name: kubexit\n"
     "        emptyDir: {{}}\n"
@@ -124,6 +125,7 @@
     "    metadata:\n"
     "      name: {job_name}\n"
     "    spec:\n"
+    "      terminationGracePeriodSeconds: {termination_grace_period}\n"
     "      volumes:\n"
     "      - name: shared-logs\n"
     "        emptyDir: {{}}\n"
diff --git a/src/suite_starter/suite_starter.py b/src/suite_starter/suite_starter.py
index 72fbab5..06677bd 100644
--- a/src/suite_starter/suite_starter.py
+++ b/src/suite_starter/suite_starter.py
@@ -71,6 +71,7 @@ def suite_runner_callback(self, event, _):
         data["etos_configmap"] = os.getenv("ETOS_CONFIGMAP")
         data["etos_rabbitmq_secret"] = os.getenv("ETOS_RABBITMQ_SECRET")
         data["ttl"] = os.getenv("ETOS_ESR_TTL", "3600")
+        data["termination_grace_period"] = os.getenv("ETOS_TERMINATION_GRACE_PERIOD", "300")
         data["docker_image"] = self.etos.config.get("suite_runner")
         data["log_listener"] = self.etos.config.get("log_listener")
         data["suite_id"] = suite_id