Question

我在使用自定义控制器对单元测试弹簧数据休息应用程序时遇到问题。

@RepositoryRestResource(path = "channels")
public interface ChannelsRepository extends PagingAndSortingRepository<Channel, Long> {
}

@RepositoryRestResource(path = "systems")
public interface SystemRepository extends PagingAndSortingRepository<System, String>{
}



@RestController
public class SystemController {

    private String query = "select a.id as fromId, b.id as toId\n" +
            "from channel a, channel b\n" +
            "where ST_Distance_Spheroid(\n" +
            "\ta.coordinates,\n" +
            "\tb.coordinates,\n" +
            "\t'SPHEROID[\"WGS 84\",6378137,298.257223563]'\n" +
            ") <= :criticalDistance and a.id != b.id";

    @Autowired
    private EntityManager manager;

    @Autowired
    private SystemRepository repository;

    @RequestMapping(value="systems/{systemId}/graph", method = RequestMethod.GET)
    public Map<BigInteger, List<BigInteger>> createNeighbourhsGraph(@PathVariable("systemId") String systemId,
                                                                    @RequestParam(value = "distance") double distance) {
        List<Object[]> results =  manager.createNativeQuery(query).setParameter("criticalDistance", distance).getResultList();
        Map<BigInteger, List<BigInteger>> map = new HashMap<BigInteger, List<BigInteger>>();
        for (Object[] result: results) {
            BigInteger fromId = (BigInteger) result[0];
            if (map.containsKey(fromId)) {
                map.get(fromId).add((BigInteger) result[1]);
            }
            else {
                List<BigInteger> neighbours = new ArrayList<BigInteger>();
                map.put(fromId, new ArrayList<BigInteger>(Arrays.asList(new BigInteger[]{(BigInteger) result[1]})));
            }
        }
        return map;
    }
}

我试图通过发布来填充数据，但没有任何运气。似乎在单元测试中没有创建任何在运行应用程序时创建的映射，所以我得到的只是说没有映射到'system / ...'等。

因此，您可以看到我尝试通过存储库填充数据。并且它也不起作用，可能是一些事务问题，因为我可以在日志中看到插入语句，但数据库中没有任何内容。

我看到我可以提供一些sql脚本来填充数据，但这似乎是非常愚蠢的事情，因为如果我们仍然使用SQL来执行像这样的简单任务，为什么我们有ORM。

    @Before
    public void setup() throws Exception {

        mockMvc = MockMvcBuilders.standaloneSetup(controller).build();
        channels.System system = new System("foo");

        GeometryFactory gf = new GeometryFactory();
        Channel[] channels = new Channel[]{
                new Channel(
                        1,
                        system,
                        gf.createPoint(new Coordinate(55.7565, 37.6153)),
                        1,
                        Arrays.asList(new String[]{"м. Охотный ряд"})
                ),
                new Channel(
                        1,
                        system,
                        gf.createPoint(new Coordinate(55.8079, 37.5808)),
                        1,
                        Arrays.asList(new String[]{"м. Дмитровская"})
                ),
                new Channel(
                        1,
                        system,
                        gf.createPoint(new Coordinate(50.4451, 30.5182)),
                        1,
                        Arrays.asList(new String[]{"м. Театральная"})
                ),

        };
        channelsRepository.save(Arrays.asList(channels));
        system.setChannels(Arrays.asList(channels));
        systemsRepository.save(system);
//        ClassLoader classLoader = getClass().getClassLoader();
//        String systemAsJson = IOUtils.toString(classLoader.getResourceAsStream("system.json"));
//        String channelsAsJson = IOUtils.toString(classLoader.getResourceAsStream("channels.json"));
//        JSONParser parser = new JSONParser();
//        JSONObject system = (JSONObject) parser.parse(systemAsJson);
//
//        String result = mockMvc.perform(post("/systems")
//                        .accept(MediaType.APPLICATION_JSON).
//                        content(system.toJSONString())
//                        ).andReturn().getResponse().getContentAsString();
//        System.out.println(result);
    }

Answer 1

您无法看到数据库中的更改，因为（如果您使用AbstractTransactionalSpringContextTests）spring会在测试执行后清除事务。因此，在测试方法内部可以获得数据。
使用sql脚本并不愚蠢，可能适用于很多情况。
查看TestNG ru

Answer 2

我已经添加了webapplication contet，现在我可以通过perform（post（））和执行（删除）方法发布和删除数据，我现在也可以使用它。

  @Autowired
    private WebApplicationContext context;

    private JSONArray channels;
    @Before
    public void setup() throws Exception {
        mockMvc = MockMvcBuilders.webAppContextSetup(context).build();

如何在spring数据休息中填充测试数据？

2 个答案: