@@ -6,15 +6,16 @@
 # answers only the calls kafka_run is expected to make -- get/wait for the
 # long-lived CLI Pod and an `exec` into it -- and runs the exec'd snippet locally
 # against the kafka-topics.sh stub. A throwaway `kubectl run -i` Pod (the old
-# attach-racing path) makes the stub exit non-zero, so any test that gets a
-# correct reply proves the reply came back through exec, not that path.
+# attach-racing path) makes the stub exit non-zero, so a test that gets the right
+# reply proves it came back through exec, not that path.
 #
-# The helpers are Bash and this runner is POSIX sh, so each case drives them
-# through `bash -c` under the same `set -euo pipefail` the demo scripts use. The
+# cozytest.sh's awk runner recognizes only @test blocks and a bare `}`; there is
+# no bats `run` or `$status`, and no setup/teardown, so each case calls
+# make_stubs itself and asserts with direct shell tests that exit non-zero. The
 # helper functions close on an indented brace because the runner rewrites a
 # column-0 `}` into `return 0`, which would mask run_topic_meta's exit status.
 
-setup() {
+make_stubs() {
     stub=$(mktemp -d)
     cat > "$stub/kubectl" <<'EOF'
 #!/bin/bash
@@ -60,16 +61,17 @@ run_topic_meta() {
     }
 
 @test "kafka_run reads the reply by exec into the long-lived CLI Pod" {
-    run run_topic_meta
-    [ "$status" -eq 0 ]
-    [ "$output" = "3 123456" ]
+    make_stubs
+    out=$(run_topic_meta)
+    [ "$out" = "3 123456" ]
     grep -q 'exec -i kafka-cli -- bash -c' "$stub/kubectl.argv"
 }
 
 @test "kafka_run starts the CLI Pod when it is absent, then execs into it" {
-    STUB_NO_POD=1 run run_topic_meta
-    [ "$status" -eq 0 ]
-    [ "$output" = "3 123456" ]
+    make_stubs
+    export STUB_NO_POD=1
+    out=$(run_topic_meta)
+    [ "$out" = "3 123456" ]
     grep -q 'run kafka-cli --image=' "$stub/kubectl.argv"
     grep -q 'exec -i kafka-cli -- bash -c' "$stub/kubectl.argv"
 }
@@ -5,16 +5,17 @@
 # kubectl and the `nats` CLI are stub executables on PATH. The kubectl stub
 # answers only the calls nats_cli is expected to make -- get/wait for the
 # long-lived CLI Pod and an `exec` into it -- and runs the exec'd `nats`
-# invocation locally against the nats stub. A throwaway `kubectl run -i` Pod
-# (the old attach-racing path) makes the stub exit non-zero, so any test that
-# gets a correct reply proves the reply came back through exec, not that path.
+# invocation locally against the nats stub. A throwaway `kubectl run -i` Pod (the
+# old attach-racing path) makes the stub exit non-zero, so a test that gets the
+# right reply proves it came back through exec, not that path.
 #
-# The helpers are Bash and this runner is POSIX sh, so each case drives them
-# through `bash -c` under the same `set -euo pipefail` the demo scripts use. The
+# cozytest.sh's awk runner recognizes only @test blocks and a bare `}`; there is
+# no bats `run` or `$status`, and no setup/teardown, so each case calls
+# make_stubs itself and asserts with direct shell tests that exit non-zero. The
 # helper functions close on an indented brace because the runner rewrites a
 # column-0 `}` into `return 0`, which would mask run_stream_count's exit status.
 
-setup() {
+make_stubs() {
     stub=$(mktemp -d)
     cat > "$stub/kubectl" <<'EOF'
 #!/bin/bash
@@ -60,16 +61,17 @@ run_stream_count() {
     }
 
 @test "nats_cli reads the reply by exec into the long-lived CLI Pod" {
-    run run_stream_count
-    [ "$status" -eq 0 ]
-    [ "$output" = "42" ]
+    make_stubs
+    out=$(run_stream_count)
+    [ "$out" = "42" ]
     grep -q 'exec -i nats-cli -- nats' "$stub/kubectl.argv"
 }
 
 @test "nats_cli starts the CLI Pod when it is absent, then execs into it" {
-    STUB_NO_POD=1 run run_stream_count
-    [ "$status" -eq 0 ]
-    [ "$output" = "42" ]
+    make_stubs
+    export STUB_NO_POD=1
+    out=$(run_stream_count)
+    [ "$out" = "42" ]
     grep -q 'run nats-cli --image=' "$stub/kubectl.argv"
     grep -q 'exec -i nats-cli -- nats' "$stub/kubectl.argv"
 }