模拟投影结果Spring Data JPA

Question

我正在使用Spring Data JPA在我的spring boot项目。

我正在激发一个JPQL查询，并使用一个投影来存储查询的结果。我的投影：

public interface VeryBasicProjection {
    String getTitle();
    String getUrl();
}

我的服务调用此投影：

public List<VeryBasicDTO> getLatestData(int limit){

        // Pageable for Limit
        Pageable pageable = new PageRequest(0, limit);

        // Get Data from DB
        List<VeryBasicProjection> latestData = tableRepository.getLatestData("live", 2,pageable);
        List<VeryBasicDTO> responseDTO = new ArrayList<>();

        // Map Projection to DTO
        for(VeryBasicProjection veryBasicProjection : latestData){
            VeryBasicDTO veryBasicDTO = new VeryBasicDTO();
            veryBasicDTO.buildDTO(veryBasicProjection);
            responseDTO.add(veryBasicDTO);
        }

        return responseDTO;
    }

现在我想使用Mockito（单元测试用例）测试这个服务，我正在使用when和thenreturn模拟对存储库的调用。

我的问题是如何模拟存储库的结果？回信里应该有什么？我的意思是，我如何创建投影实例并将数据设置到它？

Answer 1

如果希望创建投影的实例而不创建实现接口的类，则可以使用SpelawareProxyProjectionFactory。

import org.springframework.data.projection.ProjectionFactory;
import org.springframework.data.projection.SpelAwareProxyProjectionFactory;

...
ProjectionFactory factory = new SpelAwareProxyProjectionFactory();
VeryBasicProjection projection = factory.createProjection(VeryBasicProjection.class);
projection.setTitle("theTitle");
projection.setUrl("theUrl");

您还需要在投影中添加setter：

public interface VeryBasicProjection {
    String getTitle();
    String getUrl();
    void setTitle(String title);
    void setUrl(String url);
}

来源：https://github.com/spring-projects/spring-data-examples/blob/master/rest/projections/src/test/Java/example/springdata/rest/projections/simpleprojectiontests.Java