argv_test.go

package argv

import (
	"reflect"
	"testing"
)

func TestArgv(t *testing.T) {
	type testCase struct {
		Input    string
		Sections [][]string
		Error    error
	}
	cases := []testCase{
		{
			Input: " a | a|a |a`ls ~/``ls /` ",
			Sections: [][]string{
				{"a"},
				{"a"},
				{"a"},
				{"als ~/ls /"},
			},
		},
		{
			Input: "aaa |",
			Error: ErrInvalidSyntax,
		},
		{
			Input: "aaa | | aa",
			Error: ErrInvalidSyntax,
		},
		{
			Input: " | aa",
			Error: ErrInvalidSyntax,
		},
		{
			Input: `aa"aaa`,
			Error: ErrInvalidSyntax,
		},
	}
	for i, c := range cases {
		gots, err := Argv(c.Input, func(s string) (string, error) {
			return s, nil
		}, nil)
		if err != c.Error {
			t.Errorf("test failed: %d, expect error:%s, but got %s", i, c.Error, err)
		}
		if err != nil {
			continue
		}

		if !reflect.DeepEqual(gots, c.Sections) {
			t.Errorf("parse failed %d, expect: %v, got %v", i, c.Sections, gots)
		}
	}
}

func TestArgv2(t *testing.T) {
	args, err := Argv(" ls   `echo /`   |  wc  -l ", func(backquoted string) (string, error) {
		return backquoted, nil
	}, nil)
	if err != nil {
		t.Fatal(err)
	}
	expects := [][]string{
		[]string{"ls", "echo /"},
		[]string{"wc", "-l"},
	}
	if !reflect.DeepEqual(args, expects) {
		t.Fatal(args)
	}
}