package lang

import (
	"reflect"
	"testing"
)

func TestLex(t *testing.T) {
	cases := []struct {
		Input  string
		Output []int
	}{
		{
			"foo",
			[]int{STRING, lexEOF},
		},

		{
			"foo$bar",
			[]int{STRING, lexEOF},
		},

		{
			"foo ${bar}",
			[]int{STRING, PROGRAM_BRACKET_LEFT, IDENTIFIER, PROGRAM_BRACKET_RIGHT, lexEOF},
		},

		{
			"foo $${bar}",
			[]int{STRING, lexEOF},
		},

		{
			"foo $$$${bar}",
			[]int{STRING, lexEOF},
		},

		{
			"foo ${\"bar\"}",
			[]int{STRING, PROGRAM_BRACKET_LEFT, STRING, PROGRAM_BRACKET_RIGHT, lexEOF},
		},

		{
			"${bar(baz)}",
			[]int{PROGRAM_BRACKET_LEFT,
				IDENTIFIER, PAREN_LEFT, IDENTIFIER, PAREN_RIGHT,
				PROGRAM_BRACKET_RIGHT, lexEOF},
		},

		{
			"${bar(baz, foo)}",
			[]int{PROGRAM_BRACKET_LEFT,
				IDENTIFIER, PAREN_LEFT,
				IDENTIFIER, COMMA, IDENTIFIER,
				PAREN_RIGHT,
				PROGRAM_BRACKET_RIGHT, lexEOF},
		},

		{
			"${bar(42)}",
			[]int{PROGRAM_BRACKET_LEFT,
				IDENTIFIER, PAREN_LEFT, INTEGER, PAREN_RIGHT,
				PROGRAM_BRACKET_RIGHT, lexEOF},
		},

		{
			"${bar(42+1)}",
			[]int{PROGRAM_BRACKET_LEFT,
				IDENTIFIER, PAREN_LEFT,
				INTEGER, ARITH_OP, INTEGER,
				PAREN_RIGHT,
				PROGRAM_BRACKET_RIGHT, lexEOF},
		},

		{
			"${bar(3.14159)}",
			[]int{PROGRAM_BRACKET_LEFT,
				IDENTIFIER, PAREN_LEFT, FLOAT, PAREN_RIGHT,
				PROGRAM_BRACKET_RIGHT, lexEOF},
		},

		{
			"${bar(inner(baz))}",
			[]int{PROGRAM_BRACKET_LEFT,
				IDENTIFIER, PAREN_LEFT,
				IDENTIFIER, PAREN_LEFT,
				IDENTIFIER,
				PAREN_RIGHT, PAREN_RIGHT,
				PROGRAM_BRACKET_RIGHT, lexEOF},
		},

		{
			"foo ${foo.bar.baz}",
			[]int{STRING, PROGRAM_BRACKET_LEFT, IDENTIFIER, PROGRAM_BRACKET_RIGHT, lexEOF},
		},

		{
			"foo ${foo.bar.*.baz}",
			[]int{STRING, PROGRAM_BRACKET_LEFT, IDENTIFIER, PROGRAM_BRACKET_RIGHT, lexEOF},
		},

		{
			"foo ${foo(\"baz\")}",
			[]int{STRING, PROGRAM_BRACKET_LEFT,
				IDENTIFIER, PAREN_LEFT, STRING, PAREN_RIGHT,
				PROGRAM_BRACKET_RIGHT, lexEOF},
		},

		{
			`foo ${"${var.foo}"}`,
			[]int{STRING, PROGRAM_BRACKET_LEFT,
				PROGRAM_BRACKET_LEFT, IDENTIFIER, PROGRAM_BRACKET_RIGHT,
				PROGRAM_BRACKET_RIGHT, lexEOF},
		},
	}

	for _, tc := range cases {
		l := &parserLex{Input: tc.Input}
		var actual []int
		for {
			token := l.Lex(new(parserSymType))
			actual = append(actual, token)

			if token == lexEOF {
				break
			}

			// Be careful against what are probably infinite loops
			if len(actual) > 100 {
				t.Fatalf("Input:%s\n\nExausted.", tc.Input)
			}
		}

		if !reflect.DeepEqual(actual, tc.Output) {
			t.Fatalf(
				"Input: %s\n\nBad: %#v\n\nExpected: %#v",
				tc.Input, actual, tc.Output)
		}
	}
}