147 lines
2.5 KiB
Go
147 lines
2.5 KiB
Go
|
package gg
|
||
|
|
||
|
import (
|
||
|
"strconv"
|
||
|
"testing"
|
||
|
|
||
|
"github.com/stretchr/testify/assert"
|
||
|
)
|
||
|
|
||
|
func TestDecoder(t *testing.T) {
|
||
|
|
||
|
i := func(i int64) Value {
|
||
|
return Value{Number: &i}
|
||
|
}
|
||
|
|
||
|
n := func(n string) Value {
|
||
|
return Value{Name: &n}
|
||
|
}
|
||
|
|
||
|
tests := []struct {
|
||
|
in string
|
||
|
exp *Graph
|
||
|
}{
|
||
|
{
|
||
|
in: "",
|
||
|
exp: ZeroGraph,
|
||
|
},
|
||
|
{
|
||
|
in: "out = 1;",
|
||
|
exp: ZeroGraph.AddValueIn(ValueOut(i(1), Value{}), n("out")),
|
||
|
},
|
||
|
{
|
||
|
in: "out = incr < 1;",
|
||
|
exp: ZeroGraph.AddValueIn(ValueOut(i(1), n("incr")), n("out")),
|
||
|
},
|
||
|
{
|
||
|
in: "out = a < b < 1;",
|
||
|
exp: ZeroGraph.AddValueIn(
|
||
|
TupleOut(
|
||
|
[]OpenEdge{ValueOut(i(1), n("b"))},
|
||
|
n("a"),
|
||
|
),
|
||
|
n("out"),
|
||
|
),
|
||
|
},
|
||
|
{
|
||
|
in: "out = a < b < (1; c < 2; d < e < 3;);",
|
||
|
exp: ZeroGraph.AddValueIn(
|
||
|
TupleOut(
|
||
|
[]OpenEdge{TupleOut(
|
||
|
[]OpenEdge{
|
||
|
ValueOut(i(1), Value{}),
|
||
|
ValueOut(i(2), n("c")),
|
||
|
TupleOut(
|
||
|
[]OpenEdge{ValueOut(i(3), n("e"))},
|
||
|
n("d"),
|
||
|
),
|
||
|
},
|
||
|
n("b"),
|
||
|
)},
|
||
|
n("a"),
|
||
|
),
|
||
|
n("out"),
|
||
|
),
|
||
|
},
|
||
|
{
|
||
|
in: "out = a < b < (1; c < (d < 2; 3;); );",
|
||
|
exp: ZeroGraph.AddValueIn(
|
||
|
TupleOut(
|
||
|
[]OpenEdge{TupleOut(
|
||
|
[]OpenEdge{
|
||
|
ValueOut(i(1), Value{}),
|
||
|
TupleOut(
|
||
|
[]OpenEdge{
|
||
|
ValueOut(i(2), n("d")),
|
||
|
ValueOut(i(3), Value{}),
|
||
|
},
|
||
|
n("c"),
|
||
|
),
|
||
|
},
|
||
|
n("b"),
|
||
|
)},
|
||
|
n("a"),
|
||
|
),
|
||
|
n("out"),
|
||
|
),
|
||
|
},
|
||
|
{
|
||
|
in: "out = { a = 1; b = c < d < 2; };",
|
||
|
exp: ZeroGraph.AddValueIn(
|
||
|
ValueOut(
|
||
|
Value{Graph: ZeroGraph.
|
||
|
AddValueIn(ValueOut(i(1), Value{}), n("a")).
|
||
|
AddValueIn(
|
||
|
TupleOut(
|
||
|
[]OpenEdge{
|
||
|
ValueOut(i(2), n("d")),
|
||
|
},
|
||
|
n("c"),
|
||
|
),
|
||
|
n("b"),
|
||
|
),
|
||
|
},
|
||
|
Value{},
|
||
|
),
|
||
|
n("out"),
|
||
|
),
|
||
|
},
|
||
|
{
|
||
|
in: "out = a < { b = 1; } < 2;",
|
||
|
exp: ZeroGraph.AddValueIn(
|
||
|
TupleOut(
|
||
|
[]OpenEdge{
|
||
|
ValueOut(
|
||
|
i(2),
|
||
|
Value{Graph: ZeroGraph.
|
||
|
AddValueIn(ValueOut(i(1), Value{}), n("b")),
|
||
|
},
|
||
|
),
|
||
|
},
|
||
|
n("a"),
|
||
|
),
|
||
|
n("out"),
|
||
|
),
|
||
|
},
|
||
|
{
|
||
|
in: "a = 1; b = 2;",
|
||
|
exp: ZeroGraph.
|
||
|
AddValueIn(ValueOut(i(1), Value{}), n("a")).
|
||
|
AddValueIn(ValueOut(i(2), Value{}), n("b")),
|
||
|
},
|
||
|
}
|
||
|
|
||
|
for i, test := range tests {
|
||
|
t.Run(strconv.Itoa(i), func(t *testing.T) {
|
||
|
|
||
|
r := &mockReader{body: []byte(test.in)}
|
||
|
lexer := NewLexer(r)
|
||
|
|
||
|
got, err := DecodeLexer(lexer)
|
||
|
assert.NoError(t, err)
|
||
|
assert.True(t, Equal(got, test.exp), "\nexp:%v\ngot:%v", test.exp, got)
|
||
|
|
||
|
})
|
||
|
}
|
||
|
}
|