package lexer import ( "bytes" . "testing" "github.com/stretchr/testify/assert" "github.com/stretchr/testify/require" ) var lexTestSrc = ` // this is a comment // // this is also a comment a anIdentifier 1 100 1.5 1.5e9 /* some stuff */ /* this should actually work */ /*/ /* nested! /* wtf this is crazy */ */ (punctuation,is{cool}> ) -tab "this is a string", "and so is this one" "\"foo" "bar\"baz\"" "buz\0" ` func TestLex(t *T) { l := New(bytes.NewBufferString(lexTestSrc)) assertNext := func(typ TokenType, val string, row, col int) { t.Logf("asserting %s %q [row:%d col:%d]", typ, val, row, col) require.True(t, l.HasNext()) tok := l.Next() assert.Equal(t, typ, tok.TokenType) assert.Equal(t, val, tok.Val) assert.Equal(t, row, tok.Row) assert.Equal(t, col, tok.Col) } assertNext(Identifier, "a", 3, 2) assertNext(Identifier, "anIdentifier", 4, 2) assertNext(Identifier, "1", 5, 2) assertNext(Identifier, "100", 6, 2) assertNext(Identifier, "1.5", 7, 2) assertNext(Identifier, "1.5e9", 8, 2) assertNext(Wrapper, "(", 24, 2) assertNext(Identifier, "punctuation", 24, 3) assertNext(Punctuation, ",", 24, 14) assertNext(Identifier, "is", 24, 15) assertNext(Wrapper, "{", 24, 17) assertNext(Identifier, "cool", 24, 18) assertNext(Wrapper, "}", 24, 22) assertNext(Punctuation, ">", 24, 23) assertNext(Wrapper, ")", 24, 25) assertNext(Identifier, "-tab", 25, 2) assertNext(String, `"this is a string"`, 27, 2) assertNext(Punctuation, ",", 27, 20) assertNext(String, `"and so is this one"`, 27, 22) assertNext(String, `"\"foo"`, 28, 2) assertNext(String, `"bar\"baz\""`, 29, 2) assertNext(String, `"buz\0"`, 30, 2) assertNext(EOF, "EOF", 31, 0) assert.False(t, l.HasNext()) }