mirror of
https://github.com/neilotoole/sq.git
synced 2024-12-18 13:41:49 +03:00
48 lines
1.3 KiB
Go
48 lines
1.3 KiB
Go
package ast
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/neilotoole/lg/testlg"
|
|
)
|
|
|
|
func TestInspector_findSelectableSegments(t *testing.T) {
|
|
log := testlg.New(t).Strict(true)
|
|
|
|
// `@mydb1 | .user | .uid, .username`
|
|
ast, err := buildInitialAST(t, fixtSelect1)
|
|
require.Nil(t, err)
|
|
err = NewWalker(log, ast).AddVisitor(typeSelector, narrowTblSel).Walk()
|
|
require.Nil(t, err)
|
|
|
|
insp := NewInspector(log, ast)
|
|
|
|
segs := ast.Segments()
|
|
require.Equal(t, 3, len(segs))
|
|
|
|
selSegs := insp.FindSelectableSegments()
|
|
require.Equal(t, 1, len(selSegs), "should be 1 selectable segment: the tbl sel segment")
|
|
finalSelSeg, err := insp.FindFinalSelectableSegment()
|
|
require.Nil(t, err)
|
|
require.Equal(t, selSegs[0], finalSelSeg)
|
|
|
|
// `@mydb1 | .user, .address | join(.user.uid == .address.uid) | .uid, .username, .country`
|
|
ast, err = buildInitialAST(t, fixtJoinQuery1)
|
|
require.Nil(t, err)
|
|
err = NewWalker(log, ast).AddVisitor(typeSelector, narrowTblSel).Walk()
|
|
require.Nil(t, err)
|
|
insp = NewInspector(log, ast)
|
|
|
|
segs = ast.Segments()
|
|
require.Equal(t, 4, len(segs))
|
|
|
|
selSegs = insp.FindSelectableSegments()
|
|
require.Equal(t, 2, len(selSegs), "should be 2 selectable segments: the tbl selector segment, and the join segment")
|
|
|
|
finalSelSeg, err = insp.FindFinalSelectableSegment()
|
|
require.Nil(t, err)
|
|
require.Equal(t, selSegs[1], finalSelSeg)
|
|
}
|