1 files changed, 6 insertions, 16 deletions
diff --git a/test/kpod_diff.bats b/test/kpod_diff.bats
index 826fc7389..d73f462f6 100644
--- a/test/kpod_diff.bats
+++ b/test/kpod_diff.bats
@@ -2,20 +2,16 @@
 
 load helpers
 
-IMAGE="alpine:latest"
+function setup() {
+    copy_images
+}
 
 function teardown() {
     cleanup_test
 }
 
 @test "test diff of image and parent" {
-    run bash -c ${KPOD_BINARY} $KPOD_OPTIONS pull $IMAGE
-    echo "$output"
-    [ "$status" -eq 0 ]
-    run bash -c ${KPOD_BINARY} $KPOD_OPTIONS diff $IMAGE
-    echo "$output"
-    [ "$status" -eq 0 ]
-    run bash -c ${KPOD_BINARY} $KPOD_OPTIONS rmi $IMAGE
+    run ${KPOD_BINARY} $KPOD_OPTIONS diff $BB
     echo "$output"
     [ "$status" -eq 0 ]
 }
@@ -27,14 +23,8 @@ function teardown() {
 }
 
 @test "test diff with json output" {
-    run bash -c ${KPOD_BINARY} $KPOD_OPTIONS pull $IMAGE
-    echo "$output"
-    [ "$status" -eq 0 ]
-    # run bash -c bash -c "${KPOD_BINARY} ${KPOD_OPTIONS} diff --format json $IMAGE | python -m json.tool"
-    run bash -c ${KPOD_BINARY} $KPOD_OPTIONS diff --format json $IMAGE
-    echo "$output"
-    [ "$status" -eq 0 ]
-    run bash -c ${KPOD_BINARY} $KPOD_OPTIONS rmi $IMAGE
+    # run bash -c "${KPOD_BINARY} ${KPOD_OPTIONS} diff --format json $IMAGE | python -m json.tool"
+    run ${KPOD_BINARY} $KPOD_OPTIONS diff --format json $BB
     echo "$output"
     [ "$status" -eq 0 ]
 }