tests: fix image cleanup at the end of testing

test/01-samples.bats

@@ -66,6 +66,7 @@ function dq_rhel() {
     fi
     docker_build -t "${image}-${BATS_TEST_NAME}" .
     docker_run --rm --gpus 0 ${image}-${BATS_TEST_NAME}
+    docker_rmi ${image}-${BATS_TEST_NAME}
     [ "$status" -eq 0 ]
 }
 
@@ -91,6 +92,7 @@ function dq_rhel() {
     docker_build -t "${image}-${BATS_TEST_NAME}" .
     debug "docker_run --rm --gpus 0 ${image}-${BATS_TEST_NAME}"
     docker_run --rm --gpus 0 ${image}-${BATS_TEST_NAME}
+    docker_rmi ${image}-${BATS_TEST_NAME}
     [ "$status" -eq 0 ]
 }
 
@@ -115,5 +117,6 @@ function dq_rhel() {
     fi
     docker_build -t "${image}-${BATS_TEST_NAME}" .
     docker_run --rm --gpus 0 ${image}-${BATS_TEST_NAME}
+    docker_rmi ${image}-${BATS_TEST_NAME}
     [ "$status" -eq 0 ]
 }

test/helpers.bash

@@ -15,13 +15,18 @@ function check_runtime() {
     docker info | grep 'Runtimes:' | grep -q 'nvidia'
 }
 
-# Taken from runc tests
 function docker_run() {
     run docker run "$@"
     echo "docker run $@ (status=$status):" >&2
     echo "$output" >&2
 }
 
+function docker_rmi() {
+    run docker rmi "$@"
+    echo "docker rmi $@ (status=$status):" >&2
+    echo "$output" >&2
+}
+
 function docker_build() {
     run docker build --pull "$@"
     echo "docker run $@ (status=$status):" >&2

test/scripts/run_tests.sh

@@ -38,6 +38,3 @@ for test in $(find $test_path -iname "[0-9]*-*.bats" | sort); do
 done
 
 docker rmi -f ${image}
-# TODO: cleanup after pipeline
-# docker rmi -f rockylinux:9
-docker image prune -f --filter "dangling=true" || true