summaryrefslogtreecommitdiff
path: root/parser/interpreter_test.go
blob: ee927a5e490f216812fcc6d9de00d8ebfa4edbfc (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
package parser_test

import (
	"fmt"
	"log"
	"testing"

	"github.com/gnolang/parscan/lang/golang"
	"github.com/gnolang/parscan/parser"
	"github.com/gnolang/parscan/scanner"
)

var GoScanner *scanner.Scanner

func init() {
	log.SetFlags(log.Lshortfile)
	GoScanner = scanner.NewScanner(golang.GoSpec)
}

func TestEval(t *testing.T) {
	for _, test := range goTests {
		test := test
		t.Run("", func(t *testing.T) {
			interp := parser.NewInterpreter(GoScanner)
			errStr := ""
			r, e := interp.Eval(test.src)
			t.Log(r, e)
			if e != nil {
				errStr = e.Error()
			}
			if errStr != test.err {
				t.Errorf("got error %#v, want error %#v", errStr, test.err)
			}
			if res := fmt.Sprintf("%v", r); test.err == "" && res != test.res {
				t.Errorf("got %#v, want %#v", res, test.res)
			}
		})
	}
}

var goTests = []struct {
	src, res, err string
}{
	0:  {src: "", res: "<nil>"},
	1:  {src: "1+2", res: "3"},
	2:  {src: "1+", err: "block not terminated"},
	3:  {src: "a := 1 + 2; b := 0; a + 1", res: "4"},
	4:  {src: "1+(2+3)", res: "6"},
	5:  {src: "(1+2)+3", res: "6"},
	6:  {src: "(6+(1+2)+3)+5", res: "17"},
	7:  {src: "(6+(1+2+3)+5", err: "1:1: block not terminated"},
	8:  {src: "a := 2; a = 3; a", res: "3"},
	9:  {src: "a := 0; if a == 0 { a = 2 } else { a = 1 }; a", res: "2"},
	10: {src: "a := 0; if a == 1 { a = 2 } else { a = 1 }; a", res: "1"},
	11: {src: "a := 0; if a == 1 { a = 2 } else if a == 0 { a = 3 } else { a = 1 }; a", res: "3"},
	12: {src: "a := 0; if a == 1 { a = 2 } else if a == 2 { a = 3 } else { a = 1 }; a", res: "1"},
	13: {src: "func f() int {return 2}; a := f(); a", res: "2"},
	14: {src: "func f() int {return 2}; f()", res: "2"},
	15: {src: "func f(a int) int {return a+2}; f(3)", res: "5"},
	16: {src: "func f(a int) int {if a < 4 {a = 5}; return a }; f(3)", res: "5"},
	17: {src: "func f(a int) int {return a+2}; 7 - f(3)", res: "2"},
	18: {src: "func f(a int) int {return a+2}; f(5) - f(3)", res: "2"},
	19: {src: "func f(a int) int {return a+2}; f(3) - 2", res: "3"},
	20: {src: "func f(a int, b int, c int) int {return a+b-c} ; f(7, 1, 3)", res: "5"},
	21: {src: "func f(a, b, c int) int {return a+b-c} ; f(7, 1, 3)", res: "5"},
}