git-svn-id: http://code.alibabatech.com/svn/druid/trunk@82 b9813039-fb51-4c41-a8b9-e21c2acb5095

This commit is contained in:
wenshao
2011-05-14 15:27:21 +00:00
parent 6f966ac983
commit 41b1cd52c3
2 changed files with 35 additions and 0 deletions
@@ -13,6 +13,7 @@ import com.alibaba.druid.mock.MockStatement;
import com.alibaba.druid.sql.ast.SQLExpr;
import com.alibaba.druid.sql.ast.SQLStatement;
import com.alibaba.druid.sql.ast.expr.SQLCharExpr;
import com.alibaba.druid.sql.ast.expr.SQLIdentifierExpr;
import com.alibaba.druid.sql.ast.expr.SQLIntegerExpr;
import com.alibaba.druid.sql.ast.expr.SQLMethodInvokeExpr;
import com.alibaba.druid.sql.ast.expr.SQLNCharExpr;
@@ -64,6 +65,14 @@ public class MySqlMockExecuteHandlerImpl implements MockExecuteHandler {
SQLTableSource from = query.getFrom();
if (from instanceof SQLExprTableSource) {
SQLExpr expr = ((SQLExprTableSource) from).getExpr();
if (expr instanceof SQLIdentifierExpr) {
String ident = ((SQLIdentifierExpr) expr).getName();
if ("dual".equalsIgnoreCase(ident)) {
return executeQueryFromDual(statement, query);
}
}
throw new SQLException("TODO");
} else if (from == null) {
return executeQueryFromDual(statement, query);
@@ -51,4 +51,30 @@ public class MockExecuteTest extends TestCase {
stmt.close();
conn.close();
}
public void test_3() throws Exception {
Connection conn = DriverManager.getConnection("jdbc:mock:");
Statement stmt = conn.createStatement();
ResultSet rs = stmt.executeQuery("SELECT NOW() FROM DUAL");
Assert.assertTrue(rs.next());
Assert.assertTrue(rs.getObject(1) instanceof Timestamp);
rs.close();
stmt.close();
conn.close();
}
public void test_4() throws Exception {
Connection conn = DriverManager.getConnection("jdbc:mock:");
Statement stmt = conn.createStatement();
ResultSet rs = stmt.executeQuery("SELECT 'ABCDE' FROM DUAL");
Assert.assertTrue(rs.next());
Assert.assertEquals("ABCDE", rs.getString(1));
rs.close();
stmt.close();
conn.close();
}
}