add unit test for stdin and stdin err

Author: dylanhitt

pkg/app/test_command.go

@@ -64,6 +64,15 @@ func TestCommand(testPath string, ctx TestCommandContext) error {
 	return nil
 }
 
+func testFile(filePath string, fileName string, filters runtime.Filters) (runtime.Result, error) {
+	s, err := readFile(filePath, fileName)
+	if err != nil {
+		return runtime.Result{}, fmt.Errorf("Error " + err.Error())
+	}
+
+	return execute(s, filters)
+}
+
 func testDir(directory string, filters runtime.Filters) (runtime.Result, error) {
 	result := runtime.Result{}
 	files, err := ioutil.ReadDir(directory)
@@ -117,15 +126,6 @@ func convergeResults(result runtime.Result, new runtime.Result) runtime.Result {
 	return result
 }
 
-func testFile(filePath string, fileName string, filters runtime.Filters) (runtime.Result, error) {
-	s, err := readFile(filePath, fileName)
-	if err != nil {
-		return runtime.Result{}, fmt.Errorf("Error " + err.Error())
-	}
-
-	return execute(s, filters)
-}
-
 func testStdin(filters runtime.Filters) (runtime.Result, error) {
 	f, err := os.Stdin.Stat()
 	if err != nil {

pkg/app/test_command_test.go

@@ -3,6 +3,7 @@ package app
 import (
 	"bytes"
 	"io"
+	"io/ioutil"
 	"log"
 	"os"
 	"runtime"
@@ -108,6 +109,58 @@ func Test_TestCommand_Http_Err(t *testing.T) {
 	}
 }
 
+func Test_TestCommand_StdIn(t *testing.T) {
+	tmpfile, err := ioutil.TempFile("", "test")
+	if err != nil {
+		log.Fatal(err)
+	}
+	defer os.Remove(tmpfile.Name()) //clean up
+
+	content := []byte(`
+tests:
+   echo hello:
+     exit-code: 0
+`)
+
+	if _, err := tmpfile.Write(content); err != nil {
+		log.Fatal(err)
+	}
+
+	if _, err := tmpfile.Seek(0, 0); err != nil {
+		log.Fatal(err)
+	}
+
+	// set stdin to tempfile
+	oldStdin := os.Stdin
+	defer func() { os.Stdin = oldStdin }() // Restore original Stdin
+	os.Stdin = tmpfile
+
+	out := captureOutput(func() {
+		TestCommand("-", TestCommandContext{Verbose: false})
+	})
+
+	if err := tmpfile.Close(); err != nil {
+		log.Fatal(err)
+	}
+
+	assert.Contains(t, out, "✓ [local] echo hello")
+}
+
+func Test_TestCommand_StdIn_Err(t *testing.T) {
+	// set stdin to nil
+	oldStdin := os.Stdin
+	defer func() { os.Stdin = oldStdin }() // Restore original Stdin
+	os.Stdin = nil
+
+	err := TestCommand("-", TestCommandContext{Verbose: false})
+
+	if runtime.GOOS == "windows" {
+		assert.NotNil(t, err)
+	} else {
+		assert.NotNil(t, err)
+	}
+}
+
 func Test_ConvergeResults(t *testing.T) {
 	duration, _ := time.ParseDuration("5s")