mirror of
https://github.com/alibaba/druid.git
synced 2023-03-04 19:17:16 +00:00
git-svn-id: http://code.alibabatech.com/svn/druid/trunk@82 b9813039-fb51-4c41-a8b9-e21c2acb5095
This commit is contained in:
@@ -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();
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user