From 7e59b091182b2fc10c704009c68c5a4206ceef53 Mon Sep 17 00:00:00 2001
From: Sebastian Bernauer <sebastian.bernauer@stackable.tech>
Date: Thu, 15 Jan 2026 15:14:54 +0100
Subject: [PATCH] test: Improve assertion on restart-controller

---
 tests/templates/kuttl/smoke/40-assert.yaml.j2 | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/tests/templates/kuttl/smoke/40-assert.yaml.j2 b/tests/templates/kuttl/smoke/40-assert.yaml.j2
index b1d5eaa9..054f68fe 100644
--- a/tests/templates/kuttl/smoke/40-assert.yaml.j2
+++ b/tests/templates/kuttl/smoke/40-assert.yaml.j2
@@ -18,6 +18,8 @@ kind: StatefulSet
 metadata:
   name: airflow-webserver-default
   generation: 1 # There should be no unneeded Pod restarts
+  labels:
+    restarter.stackable.tech/enabled: "true"
 spec:
   template:
     spec:
@@ -32,6 +34,8 @@ kind: StatefulSet
 metadata:
   name: airflow-worker-default
   generation: 1 # There should be no unneeded Pod restarts
+  labels:
+    restarter.stackable.tech/enabled: "true"
 spec:
   template:
     spec:
@@ -46,6 +50,8 @@ kind: StatefulSet
 metadata:
   name: airflow-scheduler-default
   generation: 1 # There should be no unneeded Pod restarts
+  labels:
+    restarter.stackable.tech/enabled: "true"
 spec:
   template:
     spec:
@@ -59,6 +65,8 @@ kind: StatefulSet
 metadata:
   name: airflow-dagprocessor-default
   generation: 1 # There should be no unneeded Pod restarts
+  labels:
+    restarter.stackable.tech/enabled: "true"
 spec:
   template:
     spec:
@@ -72,6 +80,8 @@ kind: StatefulSet
 metadata:
   name: airflow-triggerer-default
   generation: 1 # There should be no unneeded Pod restarts
+  labels:
+    restarter.stackable.tech/enabled: "true"
 spec:
   template:
     spec: