From 3bdb355cc99553a225e84c5aa796395e1b0adeb2 Mon Sep 17 00:00:00 2001
From: Lars Francke <git@lars-francke.de>
Date: Wed, 2 Oct 2024 11:04:44 +0200
Subject: [PATCH] Fixes usage of custom images in the overrides test (#514)

---
 tests/templates/kuttl/overrides/10-install-airflow.yaml.j2  | 5 +++++
 tests/templates/kuttl/overrides/20-install-airflow2.yaml.j2 | 5 +++++
 2 files changed, 10 insertions(+)

diff --git a/tests/templates/kuttl/overrides/10-install-airflow.yaml.j2 b/tests/templates/kuttl/overrides/10-install-airflow.yaml.j2
index 2e7da7ff..8fe1d7f3 100644
--- a/tests/templates/kuttl/overrides/10-install-airflow.yaml.j2
+++ b/tests/templates/kuttl/overrides/10-install-airflow.yaml.j2
@@ -37,7 +37,12 @@ metadata:
   name: airflow-celery
 spec:
   image:
+{% if test_scenario['values']['airflow-latest'].find(",") > 0 %}
+    custom: "{{ test_scenario['values']['airflow-latest'].split(',')[1] }}"
+    productVersion: "{{ test_scenario['values']['airflow-latest'].split(',')[0] }}"
+{% else %}
     productVersion: "{{ test_scenario['values']['airflow-latest'] }}"
+{% endif %}
   clusterConfig:
     loadExamples: true
     exposeConfig: false
diff --git a/tests/templates/kuttl/overrides/20-install-airflow2.yaml.j2 b/tests/templates/kuttl/overrides/20-install-airflow2.yaml.j2
index bbaf47b4..c4a96ec0 100644
--- a/tests/templates/kuttl/overrides/20-install-airflow2.yaml.j2
+++ b/tests/templates/kuttl/overrides/20-install-airflow2.yaml.j2
@@ -5,7 +5,12 @@ metadata:
   name: airflow-kubernetes
 spec:
   image:
+{% if test_scenario['values']['airflow-latest'].find(",") > 0 %}
+    custom: "{{ test_scenario['values']['airflow-latest'].split(',')[1] }}"
+    productVersion: "{{ test_scenario['values']['airflow-latest'].split(',')[0] }}"
+{% else %}
     productVersion: "{{ test_scenario['values']['airflow-latest'] }}"
+{% endif %}
   clusterConfig:
     loadExamples: true
     exposeConfig: false