1- package test
1+ package parser
22
33import (
4- "testing"
5- "os"
6- "fmt"
74 "bufio"
5+ "fmt"
6+ "os"
87 "path/filepath"
98 "strings"
10-
11- "github.com/actiontech/java-sql-extractor/parser"
9+ "testing"
1210)
1311
12+ const TestFloder = "../test"
13+
1414func getJavaFiles () []string {
1515 var files []string
1616
17- testFolder := "."
1817 javaSuffix := ".java"
19- err := filepath .Walk (testFolder , func (path string , info os.FileInfo , err error ) error {
18+ err := filepath .Walk (TestFloder , func (path string , info os.FileInfo , err error ) error {
2019 if strings .HasSuffix (path , javaSuffix ) {
2120 files = append (files , path )
2221 }
@@ -39,23 +38,20 @@ func getSqlsFromSqlFile(sqlFile string) []string {
3938 scanner := bufio .NewScanner (f )
4039
4140 for scanner .Scan () {
42- sqls = append (sqls , scanner .Text ())
43- }
41+ sqls = append (sqls , scanner .Text ())
42+ }
4443
4544 return sqls
4645}
4746
4847func TestJavaFile (t * testing.T ) {
4948 javaFiles := getJavaFiles ()
5049 for _ , file := range javaFiles {
51- javaParser , err := parser . CreateJavaParser (file )
50+ sqls , err := GetSqlFromJavaFile (file )
5251 if err != nil {
5352 t .Error (err )
5453 }
55- v := parser .NewJavaVisitor ()
56- javaParser .CompilationUnit ().Accept (v )
57- sqls := parser .GetSqlsFromVisitor (v )
58- sqlFileSqls := getSqlsFromSqlFile (file + ".sql" )
54+ sqlFileSqls := getSqlsFromSqlFile (file + ".sql" )
5955 for i , sql := range sqls {
6056 if sql != sqlFileSqls [i ] {
6157 t .Error (fmt .Errorf ("sql parser failed, java file: %s" , file ))
0 commit comments