Question

我有一个方法，可以使用multipart-formdata向api发送一个休息请求，这会将文件上传到外部api。但是，我无法完成此单元测试方法。

我发现的第一个问题是，我期望的内容类型总是与方法创建的内容类型不同。由于某种原因，发送请求时，媒体类型为multipart-formdata，但除了字符集和边界外，标头还设置为该标头。后者（边界）总是在更改其值，因此我无法在单元测试中设置期望值，因为它总是不同的。

除此之外，我还如何期望请求的内容与发起测试的内容相同？如何断定有效载荷是相同的。

请检查代码：

服务类别：

@Service
@Slf4j
public class JiraService {

    private HttpHeaders createRequestHeaders(JiraClient jiraClient, MediaType contenType) {
        HttpHeaders headers = new HttpHeaders();
        headers.setContentType(contenType);
        headers.setAccept(Collections.singletonList(MediaType.APPLICATION_JSON));
        headers.setBasicAuth(jiraClient.getUsername(), jiraClient.getPassword());

        return headers;
    }

    private <EC, RC> ResponseEntity<RC> createRequestAndSend(HttpMethod method, String url, HttpHeaders headers,
            EC payload, Class<RC> responseType) {
        HttpEntity<EC> requestEntity = new HttpEntity<>(payload, headers);
        ResponseEntity<RC> responseEntity = restTemplate.exchange(url, method, requestEntity, responseType);
        // TODO deal with response

        log.error("Loggin something");

        return responseEntity;
    }

    public void addAttachment(JiraClient jiraClient, JiraIssue jiraIssue, JiraAttachment jiraAttachment)
            throws MalformedURLException, IOException {
        String url = jiraClient.getHost() + "/rest/api/2/issue/" + jiraIssue.getKey() + "/attachments";

        HttpHeaders headers = createRequestHeaders(jiraClient, MediaType.MULTIPART_FORM_DATA); // What to do here?
        headers.set("X-Atlassian-Token", "no-check");

        FileSystemResource file = jiraAttachment.downloadFileFromWeb();
        MultiValueMap<String, Object> payload = new LinkedMultiValueMap<>();
        payload.add("file", file);

        createRequestAndSend(HttpMethod.POST, url, headers, payload, String.class);

        jiraAttachment.deleteFileFromSystem();
    }
}

ServiceTest.class


@ActiveProfiles("test")
@RestClientTest(JiraService.class)
public class JiraServiceTest {

  @Value("classpath:jira/add_attachment/validJiraAttachmentAddition.json")
  private Resource validJiraAttachmentAddition;

  @Autowired
  private MockRestServiceServer server;
  @Autowired
  private JiraService jiraService;

  @Mock
  private JiraAttachment mockJiraAttachment;

  private FileSystemResource attachmentFileSystemResource;

  @BeforeEach
  public void setupTests() throws IOException {
      // initialize mocks
  }

  @Test
  public void addAttachment_WithValidData_ShouldAddAttachmentToJiraIssue() throws Exception {
    String url = host + "/rest/api/2/issue/" + issueKey + "/attachments";

    ResponseActions stub = createServiceStub(HttpMethod.POST, url, MediaType.MULTIPART_FORM_DATA_VALUE);
    stub = stub.andExpect(header("X-Atlassian-Token", "no-check"));
    stub.andRespond(withSuccess());
    // How to assert that the content of the request is the same as the resource?

    when(mockJiraAttachment.downloadFileFromWeb()).thenReturn(attachmentFileSystemResource);

    jiraService.addAttachment(mockJiraClient, mockJiraIssue, mockJiraAttachment);
  }

  private ResponseActions createServiceStub(HttpMethod method, String url, String contenType) {
    String encodedCredentials = Base64.getEncoder().encodeToString((username + ":" + password).getBytes());

    ResponseActions stub = server.expect(ExpectedCount.once(), requestTo(url));
    stub = stub.andExpect(method(method));
    stub = stub.andExpect(header("Content-Type", contenType)); // How to expect the content type here ?
    stub = stub.andExpect(header("Authorization", "Basic " + encodedCredentials));

    return stub;
  }
}

Answer 1

使用ContentRequestMatchers.contentTypeCompatibleWith(MediaType contentType)

import static org.springframework.test.web.client.match.MockRestRequestMatchers.content;
...

stub.andExpect(content().contentTypeCompatibleWith(MediaType.MULTIPART_FORM_DATA))

MockRestServiceServer测试通过Multipart-formdata发送文件上传

1 个答案: