Skip to content

Commit d857eee

Browse files
committed
change name
1 parent b06760e commit d857eee

File tree

7 files changed

+11
-15
lines changed

7 files changed

+11
-15
lines changed
Lines changed: 11 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -1,22 +1,21 @@
1-
package test
1+
package parser
22

33
import (
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+
1414
func 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

4847
func 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

Comments
 (0)