packer-cn/common/bootcommand/boot_command_ast_test.go

package bootcommand

import (
	"fmt"
	"log"
	"strings"
	"testing"

	"github.com/stretchr/testify/assert"
)

func toIfaceSlice(v interface{}) []interface{} {
	if v == nil {
		return nil
	}
	return v.([]interface{})
}

func Test_parse(t *testing.T) {
	in := "<wait><wait20><wait3s><wait4m2ns>"
	in += "foo/bar > one 界"
	in += "<fon> b<fOff>"
	in += "<foo><f3><f12><spacebar><leftalt><rightshift><rightsuper>"
	expected := []string{
		"Wait<1s>",
		"Wait<20s>",
		"Wait<3s>",
		"Wait<4m0.000000002s>",
		"LIT-Press(f)",
		"LIT-Press(o)",
		"LIT-Press(o)",
		"LIT-Press(/)",
		"LIT-Press(b)",
		"LIT-Press(a)",
		"LIT-Press(r)",
		"LIT-Press( )",
		"LIT-Press(>)",
		"LIT-Press( )",
		"LIT-Press(o)",
		"LIT-Press(n)",
		"LIT-Press(e)",
		"LIT-Press( )",
		"LIT-Press(界)",
		"LIT-On(f)",
		"LIT-Press( )",
		"LIT-Press(b)",
		"LIT-Off(f)",
		"LIT-Press(<)",
		"LIT-Press(f)",
		"LIT-Press(o)",
		"LIT-Press(o)",
		"LIT-Press(>)",
		"Spec-Press(f3)",
		"Spec-Press(f12)",
		"Spec-Press(spacebar)",
		"Spec-Press(leftalt)",
		"Spec-Press(rightshift)",
		"Spec-Press(rightsuper)",
	}

	got, err := ParseReader("", strings.NewReader(in))
	if err != nil {
		log.Fatal(err)
	}
	gL := toIfaceSlice(got)
	for i, g := range gL {
		assert.Equal(t, expected[i], fmt.Sprintf("%s", g))
		log.Printf("%s\n", g)
	}
}

func Test_special(t *testing.T) {
	var specials = []struct {
		in  string
		out string
	}{
		{
			"<rightShift><rightshift><RIGHTSHIFT>",
			"Spec-Press(rightshift)",
		},
		{
			"<delon><delON><deLoN><DELON>",
			"Spec-On(del)",
		},
		{
			"<enteroff><enterOFF><eNtErOfF><ENTEROFF>",
			"Spec-Off(enter)",
		},
	}
	for _, tt := range specials {
		got, err := ParseReader("", strings.NewReader(tt.in))
		if err != nil {
			log.Fatal(err)
		}

		gL := toIfaceSlice(got)
		for _, g := range gL {
			assert.Equal(t, tt.out, g.(*specialExpression).String())
		}
	}
}

func Test_validation(t *testing.T) {
	var expressions = []struct {
		in    string
		valid bool
	}{
		{
			"<wait1m>",
			true,
		},
		{
			"<wait-1m>",
			false,
		},
		{
			"<f1>",
			true,
		},
		{
			"<",
			true,
		},
	}
	for _, tt := range expressions {
		got, err := ParseReader("", strings.NewReader(tt.in))
		if err != nil {
			log.Fatal(err)
		}

		gL := toIfaceSlice(got)
		assert.Len(t, gL, 1)
		err = gL[0].(expression).Validate()
		if tt.valid {
			assert.NoError(t, err)
		} else {
			assert.Error(t, err)
		}
	}
}