run_test.go

package main

import (
	"io/ioutil"
	"os"
	"reflect"
	"runtime"
	"sync"
	"testing"
)

var runTests = []struct {
	args     []string
	in       string
	expected string
}{
	{
		args:     []string{"echo", "Hello,", "world!"},
		in:       ``,
		expected: `echo Hello, "world!"`,
	},
	{
		args: []string{"echo", "{{foo}},", "{{bar}}"},
		in: `Hello test
world test!
`,
		expected: `echo "Hello test," "world test!"`,
	},
	{
		args: []string{"echo", "{{foo-bar_baz}}", "{{FOO-9:BAR_2}}", "{{X}}"},
		in: `Foo bar
FOO BAR
X
`,
		expected: `echo "Foo bar" "FOO BAR" X`,
	},
	{
		args: []string{"echo", "{{foo}},", "{{bar}},", "{{foo}}-{{bar}}-{{baz}}"},
		in: `Hello
wonderful
world!
`,
		expected: `echo Hello, wonderful, "Hello-wonderful-world!"`,
	},
	{
		args: []string{"echo", "{{foo:bar}},", "{{foo:bar}},", "{{foo:baz}}"},
		in: `Hello
example world!

`,
		expected: `echo Hello, Hello, "example world!"`,
	},
	{
		args: []string{"echo", "{{foo:bar}}", "{{foo:baz}}", "{{foo:baz}}"},
		in: `Hello, world!
`,
		expected: `echo Hello "world!" "world!"`,
	},
	{
		args: []string{"echo", "{{foo:bar}}", "{{foo:bar}}", "{{foo:baz}}"},
		in: `Hello,\ world!, test,\ for,\ comma!
`,
		expected: `echo "Hello, world!" "Hello, world!" "test, for, comma!"`,
	},
	{
		args: []string{"echo", "[[foo:bar]]", "[[foo:bar]]", "[[foo:baz]]"},
		in: `Hello, world, oops!
Hello,
world?
`,
		expected: `echo Hello, Hello, "world?"`,
	},
	{
		args: []string{"echo", "{{foo}},", "[[bar]]", "{{baz}}"},
		in: `こんにちは
世界
+。:.ﾟ٩(๑>◡<๑)۶:.｡+ﾟ
`,
		expected: `echo こんにちは, 世界 "+。:.ﾟ٩(๑>◡<๑)۶:.｡+ﾟ"`,
	},
	{
		args: []string{"echo", "{{foo}}", "|", "echo", "||", "echo", "&&", "echo", ">", "/dev/null", "</dev/null", "2>&1", "1", ">&2", ">>", "foo", ">>/dev/null"},
		in: `Hello world!
`,
		expected: `echo "Hello world!" | echo || echo && echo > /dev/null </dev/null 2>&1 1 >&2 >> foo >>/dev/null`,
	},
	{
		args: []string{"echo", "{{foo}}", "{{bar}}"},
		in: `\'"${[]}|&;<>()*?!
	foo bar baz
`,
		expected: `echo "\\'\"${[]}|&;<>()*?!" "\tfoo bar baz"`,
	},
	{
		args:     []string{"echo", "{{foo}}"},
		in:       "\a\b\f'\r\t\v\\\n",
		expected: `echo "\a\b\f'\r\t\v\\"`,
	},
	{
		args: []string{"echo $(cat {{foo}} {{bar}})"},
		in: `sample1.txt
sample2.txt
`,
		expected: `echo $(cat sample1.txt sample2.txt)`,
	},
}

func TestRun(t *testing.T) {
	dir, err := ioutil.TempDir("", "fillin-")
	if err != nil {
		t.Fatal(err)
	}
	defer os.RemoveAll(dir)
	for _, test := range runTests {
		cmd, err := Run(dir, test.args, newTestPrompt(test.in))
		if err != nil {
			t.Errorf("error occurred unexpectedly: %+v", err)
		}
		if !reflect.DeepEqual(cmd, test.expected) {
			t.Errorf("command not correct (expected: %+v, got: %+v)", test.expected, cmd)
		}
	}
}

func TestRun_concurrently(t *testing.T) {
	if runtime.GOOS == "windows" {
		t.Skip("skip test on Windows")
	}
	dir, err := ioutil.TempDir("", "fillin-")
	if err != nil {
		t.Fatal(err)
	}
	defer os.RemoveAll(dir)
	test := runTests[1]
	var wg sync.WaitGroup
	for i := 0; i < 20; i++ {
		wg.Add(1)
		go func() {
			defer wg.Done()
			cmd, err := Run(dir, test.args, newTestPrompt(test.in))
			if err != nil {
				t.Errorf("error occurred unexpectedly: %+v", err)
			}
			if !reflect.DeepEqual(cmd, test.expected) {
				t.Errorf("command not correct (expected: %+v, got: %+v)", test.expected, cmd)
			}
		}()
	}
	wg.Wait()
}