milvus/internal/parser/planparserv2/pool_test.go
Jiquan Long e756fdccd4
Fix pool of lexer and parser (#17129)
Signed-off-by: longjiquan <jiquan.long@zilliz.com>
2022-05-20 18:47:57 +08:00

40 lines
880 B
Go

package planparserv2
import (
"testing"
"github.com/stretchr/testify/assert"
"github.com/antlr/antlr4/runtime/Go/antlr"
antlrparser "github.com/milvus-io/milvus/internal/parser/planparserv2/generated"
)
func genNaiveInputStream() *antlr.InputStream {
return antlr.NewInputStream("a > 2")
}
func Test_getLexer(t *testing.T) {
var lexer *antlrparser.PlanLexer
lexer = getLexer(genNaiveInputStream(), &errorListener{})
assert.NotNil(t, lexer)
lexer = getLexer(genNaiveInputStream(), &errorListener{})
assert.NotNil(t, lexer)
}
func Test_getParser(t *testing.T) {
var lexer *antlrparser.PlanLexer
var parser *antlrparser.PlanParser
lexer = getLexer(genNaiveInputStream(), &errorListener{})
assert.NotNil(t, lexer)
parser = getParser(lexer, &errorListener{})
assert.NotNil(t, parser)
parser = getParser(lexer, &errorListener{})
assert.NotNil(t, parser)
}