Merge pull request #94 from TypedDevs/fix/unset-setup-functions

Unset setup functions

src/assert.sh

@@ -129,7 +129,7 @@ function assertSuccessfulCode() {
   local expected_exit_code=0
   local label="${3:-$(normalizeTestFunctionName "${FUNCNAME[1]}")}"
 
-  if [ $actual_exit_code -ne "$expected_exit_code" ]; then
+  if [[ $actual_exit_code != "$expected_exit_code" ]]; then
     ((_ASSERTIONS_FAILED++))
     printFailedTest  "${label}" "${actual_exit_code}" "to be exactly" "${expected_exit_code}"
     return

src/helpers.sh

@@ -59,3 +59,14 @@ function executeFunctionIfExists() {
     "$function_name"
   fi
 }
+
+function unsetIfExists() {
+  local function_name=$1
+
+  if declare -F | awk '{print $3}' | grep -Eq "^${function_name}$"; then
+    unset "$function_name"
+    return 0
+  fi
+
+  return 1
+}

src/runner.sh

@@ -57,6 +57,16 @@ function runTearDownAfterScript() {
   executeFunctionIfExists 'tearDownAfterScript'
 }
 
+function cleanSetUpAndTearDownAfterTest() {
+  unsetIfExists 'setUp'
+  unsetIfExists 'tearDown'
+}
+
+function cleanSetUpAndTearDownAfterScript() {
+  unsetIfExists 'setUpBeforeScript'
+  unsetIfExists 'tearDownAfterScript'
+}
+
 ###############
 #### MAIN #####
 ###############
@@ -101,6 +111,8 @@ function loadTestFiles() {
       wait
     fi
     runTearDownAfterScript
+
+    cleanSetUpAndTearDownAfterScript
   done
 }
 

tests/unit/helpers_test.sh

@@ -55,3 +55,20 @@ function test_no_function_is_executed_with_execute_function_if_exists() {
 
   assertEmpty "$(executeFunctionIfExists "$function_name")"
 }
+
+function test_unsuccessful_unsetIfExists() {
+  assertGeneralError "$(unsetIfExists "fake_function")"
+}
+
+function test_successful_unsetIfExists() {
+  # shellcheck disable=SC2317
+  function fake_function() {
+    return 0
+  }
+
+  assertSuccessfulCode "$(unsetIfExists "fake_function")"
+}
+
+function tearDown() {
+  unset fake_function
+}