feat:added tests for pre-exit hook

hooks/pre-exit

@@ -3,13 +3,11 @@ set -euo pipefail
 
 # Allow to pass TMP_DIR for testing purposes
 if [[ -z "${TMP_DIR:-}" ]]; then
-
   TMP_DIR="${BUILDKITE_PLUGIN_LAMBDATEST_TMP_DIR:-}"
   function cleanup {
-
-    echo "Removing ${TMP_DIR}"
+    echo "Removing TMP_DIR ${TMP_DIR}"
     if [[ -d "${TMP_DIR}" ]]; then
-        rm -rf "${TMP_DIR}"
+      rm -rf "${TMP_DIR}"
     fi
   }
 else
@@ -25,9 +23,9 @@ is_pid_alive() {
 }
 
 stop_tunnel() {
-
   echo "Stopping LambdaTest Tunnel"
   pushd "${TMP_DIR}" >/dev/null
+
   local find_pidfiles
   find_pidfiles=$(find . -maxdepth 1 -name 'pid.*' -type f)
 
@@ -37,21 +35,21 @@ stop_tunnel() {
       local pid
       pid=$(cat "${pidfile}")
       kill -SIGINT "${pid}" || true
+
       local counter=0
       while is_pid_alive "${pid}"; do
         sleep 1
-        counter=$((counter+1))
+        counter=$((counter + 1))
         if [[ "${counter}" -gt 60 ]]; then
           # try to kill again after a minute
           kill -SIGINT "${pid}" 2>/dev/null || true
         fi
-        echo "Tunnel Successfully Stopped"
       done
-      echo
+      echo "Tunnel Successfully Stopped"
     fi
   done <<< "${find_pidfiles}"
+
   popd >/dev/null
-
 }
 
 stop_tunnel

tests/pre-exit.bats

@@ -0,0 +1,36 @@
+#!/usr/bin/env bats
+
+load "$BATS_PLUGIN_PATH/load.bash"
+
+@test "Ensure TMP_DIR is not removed if it existed before" {
+  export TMP_DIR=$(mktemp -d)
+
+  run "${PWD}/hooks/pre-exit"
+
+  assert_success
+  assert_output --partial "Not removing given TMP_DIR (${TMP_DIR})"
+
+  rm -rf "$TMP_DIR"
+  unset TMP_DIR
+}
+
+@test "Ensure TMP_DIR is removed if didnt exist" {
+  run "${PWD}/hooks/pre-exit"
+  assert_success
+  assert_output --partial "Removing TMP_DIR"
+}
+
+@test "Ensure tunnel is stopped" {
+  export TMP_DIR=$(mktemp -d)
+  touch "${TMP_DIR}/pid.$$" 
+
+  run "${PWD}/hooks/pre-exit"
+
+  assert_success
+  assert_output --partial "Stopping LambdaTest Tunnel"
+  assert_output --partial "Tunnel Successfully Stopped"
+
+
+  unset TMP_DIR
+  rm -rf "$TMP_DIR"
+}