From dcd9904ce1736c47e587958b4e4e545284ab03e8 Mon Sep 17 00:00:00 2001
From: ttionya <git@ttionya.com>
Date: Wed, 23 Apr 2025 22:52:26 +0800
Subject: [PATCH] test: cleanup environment variable after every test

---
 tests/units/backup-7z-file/test.sh   | 10 ++++++++++
 tests/units/backup-cron/test.sh      | 16 +++++++++++++---
 tests/units/backup-unpackage/test.sh |  8 ++++++++
 tests/units/backup-zip-file/test.sh  | 10 ++++++++++
 4 files changed, 41 insertions(+), 3 deletions(-)

diff --git a/tests/units/backup-7z-file/test.sh b/tests/units/backup-7z-file/test.sh
index e8aab78..e6bbabb 100644
--- a/tests/units/backup-7z-file/test.sh
+++ b/tests/units/backup-7z-file/test.sh
@@ -48,8 +48,18 @@ function test() {
     fi
 }
 
+function cleanup() {
+    rm -rf "${TEST_OUTPUT_DIR}" "${TEST_EXTRACT_DIR}"
+
+    unset TEST_OUTPUT_DIR
+    unset TEST_EXTRACT_DIR
+    unset PASSWORD
+    unset BACKUP_FILE
+}
+
 prepare
 start
 test
+cleanup
 
 test_result "${TEST_NAME}" "${FAILED_NUM}"
diff --git a/tests/units/backup-cron/test.sh b/tests/units/backup-cron/test.sh
index 6cc05bc..0ae5574 100644
--- a/tests/units/backup-cron/test.sh
+++ b/tests/units/backup-cron/test.sh
@@ -43,9 +43,6 @@ function test() {
         ((TIMER++))
     done
 
-    # stop the container
-    docker stop "${TEST_CONTAINER_NAME}"
-
     ls -l "${BACKUP_FILE}"
 
     if [[ "${SUCCESS}" == "FALSE" ]]; then
@@ -53,8 +50,21 @@ function test() {
     fi
 }
 
+function cleanup() {
+    # stop the container
+    docker stop "${TEST_CONTAINER_NAME}"
+
+    rm -rf "${TEST_OUTPUT_DIR}"
+
+    unset TEST_CONTAINER_NAME
+    unset TEST_OUTPUT_DIR
+    unset PASSWORD
+    unset BACKUP_FILE
+}
+
 prepare
 start
 test
+cleanup
 
 test_result "${TEST_NAME}" "${FAILED_NUM}"
diff --git a/tests/units/backup-unpackage/test.sh b/tests/units/backup-unpackage/test.sh
index 948a8eb..c231cee 100644
--- a/tests/units/backup-unpackage/test.sh
+++ b/tests/units/backup-unpackage/test.sh
@@ -45,8 +45,16 @@ function test() {
     fi
 }
 
+function cleanup() {
+    rm -rf "${TEST_OUTPUT_DIR}" "${TEST_EXTRACT_DIR}"
+
+    unset TEST_OUTPUT_DIR
+    unset TEST_EXTRACT_DIR
+}
+
 prepare
 start
 test
+cleanup
 
 test_result "${TEST_NAME}" "${FAILED_NUM}"
diff --git a/tests/units/backup-zip-file/test.sh b/tests/units/backup-zip-file/test.sh
index 06d6fab..e2a69ec 100644
--- a/tests/units/backup-zip-file/test.sh
+++ b/tests/units/backup-zip-file/test.sh
@@ -47,8 +47,18 @@ function test() {
     fi
 }
 
+function cleanup() {
+    rm -rf "${TEST_OUTPUT_DIR}" "${TEST_EXTRACT_DIR}"
+
+    unset TEST_OUTPUT_DIR
+    unset TEST_EXTRACT_DIR
+    unset PASSWORD
+    unset BACKUP_FILE
+}
+
 prepare
 start
 test
+cleanup
 
 test_result "${TEST_NAME}" "${FAILED_NUM}"