46 lines
833 B
Go
46 lines
833 B
Go
package evaluator
|
|
|
|
import (
|
|
"monkey/lexer"
|
|
"monkey/object"
|
|
"monkey/parser"
|
|
"testing"
|
|
)
|
|
|
|
func TestEvalIntegerExpression(t *testing.T) {
|
|
tests := []struct {
|
|
input string
|
|
expected int64
|
|
}{
|
|
{"5", 5},
|
|
{"10", 10},
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
evaluated := testEval(tt.input)
|
|
testIntegerObject(t, evaluated, tt.expected)
|
|
}
|
|
}
|
|
|
|
func testEval(input string) object.Object {
|
|
l := lexer.New(input)
|
|
p := parser.New(l)
|
|
program := p.ParseProgram()
|
|
|
|
return Eval(program)
|
|
}
|
|
|
|
func testIntegerObject(t *testing.T, obj object.Object, expected int64) bool {
|
|
result, ok := obj.(*object.Integer)
|
|
if !ok {
|
|
t.Errorf("object is not Integer. got=%T (%+v)", obj, obj)
|
|
return false
|
|
}
|
|
if result.Value != expected {
|
|
t.Errorf("object has wrong value. got=%d, want=%d", result.Value, expected)
|
|
return false
|
|
}
|
|
|
|
return true
|
|
}
|