1 files changed, 22 insertions, 0 deletions
diff --git a/runner/shell/shell_test.go b/runner/shell/shell_test.go
new file mode 100644
index 0000000..b28bd6c
--- /dev/null
+++ b/runner/shell/shell_test.go
@@ -0,0 +1,22 @@
+package shell
+
+import (
+	"github.com/stretchr/testify/assert"
+	"testing"
+)
+
+func TestShell_ShellByFilename(t *testing.T) {
+	s := &shell{
+		Locations: map[string]string{"powershell": "/pws", "bash": "/bash"},
+	}
+	inputs := map[string]string{
+		"script.ps1": "/pws",
+		"script.sh":  "/bash",
+		"asdf":       "",
+	}
+
+	for k, v := range inputs {
+		sh := s.ShellByFilename(k)
+		assert.Equal(t, sh, v)
+	}
+}