protected ResultSet createResultSet(MockPreparedStatement stmt) {
MockResultSet rs = new MockResultSet(stmt);
String sql = stmt.getSql();
if ("SELECT 1".equalsIgnoreCase(sql)) {
rs.getRows().add(new Object[] {1});
} else if ("SELECT NOW()".equalsIgnoreCase(sql)) {
rs.getRows().add(new Object[] {new java.sql.Timestamp(System.currentTimeMillis())});
} else if ("SELECT ?".equalsIgnoreCase(sql)) {
rs.getRows().add(new Object[] {stmt.getParameters().get(0)});
}
return rs;
}
public JdbcTestSupport() {
mockConnection = new MockConnection();
mockPreparedStatement = new MockPreparedStatement();
mockConnection.prepareStatementReturn = mockPreparedStatement;
mockStatement = new MockStatement();
mockConnection.createStatementReturn = mockStatement;
mockResultSet = new MockResultSet();
mockResultSet.returnedRows = new ArrayList();
mockPreparedStatement.executeQueryReturn = mockResultSet;
mockStatement.executeQueryReturn = mockResultSet;
}
