我希望能够模拟一个ResultSet。认真的。 我正在重构一个大而复杂的代码块，它从ResultSet解析数据，并且我希望我的代码行为完全一致。因此，我需要编写单元测试以测试被重构的部分，以便测试这个。
在Google搜索之后，我想到了两个想法：
第二种方法看起来更容易一些，而且更易于支持。
您有什么建议可以创建这样的模拟吗？（当然要除了医生）？我是否遗漏了某个神器？可能，DBUnit是这个工具吗？
我曾经使用MockRunner中的MockResultSet类获得了成功。它允许您创建实现ResultSet接口的类，并为每个列和行设置值。
如果您的方法使用的结果集大小合理，那么您应该能够轻松地创建返回所需值的测试。
这里是一个简单的示例：
MockResultSet rs = new MockResultSet("myMock");
rs.addColumn("columnA", new Integer[]{1});
rs.addColumn("columnB", new String[]{"Column B Value"});
rs.addColumn("columnC", new Double[]{2});
// make sure to move the cursor to the first row
try
{
rs.next();
}
catch (SQLException sqle)
{
fail("unable to move resultSet");
}
// process the result set
MyObject obj = processor.processResultSet(rs);
// run your tests using the ResultSet like you normally would
assertEquals(1, obj.getColumnAValue());
assertEquals("Column B Value", obj.getColumnBValue());
assertEquals(2.0d, obj.getColumnCValue());
private static class SimpleInvocationHandler implements InvocationHandler {
private Object invokee;
public SimpleInvocationHandler(Object invokee) {
this.invokee = invokee;
}
public Object invoke(Object proxy, Method method, Object[] args)
throws Throwable {
method = invokee.getClass().getMethod(method.getName(), method.getParameterTypes());
if (!method.isAccessible()) {
method.setAccessible(true);
}
try {
return method.invoke(invokee, args);
} catch (InvocationTargetException e) {
throw e.getTargetException();
}
}
}
public static <T> T generateProxy(Object realObject, Class... interfaces) {
return (T) Proxy.newProxyInstance(realObject.getClass().getClassLoader(), interfaces, new SimpleInvocationHandler(realObject));
}
Mockrunner 可以自动加载 CSV 或 XML 文件并创建 MockResultSet。它还可以模拟 Connection 和 Statement，因此所有的 JDBC 都可以正常工作，而无需将 JDBC 驱动程序添加到类路径中。
我为这个相同的情况写了一些东西。你可以使用Mockito模拟结果集。你也可以通过模拟resultset.next()来循环遍历模拟行。
// two dimensional array mocking the rows of database.
String[][] result = { { "column1", "column2" }, { "column1", "column2" } };
@InjectMocks
@Spy
private TestableClass testableClass;
@Mock
private Connection connection;
@Mock
private Statement statement;
@Mock
private ResultSet resultSet;
@BeforeTest
public void beforeTest() {
MockitoAnnotations.initMocks(this);
}
@BeforeMethod
public void beforeMethod() throws SQLException {
doAnswer(new Answer<Connection>() {
public Connection answer(InvocationOnMock invocation)
throws Throwable {
return connection;
}
}).when(testableClass).getConnection();
when(connection.createStatement()).thenReturn(statement);
when(statement.executeQuery(anyString())).thenReturn(resultSet);
final AtomicInteger idx = new AtomicInteger(0);
final MockRow row = new MockRow();
doAnswer(new Answer<Boolean>() {
@Override
public Boolean answer(InvocationOnMock invocation) throws Throwable {
int index = idx.getAndIncrement();
if (result.length > index) {
String[] current = result[index];
row.setCurrentRowData(current);
return true;
} else
return false;
}
;
}).when(resultSet).next();
doAnswer(new Answer<String>() {
@Override
public String answer(InvocationOnMock invocation) throws Throwable {
Object[] args = invocation.getArguments();
int idx = ((Integer) args[0]).intValue();
return row.getColumn(idx);
}
;
}).when(resultSet).getString(anyInt());
}
static class MockRow {
String[] rowData;
public void setCurrentRowData(String[] rowData) {
this.rowData = rowData;
}
public String getColumn(int idx) {
return rowData[idx - 1];
}
}
如果适用的话，您现在可以从真实数据源获取结果集，对其进行序列化并保存文件。然后，您可以为每个单元测试反序列化该结果集，这样您就可以顺利进行了。
ResultSet方法，我可能会将分隔文本文件加载到二维数组中，并实现我实际需要的方法，让其余方法抛出UnsupportedOperationException（这是我的IDE中未完成方法的默认实现）。