Springboot编写单元测试上传文件和参数

1、工程pom文件引入工具包hutool包和fastjson

<dependency>
    <groupId>cn.hutool</groupId>
    <artifactId>hutool-core</artifactId>
    <version>5.6.5</version>
</dependency>
<!-- https://mvnrepository.com/artifact/cn.hutool/hutool-crypto -->
<dependency>
    <groupId>cn.hutool</groupId>
    <artifactId>hutool-crypto</artifactId>
    <version>5.6.5</version>
</dependency>
<dependency>
    <groupId>cn.hutool</groupId>
    <artifactId>hutool-http</artifactId>
    <version>5.6.5</version>
</dependency>
<dependency>
    <groupId>com.alibaba</groupId>
    <artifactId>fastjson</artifactId>
    <version>1.2.76</version>
</dependency>

2、控制器接收文件和参数代码

import com.lxdz.keyijie.api.service.MembershipService;
import com.lxdz.keyijie.common.restful.RestBody;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.RestController;
import org.springframework.web.multipart.MultipartFile;

@RestController("membershipController")
@RequestMapping("terminal/employee")
public class MembershipController {

    private final MembershipService service;

    public MembershipController(MembershipService service) {
        this.service = service;
    }

    @PostMapping("upload/face")
    public RestBody<String> uploadFace(@RequestParam(value = "empId") String empId, @RequestParam("file") MultipartFile file) {
        return service.uploadFace(empId, file);
    }

}

3、service将MultipartFile转为File处理，如果有需要的同学，不需要用File处理的请直接看最后。

MultipartFile和File转换工具类

package com.lxdz.keyijie.common.util;

import org.springframework.web.multipart.MultipartFile;
import java.io.File;
import java.io.FileOutputStream;
import java.io.InputStream;
import java.io.OutputStream;

/**
 * @ClassName MultipartFileToFile
 * @Description MultipartFile转fie
 **/
public class MultipartFileToFile {

    /**
     * MultipartFile 转 File
     *
     * @param file
     * @throws Exception
     */
    public static File multipartFileToFile(MultipartFile file) throws Exception {

        File toFile = null;
        if (file.equals("") || file.getSize() <= 0) {
            file = null;
        } else {
            InputStream ins = null;
            ins = file.getInputStream();
            toFile = new File(file.getOriginalFilename());
            inputStreamToFile(ins, toFile);
            ins.close();
        }
        return toFile;
    }

    //获取流文件
    private static void inputStreamToFile(InputStream ins, File file) {
        try {
            OutputStream os = new FileOutputStream(file);
            int bytesRead = 0;
            byte[] buffer = new byte[8192];
            while ((bytesRead = ins.read(buffer, 0, 8192)) != -1) {
                os.write(buffer, 0, bytesRead);
            }
            os.close();
            ins.close();
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

    /**
     * 删除本地临时文件
     * @param file
     */
    public static void delteTempFile(File file) {
        if (file != null) {
            File del = new File(file.toURI());
            del.delete();
        }
    }
}

在业务需要处理文件的地方转换

	File f = null;
    try {
        f = MultipartFileToFile.multipartFileToFile(file);
    } catch (Exception e) {
        return new RestBody<>(RestStatus.FAILED, "文件转换失败");
    }
    HttpResponse response = HttpUtil.createPost(AppConfig.getUploadHost() + "/upload/faceimage")
            .contentType(MediaType.MULTIPART_FORM_DATA_VALUE)
            .form("file", f)
            .execute();
    MultipartFileToFile.delteTempFile(f);

4、单元测试类

import cn.hutool.http.HttpResponse;
import cn.hutool.http.HttpUtil;
import com.alibaba.fastjson.JSONObject;
import com.lxdz.keyijie.api.pojo.form.MembershipForm;
import org.junit.jupiter.api.Test;
import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.http.MediaType;

import java.io.File;
import java.util.HashMap;
import java.util.Map;


@SpringBootTest(webEnvironment = SpringBootTest.WebEnvironment.RANDOM_PORT)
public class MembershipControllerTests {

    protected String basicUrl = "http://localhost:9900";
    protected <T> String post(String url, T data, String token) {
        HttpResponse response = HttpUtil.createPost(basicUrl + url)
                .contentType(MediaType.APPLICATION_JSON_VALUE)
                .auth("Bearer " + token)
                .body(JSONObject.toJSONString(data))
                .execute();
        return response.body();
    }

    protected <T> String postFile(String url, String empId, File f, String token) {
        HttpResponse response = HttpUtil.createPost(basicUrl + url)
                .contentType(MediaType.MULTIPART_FORM_DATA_VALUE)
                .auth("Bearer " + token)
                .form("empId", empId)
                .form("file", f)
                .execute();
        return response.body();
    }

    /**
     * 登录
     *
     * @return token
     */
    protected String login() {
        String url = "http://localhost:8602/oauth/token";
        Map<String, Object> param = new HashMap<>();
        param.put("grant_type", "client_credentials");
        HttpResponse response = HttpUtil.createPost(url)
                .contentType(MediaType.APPLICATION_FORM_URLENCODED_VALUE)
                .form(param)
                .basicAuth("client", "580488702cb6db1a307f8072e543a666")
                .execute();
        JSONObject object = JSONObject.parseObject(response.body());
        return object.getString("access_token");
    }

    private static final String MODEL_URL = "/terminal/employee";

    /**
     * @description 中台添加人员并开卡
     * @author Haixing.Wang
     * @since V1.0.0
     * 2021/6/17 11:31 上午
     */
    @Test
    public void testAddMembership() {
        String token = login();
        MembershipForm form = new MembershipForm();
        String s = "11";
        form.setEmpId("1"+s);
        form.setName("赵"+s);
        form.setPhoneNumber("123456789"+s);
        form.setAccount("1"+s);
        form.setOrgId("101");
        form.setMealCardTypeName("员工卡");
        System.out.println(post(MODEL_URL+"/add", form, token));
    }

    /**
     * @description 中台上传人脸图片
     * @author Haixing.Wang
     * @since V1.0.0
     * 2021/6/17 11:32 上午
     */
    @Test
    public void testUploadFace() {
        String token = login();
        File f = new File("/Users/wanghaixing/Desktop/test3.jpg");
        System.out.println(postFile(MODEL_URL+"/upload/face", "111", f, token));
    }

    /**
     * @description 调用upload服务上传人脸
     * @author Haixing.Wang
     * @since V1.0.0
     * 2021/6/17 11:32 上午
     */
    @Test
    public void testUploadFaceServer() {
        File f =new File("/Users/wanghaixing/Desktop/test.jpg");
        HttpResponse response = HttpUtil.createPost("http://192.168.1.10:8100/upload/faceimage")
                .contentType(MediaType.MULTIPART_FORM_DATA_VALUE)
                .form("file", f)
                .execute();
        System.out.println("response.body() =========>>>"+response.body());
    }
}