From 4440d0ebed4842f689e2b1f4cf88a066a0e7e2ff Mon Sep 17 00:00:00 2001
From: Razvan-Daniel Mihai <84674+razvan@users.noreply.github.com>
Date: Mon, 1 Sep 2025 11:23:22 +0200
Subject: [PATCH] test: fix spark connect client image

---
 .../kuttl/spark-connect/20-run-connect-client.yaml.j2  | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

diff --git a/tests/templates/kuttl/spark-connect/20-run-connect-client.yaml.j2 b/tests/templates/kuttl/spark-connect/20-run-connect-client.yaml.j2
index 89feba22..67842a27 100644
--- a/tests/templates/kuttl/spark-connect/20-run-connect-client.yaml.j2
+++ b/tests/templates/kuttl/spark-connect/20-run-connect-client.yaml.j2
@@ -57,11 +57,11 @@ spec:
       activeDeadlineSeconds: 600
       containers:
         - name: simple-connect-app
-{% if test_scenario['values']['spark-connect'].find(",") > 0 %}
-          image: "{{ test_scenario['values']['spark-connect'].split(',')[1] }}"
-{% else %}
-          image: oci.stackable.tech/sdp/spark-connect-client:{{ test_scenario['values']['spark-connect'] }}-stackable0.0.0-dev
-{% endif %}
+{% set spark_connect_client_version = test_scenario['values']['spark-connect'].split(',')[0] %}
+          # Extract the Spark Connect version and use it for the client.
+          # Using a separate dimension for the client doesn't work because beku would generate tests with
+          # version mismatches between client and server.
+          image: oci.stackable.tech/stackable/spark-connect-client:{{ spark_connect_client_version }}-stackable0.0.0-dev
           imagePullPolicy: IfNotPresent
           command:
             [