使用 Citrus 框架接收 HTTP Post 请求中的 PDF 并写入文件

import org.springframework.http.HttpStatus;
import org.springframework.http.MediaType;
import org.springframework.core.io.FileSystemResource;

import java.io.IOException;
import java.nio.file.Files;
import java.nio.file.Paths;

import com.consol.citrus.dsl.testng.TestNGCitrusTestRunner;
import org.testng.annotations.Test;

public class PdfHandlingTest extends TestNGCitrusTestRunner {

    @Test
    public void testReceiveAndWritePdf() {
        String pdfFileName = "received.pdf";

        // 发送请求并接收PDF响应
        http(httpActionBuilder -> httpActionBuilder
                .client("vdeClient")
                .send()
                .post("/generatePdf")
                .payload(new FileSystemResource("src/test/resources/payload.xml")) // 假设有一个payload.xml文件
                .contentType(MediaType.APPLICATION_XML_VALUE));

        http(httpActionBuilder -> httpActionBuilder
                .client("vdeClient")
                .receive()
                .response(HttpStatus.OK)
                .contentType(MediaType.APPLICATION_PDF_VALUE)
                .name("pdfResponse"));

        // 获取Payload并写入文件
        receive(receiveBuilder -> receiveBuilder
                .message(context -> {
                    byte[] pdfBytes = (byte[]) context.getMessageStore().getMessage("pdfResponse").getPayload();
                    try {
                        Files.write(Paths.get(pdfFileName), pdfBytes);
                    } catch (IOException e) {
                        e.printStackTrace();
                    }
                    return null;
                }));
    }
}

import com.mgnt.lifecycle.management.httpclient.HttpClient;
import com.mgnt.utils.TextUtils;

import java.io.IOException;
import java.nio.ByteBuffer;
import java.nio.file.Files;
import java.nio.file.Paths;

public class HttpClientExample {

    public static void downloadFile() {
        HttpClient httpClient = new HttpClient();
        httpClient.setRequestHeader("Accept", "application/octet-stream");
        httpClient.setConnectionUrl("https://dl.k8s.io/release/v1.20.0//bin/linux/amd64/kubectl");
        ByteBuffer buffer = null;
        try {
            buffer = httpClient.sendHttpRequestForBinaryResponse(HttpClient.HttpMethod.GET);
            System.out.println(httpClient.getLastResponseCode() + " " + httpClient.getLastResponseMessage());
        } catch (IOException ioe) {
            System.out.println(httpClient.getLastResponseCode() + " " + httpClient.getLastResponseMessage());
            System.out.println(TextUtils.getStacktrace(ioe, "com.mgnt.stam."));
        }
        try {
            Files.write(Paths.get("C:\Michael\work\Installations\Kubernetes\kubectl"), buffer.array());
        } catch (IOException e) {
            System.out.println(TextUtils.getStacktrace(e, "com.mgnt.stam."));
        }
    }
}