File: lex_test.go

package info (click to toggle)
golang-sourcehut-sircmpwn-go-bare 0.0~git20210406.ab86bc2-2
  • links: PTS, VCS
  • area: main
  • in suites: sid, trixie
  • size: 268 kB
  • sloc: makefile: 5
file content (152 lines) | stat: -rw-r--r-- 4,352 bytes parent folder | download
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
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
package schema

import (
	"io"
	"strings"
	"testing"

	"github.com/stretchr/testify/assert"
)

func TestScanWords(t *testing.T) {
	cases := map[string]TokenKind{
		"uint": TUINT,
		"u8": TU8,
		"u16": TU16,
		"u32": TU32,
		"u64": TU64,
		"int": TINT,
		"i8": TI8,
		"i16": TI16,
		"i32": TI32,
		"i64": TI64,
		"f32": TF32,
		"f64": TF64,
		"bool": TBOOL,
		"string": TSTRING,
		"data": TDATA,
		"void": TVOID,
		"map": TMAP,
		"optional": TOPTIONAL,
	}

	for input, reference := range cases {
		scanner := NewScanner(strings.NewReader(input))
		tok, err := scanner.Next()
		assert.NoError(t, err, "Expected Scan to return without error")
		assert.Empty(t, tok.Value, "Expected Scan to return no value")
		assert.Equal(t, reference, tok.Token,
			"Expected Scan to return reference value for %s", input)
		_, err = scanner.Next()
		assert.Equal(t, io.EOF, err, "Expected Scan to return EOF")
	}

	scanner := NewScanner(strings.NewReader("hello"))
	tok, err := scanner.Next()
	assert.NoError(t, err, "Expected Scan to return without error")
	assert.Equal(t, tok.Value, "hello", "Expected Scan to return value 'hello'")
	assert.Equal(t, TNAME, tok.Token, "Expected Scan to return TNAME")
	_, err = scanner.Next()
	assert.Equal(t, io.EOF, err, "Expected Scan to return EOF")
}

func TestScanInteger(t *testing.T) {
	scanner := NewScanner(strings.NewReader("12345"))
	tok, err := scanner.Next()
	assert.NoError(t, err, "Expected Scan to return without error")
	assert.Equal(t, tok.Value, "12345", "Expected Scan to return value '12345'")
	assert.Equal(t, TINTEGER, tok.Token, "Expected Scan to return TINTEGER")
	_, err = scanner.Next()
	assert.Equal(t, io.EOF, err, "Expected Scan to return EOF")
}

func TestScanSymbols(t *testing.T) {
	cases := map[string]TokenKind{
		"<": TLANGLE,
		">": TRANGLE,
		"{": TLBRACE,
		"}": TRBRACE,
		"[": TLBRACKET,
		"]": TRBRACKET,
		"(": TLPAREN,
		")": TRPAREN,
	}

	for input, reference := range cases {
		scanner := NewScanner(strings.NewReader(input))
		tok, err := scanner.Next()
		assert.NoError(t, err, "Expected Scan to return without error")
		assert.Empty(t, tok.Value, "Expected Scan to return no value")
		assert.Equal(t, reference, tok.Token,
			"Expected Scan to return reference value for %s", input)
		_, err = scanner.Next()
		assert.Equal(t, io.EOF, err, "Expected Scan to return EOF")
	}
}

func TestScanSample(t *testing.T) {
	sample := `
	type PublicKey data<128>
	type Time string # ISO T8601

	enum Department u8 {
		ACCOUNTING
		ADMINISTRATION
		CUSTOMER_SERVICE
		DEVELOPMENT

		# Reserved for the CEO
		JSMITH = 99
	}

	type Customer {
		name: string
		email: string
		address: Address
		orders: []{
			orderId: i64
			quantity: i32
		}
		metadata: map[string]data
	}

	type Person (Customer | Employee)`
	reference := []Token{
		{TTYPE, ""}, {TNAME, "PublicKey"}, {TDATA, ""},
			{TLANGLE, ""}, {TINTEGER, "128"}, {TRANGLE, ""},
		{TTYPE, ""}, {TNAME, "Time"}, {TSTRING, ""},
		{TENUM, ""}, {TNAME, "Department"}, {TU8, ""}, {TLBRACE, ""},
			{TNAME, "ACCOUNTING"},
			{TNAME, "ADMINISTRATION"},
			{TNAME, "CUSTOMER_SERVICE"},
			{TNAME, "DEVELOPMENT"},
			{TNAME, "JSMITH"}, {TEQUAL, ""}, {TINTEGER, "99"},
		{TRBRACE, ""},
		{TTYPE, ""}, {TNAME, "Customer"}, {TLBRACE, ""},
		{TNAME, "name"}, {TCOLON, ""}, {TSTRING, ""},
		{TNAME, "email"}, {TCOLON, ""}, {TSTRING, ""},
		{TNAME, "address"}, {TCOLON, ""}, {TNAME, "Address"},
		{TNAME, "orders"}, {TCOLON, ""}, {TLBRACKET, ""}, {TRBRACKET, ""}, {TLBRACE, ""},
			{TNAME, "orderId"}, {TCOLON, ""}, {TI64, ""},
			{TNAME, "quantity"}, {TCOLON, ""}, {TI32, ""},
		{TRBRACE, ""},
		{TNAME, "metadata"}, {TCOLON, ""},
			{TMAP, ""}, {TLBRACKET, ""}, {TSTRING, ""}, {TRBRACKET, ""},
				{TDATA, ""},
		{TRBRACE, ""},
		{TTYPE, ""}, {TNAME, "Person"},
			{TLPAREN, ""}, {TNAME, "Customer"},
			{TPIPE, ""}, {TNAME, "Employee"},
			{TRPAREN, ""},
	}
	scanner := NewScanner(strings.NewReader(sample))
	for i, ref := range reference {
		tok, err := scanner.Next()
		assert.NoError(t, err, "Expected Scan to return without error for reference %d", i)
		assert.Equal(t, ref.Token, tok.Token, "Expected Scan to return correct token for reference %d", i)
		assert.Equal(t, ref.Value, tok.Value, "Expected Scan to return correct value for reference %d", i)
	}

	_, err := scanner.Next()
	assert.Equal(t, io.EOF, err, "Expected Scan to return EOF")
}