improve SchemaStatVisitor

This commit is contained in:
wenshao
2013-05-26 00:37:39 +08:00
parent 06d25a84b3
commit e324eef0d9
5 changed files with 142 additions and 12 deletions
@@ -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());
}
}
@@ -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;