ADD git-sync container lifecycle hooks

chart/templates/_helpers.yaml

@@ -309,6 +309,9 @@ If release name contains chart name it will be used as a full name.
   {{- if .Values.dags.gitSync.extraVolumeMounts }}
     {{- tpl (toYaml .Values.dags.gitSync.extraVolumeMounts) . | nindent 2 }}
   {{- end }}
+  {{- if and .Values.dags.gitSync.containerLifecycleHooks (not .is_init) }}
+  lifecycle: {{- tpl (toYaml .Values.dags.gitSync.containerLifecycleHooks) . | nindent 4 }}
+  {{- end }}
 {{- end }}
 
 {{/* This helper will change when customers deploy a new image */}}

chart/values.yaml

@@ -57,7 +57,7 @@ securityContexts:
   pod: {}
   containers: {}
 
-# Default container lifecycle hooks for every service except for redis, statsd and pgbouncer.
+# Global container lifecycle hooks for airflow containers
 containerLifecycleHooks: {}
 
 # Airflow home directory

helm_tests/other/test_git_sync_worker.py

@@ -168,3 +168,40 @@ def test_validate_if_ssh_params_are_added_with_git_ssh_key(self):
             "name": "git-sync-ssh-key",
             "secret": {"secretName": "release-name-ssh-secret", "defaultMode": 288},
         } in jmespath.search("spec.template.spec.volumes", docs[0])
+
+    def test_container_lifecycle_hooks(self):
+        docs = render_chart(
+            values={
+                "dags": {
+                    "gitSync": {
+                        "enabled": True,
+                        "containerLifecycleHooks": {
+                            "postStart": {
+                                "exec": {
+                                    "command": [
+                                        "/bin/sh",
+                                        "-c",
+                                        "echo postStart handler > /git/message_start"
+                                    ]
+                                }
+                            },
+                            "preStop": {
+                                "exec": {
+                                    "command": [
+                                        "/bin/sh",
+                                        "-c",
+                                        "echo preStop handler > /git/message_start"
+                                    ]
+                                }
+                            }
+                        }
+                    },
+                }
+            },
+            show_only=["templates/workers/worker-deployment.yaml"],
+        )
+        assert {
+            "postStart": {"exec": {"command": ["/bin/sh","-c","echo postStart handler > /git/message_start"]}},
+            "preStop": {"exec": {"command": ["/bin/sh","-c","echo preStop handler > /git/message_start"]}}
+       } == jmespath.search("spec.template.spec.containers[1].lifecycle", docs[0])
+