Add ttlSecondsAfterFinished to migrateDatabaseJob and createUserJob

chart/templates/jobs/create-user-job.yaml

@@ -48,6 +48,9 @@ metadata:
     {{- $annotations | toYaml | nindent 4 }}
   {{- end }}
 spec:
+  {{- if not (kindIs "invalid" .Values.createUserJob.ttlSecondsAfterFinished) }}
+  ttlSecondsAfterFinished: {{ .Values.createUserJob.ttlSecondsAfterFinished }}
+  {{- end }}
   template:
     metadata:
       labels:

chart/templates/jobs/migrate-database-job.yaml

@@ -48,6 +48,9 @@ metadata:
     {{- $annotations | toYaml | nindent 4 }}
   {{- end }}
 spec:
+  {{- if not (kindIs "invalid" .Values.migrateDatabaseJob.ttlSecondsAfterFinished) }}
+  ttlSecondsAfterFinished: {{ .Values.migrateDatabaseJob.ttlSecondsAfterFinished }}
+  {{- end }}
   template:
     metadata:
       labels:

chart/values.schema.json

@@ -2945,6 +2945,14 @@
                     "type": "boolean",
                     "default": true
                 },
+                "ttlSecondsAfterFinished": {
+                    "description": "Limit the lifetime of the job object after it finished execution",
+                    "type": [
+                        "integer",
+                        "null"
+                    ],
+                    "default": null
+                },
                 "env": {
                     "description": "Add additional env vars to the create user job pod.",
                     "type": "array",
@@ -3144,6 +3152,14 @@
                     "description": "Specify if you want additional configured env vars applied to this job",
                     "type": "boolean",
                     "default": true
+                },
+                "ttlSecondsAfterFinished": {
+                    "description": "Limit the lifetime of the job object after it finished execution",
+                    "type": [
+                        "integer",
+                        "null"
+                    ],
+                    "default": null
                 }
             }
         },

chart/values.yaml

@@ -759,6 +759,8 @@ scheduler:
 
 # Airflow create user job settings
 createUserJob:
+  # Limit the lifetime of the job object after it finished execution.
+  ttlSecondsAfterFinished: ~
   # Command to use when running the create user job (templated).
   command: ~
   # Args to use when running the create user job (templated).
@@ -837,6 +839,8 @@ createUserJob:
 # Airflow database migration job settings
 migrateDatabaseJob:
   enabled: true
+  # Limit the lifetime of the job object after it finished execution.
+  ttlSecondsAfterFinished: ~
   # Command to use when running the migrate database job (templated).
   command: ~
   # Args to use when running the migrate database job (templated).

tests/charts/test_create_user_job.py

@@ -241,6 +241,27 @@ def test_should_disable_custom_env(self):
         assert {"name": "foo", "value": "bar"} not in envs
         assert {"name": "extraFoo", "value": "extraBar"} not in envs
 
+    def test_job_ttl_after_finished(self):
+        docs = render_chart(
+            values={"createUserJob": {"ttlSecondsAfterFinished": 1}},
+            show_only=["templates/jobs/create-user-job.yaml"],
+        )
+        ttl = jmespath.search("spec.ttlSecondsAfterFinished", docs[0])
+        assert ttl == 1
+
+    def test_job_ttl_after_finished_zero(self):
+        docs = render_chart(
+            values={"createUserJob": {"ttlSecondsAfterFinished": 0}},
+            show_only=["templates/jobs/create-user-job.yaml"],
+        )
+        ttl = jmespath.search("spec.ttlSecondsAfterFinished", docs[0])
+        assert ttl == 0
+
+    def test_job_ttl_after_finished_nil(self):
+        docs = render_chart(show_only=["templates/jobs/create-user-job.yaml"])
+        spec = jmespath.search("spec", docs[0])
+        assert "ttlSecondsAfterFinished" not in spec
+
     @pytest.mark.parametrize(
         "airflow_version, expected_arg",
         [

tests/charts/test_migrate_database_job.py

@@ -235,6 +235,27 @@ def test_should_add_global_volume_and_global_volume_mount(self):
             "spec.template.spec.containers[0].volumeMounts[-1]", docs[0]
         )
 
+    def test_job_ttl_after_finished(self):
+        docs = render_chart(
+            values={"migrateDatabaseJob": {"ttlSecondsAfterFinished": 1}},
+            show_only=["templates/jobs/migrate-database-job.yaml"],
+        )
+        ttl = jmespath.search("spec.ttlSecondsAfterFinished", docs[0])
+        assert ttl == 1
+
+    def test_job_ttl_after_finished_zero(self):
+        docs = render_chart(
+            values={"migrateDatabaseJob": {"ttlSecondsAfterFinished": 0}},
+            show_only=["templates/jobs/migrate-database-job.yaml"],
+        )
+        ttl = jmespath.search("spec.ttlSecondsAfterFinished", docs[0])
+        assert ttl == 0
+
+    def test_job_ttl_after_finished_nil(self):
+        docs = render_chart(show_only=["templates/jobs/migrate-database-job.yaml"])
+        spec = jmespath.search("spec", docs[0])
+        assert "ttlSecondsAfterFinished" not in spec
+
     @pytest.mark.parametrize(
         "airflow_version, expected_arg",
         [