java - 带有模拟测试的 Dgs 框架？

Question

目前，我正在使用 dgs 框架从 GraphQL 服务器调用数据。现在我想创建一个 mocking(mockito) 和 junit 来测试它。我如何实现它的单元测试？

    public List<AuditData> getAuditData(RestTemplate restTemplate, int retLabelId, int countryId, int periodId, int versionNo, String URL, String token) {
        DefaultGraphQLClient graphQLClient = new DefaultGraphQLClient(URL);
        String query = QueryUtils.getAuditDataQuery(retLabelId, countryId, periodId, versionNo, PartitionUtil.getReadPartition(token));
        GraphQLResponse response = graphQLClient.executeQuery(query, new HashMap<>(), "", (url, headers, body) -> {
            HttpHeaders requestHeaders = new HttpHeaders();
            requestHeaders.setContentType(MediaType.APPLICATION_JSON);
            requestHeaders.set("Authorization", "Bearer " + token );
            headers.forEach(requestHeaders::put);
            ResponseEntity<String> exchange = restTemplate.exchange(url, HttpMethod.POST, new HttpEntity(body, requestHeaders), String.class);
            return  new HttpResponse(exchange.getStatusCodeValue(), exchange.getBody());
            }
        );
        return response.extractValueAsObject("data.auditData[*]",new TypeRef<List<AuditData>>(){});
    }