mirror of
https://github.com/alibaba/druid.git
synced 2023-03-04 19:17:16 +00:00
improve SchemaStatVisitor
This commit is contained in:
+10
-1
@@ -17,9 +17,10 @@ package com.alibaba.druid.bvt.sql.mysql.visitor;
|
||||
|
||||
import java.util.List;
|
||||
|
||||
import junit.framework.Assert;
|
||||
import junit.framework.TestCase;
|
||||
|
||||
import org.junit.Assert;
|
||||
|
||||
import com.alibaba.druid.sql.ast.SQLStatement;
|
||||
import com.alibaba.druid.sql.dialect.mysql.parser.MySqlStatementParser;
|
||||
import com.alibaba.druid.sql.dialect.mysql.visitor.MySqlSchemaStatVisitor;
|
||||
@@ -81,5 +82,13 @@ public class MySqlSchemaStatVisitorTest1 extends TestCase {
|
||||
Assert.assertEquals(true, visitor.getColumns().contains(new Column("usergroups", "id")));
|
||||
Assert.assertEquals(true, visitor.getColumns().contains(new Column("usergroups", "name")));
|
||||
|
||||
Assert.assertTrue(visitor.getColumn("users", "name").isSelect());
|
||||
Assert.assertTrue(visitor.getColumn("usergroups", "name").isSelect());
|
||||
|
||||
Assert.assertTrue(visitor.getColumn("users", "groupId").isJoin());
|
||||
Assert.assertTrue(visitor.getColumn("usergroups", "id").isJoin());
|
||||
|
||||
Assert.assertTrue(visitor.getColumn("users", "groupId").isWhere());
|
||||
|
||||
}
|
||||
}
|
||||
|
||||
+2
-1
@@ -17,9 +17,10 @@ package com.alibaba.druid.bvt.sql.mysql.visitor;
|
||||
|
||||
import java.util.List;
|
||||
|
||||
import junit.framework.Assert;
|
||||
import junit.framework.TestCase;
|
||||
|
||||
import org.junit.Assert;
|
||||
|
||||
import com.alibaba.druid.sql.ast.SQLStatement;
|
||||
import com.alibaba.druid.sql.dialect.mysql.parser.MySqlStatementParser;
|
||||
import com.alibaba.druid.sql.dialect.mysql.visitor.MySqlSchemaStatVisitor;
|
||||
|
||||
Reference in New Issue
Block a user