如何测试 SimpleJdbcCall

Question

我需要为此代码创建测试。

@Autowired
JdbcTemplate jdbcTemplate;

public List<Row> getData(int id) {
    // Preconditions here

    SimpleJdbcCall getCall = new SimpleJdbcCall(jdbcTemplate)
            .withSchemaName(SCHEMA)
            .withProcedureName(SP)
            .declareParameters( 
                // ...
            )
            .returningResultSet("result", (RowMapper<QuestionAnswerRow>) (rs, rowNum) -> .....);

    MapSqlParameterSource params = new MapSqlParameterSource();
    params.addValue("id", id);
    // other parameters here

    Map queryRes = getCall.execute(params);
    List<row> res = (List<row>) queryRes.get("result");
    return res;
}

你能告诉我如何用 Mockito 在这里模拟 getCall.execute(params) 响应吗？

Answer 1

你的问题是，你在你的方法中创建了整个 getCall 对象，这使得几乎不可能以某种方式将模拟注入该过程。

这里有几种可能性：

a) 您可以模拟您的 jdbcTemplate，然后尝试通过模拟 getCall 将对 jdbcTemplate 进行的所有调用来使其正常工作。由于 jdbcTemplate 可能会完成所有实际的数据库工作，这可以工作，但老实说，这可能不值得付出努力（因为它绝对不平凡）。

b) 您可以将整个测试切换为集成测试，例如，让它运行针对内存中的数据库。当然，有很多反对使用集成测试代替单元测试的论据，所以这可能也不是最好的方法。不过，这是可能的，并且使用 Spring 测试实用程序和注释可以非常简单。

c) 这给我们留下了一些工作，在这种情况下意味着重构：

由于您的问题是您在内部创建了 SimpleJdbcCall，因此一种解决方案是将该部分提取到工厂中。这个简化的例子表明：

@Component
class SimpleJdbcCallFactory {
   public SimpleJdbcCall create(JdbcTemplate template) {
       return new SimpleJdbcCall(template);
   }
}

现在您可以向您的 class 添加一个 @Autowired 依赖项，然后在您的单元测试中模拟该依赖项...

@RunWith(MockitoJUnitRunner.class)
public class YourTestClassHere {

   @Mock
   private SimpleJdbcCallFactory simpleJdbcCallFactory;

   @InjectMocks
   private YourClassHere classToTest;

   @Test
   public void test() {
        SimpleJdbcCall mockedCall = Mockito.mock(SimpleJdbcCall.class);
        Mockito.when( simpleJdbcCallFactory.create(Mockito.any())).thenReturn(mockedCall);
        Mockito.when( mockedCall ).withSchemaName(Mockito.anyString()).thenReturn(mockedCall);
        // etc. unfortunately needed for fluent apis (unless they added those in mockito)
        Mockito.when( mockedCall.execute(Mockito.any()).thenReturn( ... );

        classToTest.getData(123);
   }

}

Answer 2

为了解决no datasource specified我们需要模拟SimpleJdbcCall，上面的代码是调用原始方法。所以我们得到了上面的错误。

请使用 power mockito 模拟参数化构造函数，如下所示

JdbcTemplate jdbcTemplateMock = Mockito.mock(JdbcTemplate.class);
SimpleJdbcCall mockedSimpleJdbcCall = Mockito.mock(SimpleJdbcCall.class);

PowerMockito.whenNew(SimpleJdbcCall.class).withArguments(jdbcTemplateMock).thenReturn(mockedSimpleJdbcCall);

如何测试 SimpleJdbcCall

How to test SimpleJdbcCall

mocking

jdbctemplate

mockito