lexer test pass
This commit is contained in:
parent
7d8967403d
commit
d93f87c889
@ -1 +1,61 @@
|
|||||||
package lexer
|
package lexer
|
||||||
|
|
||||||
|
import (
|
||||||
|
"gitlab.com/Tyrel/monkey/token"
|
||||||
|
)
|
||||||
|
|
||||||
|
type Lexer struct {
|
||||||
|
input string
|
||||||
|
position int
|
||||||
|
readPosition int
|
||||||
|
ch byte
|
||||||
|
}
|
||||||
|
|
||||||
|
func New(input string) *Lexer {
|
||||||
|
l := &Lexer{input: input}
|
||||||
|
l.readChar()
|
||||||
|
return l
|
||||||
|
}
|
||||||
|
|
||||||
|
func (l *Lexer) readChar() {
|
||||||
|
if l.readPosition >= len(l.input) {
|
||||||
|
l.ch = 0
|
||||||
|
} else {
|
||||||
|
l.ch = l.input[l.readPosition]
|
||||||
|
}
|
||||||
|
l.position = l.readPosition
|
||||||
|
l.readPosition += 1
|
||||||
|
}
|
||||||
|
|
||||||
|
func (l *Lexer) NextToken() token.Token {
|
||||||
|
var tok token.Token
|
||||||
|
|
||||||
|
switch l.ch {
|
||||||
|
case '=':
|
||||||
|
tok = newToken(token.ASSIGN, l.ch)
|
||||||
|
case '+':
|
||||||
|
tok = newToken(token.PLUS, l.ch)
|
||||||
|
case '(':
|
||||||
|
tok = newToken(token.LPAREN, l.ch)
|
||||||
|
case ')':
|
||||||
|
tok = newToken(token.RPAREN, l.ch)
|
||||||
|
case '{':
|
||||||
|
tok = newToken(token.LBRACE, l.ch)
|
||||||
|
case '}':
|
||||||
|
tok = newToken(token.RBRACE, l.ch)
|
||||||
|
case ',':
|
||||||
|
tok = newToken(token.COMMA, l.ch)
|
||||||
|
case ';':
|
||||||
|
tok = newToken(token.SEMICOLON, l.ch)
|
||||||
|
case 0:
|
||||||
|
tok.Literal = ""
|
||||||
|
tok.Type = token.EOF
|
||||||
|
}
|
||||||
|
|
||||||
|
l.readChar()
|
||||||
|
return tok
|
||||||
|
}
|
||||||
|
|
||||||
|
func newToken(tokenType token.TokenType, ch byte) token.Token {
|
||||||
|
return token.Token{Type: tokenType, Literal: string(ch)}
|
||||||
|
}
|
||||||
|
@ -1,38 +1,40 @@
|
|||||||
package lexer
|
package lexer
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"testing"
|
"testing"
|
||||||
"monkey/token"
|
|
||||||
|
"gitlab.com/Tyrel/monkey/token"
|
||||||
)
|
)
|
||||||
|
|
||||||
func TestNextToken(t *testing.T) {
|
func TestNextToken(t *testing.T) {
|
||||||
input := `=+(){},;`
|
input := `=+(){},;`
|
||||||
|
|
||||||
tests := []struct{
|
|
||||||
expectedType token.TokenType
|
|
||||||
expectedLiteral string
|
|
||||||
}{
|
|
||||||
{token.ASSIGN, "="},
|
|
||||||
{token.PLUS, "+"},
|
|
||||||
{token.LPAREN, "("},
|
|
||||||
{token.RPAREN, ")"},
|
|
||||||
{token.LBRACE, "{"},
|
|
||||||
{token.RBRACE, "}"},
|
|
||||||
{token.COMMA, ","},
|
|
||||||
{token.SEMICOLON, ";"},
|
|
||||||
{token.EOF, ""},
|
|
||||||
}
|
|
||||||
|
|
||||||
l := New(input)
|
tests := []struct {
|
||||||
for i, tt := range tests {
|
expectedType token.TokenType
|
||||||
tok := l.NextToken()
|
expectedLiteral string
|
||||||
|
}{
|
||||||
|
{token.ASSIGN, "="},
|
||||||
|
{token.PLUS, "+"},
|
||||||
|
{token.LPAREN, "("},
|
||||||
|
{token.RPAREN, ")"},
|
||||||
|
{token.LBRACE, "{"},
|
||||||
|
{token.RBRACE, "}"},
|
||||||
|
{token.COMMA, ","},
|
||||||
|
{token.SEMICOLON, ";"},
|
||||||
|
{token.EOF, ""},
|
||||||
|
}
|
||||||
|
|
||||||
if tok.Type != tt.expectedType {
|
l := New(input)
|
||||||
t.Fatalf("tests[%d] - tokentype wrong. expected=%q, got %q",
|
for i, tt := range tests {
|
||||||
i, tt.expectedType, tok.Type)
|
tok := l.NextToken()
|
||||||
}
|
|
||||||
if tok.Literal != tt.expectedLiteral {
|
if tok.Type != tt.expectedType {
|
||||||
t.Fatalf("tests[%d] - literal wrong. expected=%q, got %q",
|
t.Fatalf("tests[%d] - tokentype wrong. expected=%q, got %q",
|
||||||
i, tt.expectedLiteral, tok.Literal)
|
i, tt.expectedType, tok.Type)
|
||||||
}
|
}
|
||||||
}
|
if tok.Literal != tt.expectedLiteral {
|
||||||
|
t.Fatalf("tests[%d] - literal wrong. expected=%q, got %q",
|
||||||
|
i, tt.expectedLiteral, tok.Literal)
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
Loading…
Reference in New Issue
Block a user