Allow optional space in shebang

zunit-zsh · Jul 11, 2017 · 792f96d · 792f96d
1 parent d338dee
commit 792f96d
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/commands/run.zsh b/src/commands/run.zsh
@@ -409,7 +409,7 @@ function _zunit_parse_argument() {
     line=$(cat $argument | head -n 1)
 
     # Check for the zunit shebang
-    if [[ $line = "#!/usr/bin/env zunit" ]]; then
+    if [[ $line =~ "#! ?/usr/bin/env zunit" ]]; then
       # Add it to the array
       testfiles[(( ${#testfiles} + 1 ))]=("$argument${test_name+"@$test_name"}")
       return

diff --git a/tests/assertions/exists.zunit b/tests/assertions/exists.zunit
@@ -1,4 +1,4 @@
-#!/usr/bin/env zunit
+#! /usr/bin/env zunit
 
 @test 'Test _zunit_assert_exists success' {
   run assert './exists.zunit' exists