SpringBoot2.x之MockMvc进阶使用

之前写单元测试都是Junit进行测试，由于项目需要，需要对服务接口进行流程化测试，所以我选择了MockMvc，接下来就结合我们项目上的实际情况进行了整合，希望对大家有所帮助！

一.MockMvc

1.1 依赖引入

        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-test</artifactId>
        </dependency>

1.2 配置MockMvc

这里小编进行了简单的封装，因为项目实际需求，便捷开发，可以直接用

@Slf4j
@Data
@Transactional
@RunWith(SpringJUnit4ClassRunner.class)
public class BaseTest{

    @Autowired
    private WebApplicationContext context ;
    private MockMvc mvc ;

    /**
     * @Description 初始化
     */
    @Before
    public void setUp() {
        MockitoAnnotations.initMocks(this);
        mvc = MockMvcBuilders.webAppContextSetup(context).build();  //构造MockMvc
    }

    /**
     * @Description 统一输出
     * @param mvcResult 返回值
     */
    public void sysLog(MvcResult mvcResult){
        try {
            MockHttpServletResponse response = mvcResult.getResponse();
            MatcherAssert.assertThat("Service execution successful", response.getStatus(), equalTo(HttpServletResponse.SC_OK));
            log.info("Service execution return value <---> {}",response.getContentAsString());
        } catch (UnsupportedEncodingException e) {
            log.info("Service execution return value resolution failure");
            e.printStackTrace();
        }
    }

    /**
     * @Description 发送json请求
     * @param url           访问地址
     * @param authorization 所需要的权限
     * @param obj           参数
     * @return MvcResult    返回值
     */
    public MvcResult doJson(String url, String authorization, Object obj){
        try {
            MvcResult mvcResult =
                    this.getMvc().perform(MockMvcRequestBuilders.post(url)
                            .header("Authorization", authorization)
                            .contentType(MediaType.APPLICATION_JSON_UTF8)
                            .content(JSON.toJSONString(obj)))
                            .andExpect(status().isOk())
                            .andDo(print()).andReturn();
            return mvcResult;
        } catch (Exception e) {
            log.info("Please check the parameter or call mode");
            e.printStackTrace();
            return null;
        }
    }


    /**
     * @Description 发送from请求
     * @param url           访问地址
     * @param authorization 所需要的权限
     * @param params        参数
     * @return MvcResult    返回值
     */
    public MvcResult doFrom(String url, String authorization,  MultiValueMap<String, String> params){
        try {
            MvcResult mvcResult =
                    this.getMvc().perform(MockMvcRequestBuilders.post(url)
                            .header("Authorization", authorization)
                            .contentType(MediaType.ALL)
                            .params(params))
                            .andExpect(status().isOk())
                            .andDo(print()).andReturn();
            return mvcResult;
        } catch (Exception e) {
            log.info("Please check the parameter or call mode");
            e.printStackTrace();
            return null;
        }
    }
}

注：
由于项目是整合JWT验证，这里访问还需要进行权限的验证，所以我在.header(“Authorization”, authorization)，这里进行了配置

二.创建JWT用户加密

  /**
     * @Description 获取学生token
     * @UserModule: exam-common
     * @author Dylan
     * @date 2019/12/10
     * @param
     * @return java.lang.String
     */
    public static String getStudentToken(String currSystemFlag){
        User user =new User();
        user.setId("1df391615d9b4085b5d20fc0a476de22");
        user.setLoginName("gyc42");
        user.setPassword("123456");
        Role role = new Role();
        role.setId("a220425221ae4c4991f069e586b91094");
        role.setName("冯凯1班");
        role.setEnname(Constants.ROLE_ENNAME_STUDENT);
        user.setRoleList(Arrays.asList(role));
        return releaseToken(user,currSystemFlag,10000L);
    }


    /**
     * @Description token 加密
     */
    public static String releaseToken(User user, String currSystemFlag, Long validity) {
        String key = "JAVAYH";
        Algorithm alg = Algorithm.HMAC256(key);
        Role role = new Role();
        if(user.getRoleList()!=null&&user.getRoleList().size()>0) {
            role = user.getRoleList().get(0);
        }
        //签发Token
        Date currentTime = new Date();
        String token = JWT.create()
                .withIssuer("YHJ") // 发行者
                .withSubject(user.getId()) // 用户身份标识
                .withIssuedAt(currentTime) // 签发时间
                .withExpiresAt(new Date(currentTime.getTime() + validity))
                .withJWTId(UUID.randomUUID().toString()) // 分配JWT的ID
                .withClaim("currSystemFlag", currSystemFlag) // 定义公共域信息,可以自定义多个
                .withClaim("roleName", role.getEnname())
                .sign(alg);
        return token;
    }

三.实际调用

    @Test
    public void setPaperNameWindow(){
        VO vo= new VO();
        MvcResult mvcResult = this.doJson(setPaperNameWindow, getAdminToken(1), vo);
        sysLog(mvcResult);
    }

到这里我们的测试封装完成，如果遇到问题，可以加我微信，微信在下方

关注 Java有货领取更多资料

联系小编。微信：372787553，带您进群互相学习
左侧小编微信，右侧获取免费资料

技术博客：https://blog.csdn.net/weixin_38937840

SpringCloud学习代码： https://github.com/Dylan-haiji/javayh-cloud

Redis、Mongo、Rabbitmq、Kafka学习代码： https://github.com/Dylan-haiji/javayh-middleware

AlibabaCloud学习代码：https://github.com/Dylan-haiji/javayh-cloud-nacos

SpringBoot+SpringSecurity实现自定义登录学习代码：https://github.com/Dylan-haiji/javayh-distribution