mirror of
https://github.com/neilotoole/sq.git
synced 2024-12-20 22:51:35 +03:00
93 lines
2.4 KiB
Go
93 lines
2.4 KiB
Go
package ast
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/neilotoole/sq/libsq/core/lg/lgt"
|
|
)
|
|
|
|
func TestChildIndex(t *testing.T) {
|
|
const q1 = `@mydb1 | .user | join(.address, .user.uid == .address.uid) | .uid, .username, .country`
|
|
|
|
p := getSLQParser(q1)
|
|
query := p.Query()
|
|
ast, err := buildAST(lgt.New(t), query)
|
|
require.Nil(t, err)
|
|
require.NotNil(t, ast)
|
|
require.Equal(t, 4, len(ast.Segments()))
|
|
|
|
for i, seg := range ast.Segments() {
|
|
index := nodeChildIndex(ast, seg)
|
|
require.Equal(t, i, index)
|
|
}
|
|
}
|
|
|
|
func TestNodesWithType(t *testing.T) {
|
|
nodes := []Node{&ColSelectorNode{}, &ColSelectorNode{}, &TblSelectorNode{}, &RowRangeNode{}}
|
|
|
|
require.Equal(t, 2, len(nodesWithType(nodes, typeColSelectorNode)))
|
|
require.Equal(t, 1, len(nodesWithType(nodes, typeTblSelectorNode)))
|
|
require.Equal(t, 1, len(nodesWithType(nodes, typeRowRangeNode)))
|
|
require.Equal(t, 0, len(nodesWithType(nodes, typeJoinNode)))
|
|
}
|
|
|
|
func TestNodePrevNextSibling(t *testing.T) {
|
|
const in = `@sakila | .actor | .actor_id == 2`
|
|
log := lgt.New(t)
|
|
a, err := Parse(log, in)
|
|
require.NoError(t, err)
|
|
|
|
equalsNode := NodesHavingText(a, "==")[0]
|
|
|
|
gotPrev := NodePrevSibling(equalsNode)
|
|
require.Equal(t, ".actor_id", gotPrev.Text())
|
|
require.Nil(t, NodePrevSibling(gotPrev))
|
|
|
|
gotNext := NodeNextSibling(equalsNode)
|
|
require.Equal(t, "2", gotNext.Text())
|
|
require.Nil(t, NodeNextSibling(gotNext))
|
|
}
|
|
|
|
func TestNodeUnwrap(t *testing.T) {
|
|
var ok bool
|
|
exprA := &ExprNode{}
|
|
exprB := &ExprNode{}
|
|
|
|
var gotExpr *ExprNode
|
|
|
|
gotExpr, ok = NodeUnwrap[*ExprNode](exprA)
|
|
require.True(t, ok)
|
|
require.True(t, exprA == gotExpr)
|
|
|
|
require.NoError(t, exprA.AddChild(exprB))
|
|
gotExpr, ok = NodeUnwrap[*ExprNode](exprA)
|
|
require.True(t, ok)
|
|
require.True(t, exprB == gotExpr)
|
|
|
|
litA := &LiteralNode{}
|
|
var gotLit *LiteralNode
|
|
require.NoError(t, exprB.AddChild(litA))
|
|
|
|
gotLit, ok = NodeUnwrap[*LiteralNode](exprA)
|
|
require.True(t, ok)
|
|
require.True(t, litA == gotLit)
|
|
|
|
litB := &LiteralNode{}
|
|
require.NoError(t, exprB.AddChild(litB))
|
|
gotLit, ok = NodeUnwrap[*LiteralNode](exprA)
|
|
require.False(t, ok, "should fail because exprB has multiple children")
|
|
require.Nil(t, gotLit)
|
|
}
|
|
|
|
func TestFindNodes(t *testing.T) {
|
|
const in = `@sakila | .actor | .actor_id == 2 | .actor_id, .first_name, .last_name`
|
|
a, err := Parse(lgt.New(t), in)
|
|
require.NoError(t, err)
|
|
|
|
handles := FindNodes[*HandleNode](a)
|
|
require.Len(t, handles, 1)
|
|
require.Equal(t, "@sakila", handles[0].Handle())
|
|
}
|