unit-test-in-SpringBoot-with-login-token

classify

precondition

test class

SpringBootTest(webEnvironment = SpringBootTest.WebEnvironment.RANDOM_PORT)
@AutoConfigureMockMvc
class ClsName {

	// bind the above RANDOM_PORT
    @LocalServerPort
    private int port;

    @Autowired
    private TestRestTemplate restTemplate;

    private String token;

}

pom.xml

<!-- jUnit test start -->
<dependency>
   <groupId>org.junit.jupiter</groupId>
   <artifactId>junit-jupiter</artifactId>
   <scope>test</scope>
   <version>5.9.0</version>
</dependency>
<dependency>
   <groupId>org.mockito</groupId>
   <artifactId>mockito-junit-jupiter</artifactId>
   <version>4.8.0</version>
   <scope>test</scope>
</dependency>
<dependency>
   <groupId>org.springframework</groupId>
   <artifactId>spring-test</artifactId>
   <version>5.3.22</version>
   <scope>test</scope>
</dependency>
<dependency>
   <groupId>org.springframework.boot</groupId>
   <artifactId>spring-boot-test</artifactId>
   <version>2.7.3</version>
   <scope>test</scope>
</dependency>
<dependency>
   <groupId>org.hamcrest</groupId>
   <artifactId>hamcrest-library</artifactId>
   <version>1.3</version>
   <scope>test</scope>
</dependency>
<dependency>
   <groupId>org.springframework.boot</groupId>
   <artifactId>spring-boot-test-autoconfigure</artifactId>
   <version>2.6.6</version>
   <scope>test</scope>
</dependency>
<!-- jUnit test end -->

with login, use token

CRUD: create read update delete

auto login beforeEach

/**
 * 模拟登录
 */
@BeforeEach
public void setLoginUser() throws MalformedURLException {
    LoginBodyEntity fdLoginBodyEntity = new LoginBodyEntity();
    loginBodyEntity.setUsername("username");
    loginBodyEntity.setPassword("password");
    ResponseEntity<AjaxResult> response = restTemplate.postForEntity(
    new URL("http://localhost:" + port + "/login").toString, loginBodyEntity, AjaxResult.class);

    token = (String) Objects.requireNonNull(response.getBody()).get("data");
}

create

/**
 * 测试增加nbhd
 * name唯一性
 * 新建的各项数据合理性
 */
@Test
@DisplayName("test add unique name success")
void addNbhd() throws Exception {
    AddModel aEntity = new AddModel();
    aEntity.setName("name");

    HttpHeaders headers = new HttpHeaders();
    headers.setBearerAuth(token);
    HttpEntity<AddModel> entity = new HttpEntity<>(aEntity, headers);

    ResponseEntity<AjaxResult> response = restTemplate.postForEntity(
       new URL("http://localhost:" + port + "/add").toString, entity, AjaxResult.class);

    System.out.println("code: " + Objects.requireNonNull(response.getBody()).get("code") + "; msg: "
        + Objects.requireNonNull(response.getBody()).get("msg"));
    assertEquals(200, Objects.requireNonNull(response.getBody()).get("code"));
    assertEquals("操作成功", response.getBody().get("msg"));
}

read

/**
 * 测试 list
*/
@Test
public void getList() throws Exception {
    HttpHeaders headers = new HttpHeaders();
    headers.setBearerAuth(token);
    HttpEntity<String> requestHeader = new HttpEntity<>(headers);

    String urlTemplate = UriComponentsBuilder.fromHttpUrl(new URL("http://localhost:" + port + "/list").toString, port))
        .queryParam("pageNum", "1")
        .queryParam("pageSize", "12")
        .queryParam("name", "nbhd2")
//        .queryParam("orderByColumn", "arm_type")
//        .queryParam("isAsc", "desc")
        .build()
        .toString();

    ResponseEntity<TableDataInfo> response = restTemplate.exchange(
        urlTemplate, HttpMethod.GET, requestHeader, TableDataInfo.class);

    System.out.println("size is " + Objects.requireNonNull(response.getBody()).getData().getList().size()
        + "; nbhd: " + Objects.requireNonNull(response.getBody()).getData().getList().toString());
    assertEquals(200, Objects.requireNonNull(response.getBody()).getCode());
    assertThat("row", Long.parseLong(String.valueOf(response.getBody().getData().getList().size())), greaterThan(0L));
}

/**
 * test detail getInfo
*/
@Test
public void getDetail() throws Exception {
    HttpHeaders headers = new HttpHeaders();
    headers.setBearerAuth(token);

    HttpEntity<String> requestHeader = new HttpEntity<>(headers);

    String urlTemplate = UriComponentsBuilder.fromHttpUrl(new URL("http://localhost:" + port + "/12").toString, port))
        .build()
        .toString();

    ResponseEntity<AjaxResult> response = restTemplate.exchange(
        urlTemplate, HttpMethod.GET, requestHeader, AjaxResult.class);

    System.out.println("nbhd: " + Objects.requireNonNull(response.getBody()).get("data").toString());
    assertEquals(200, Objects.requireNonNull(response.getBody()).get("code"));
    assertEquals("操作成功", response.getBody().get("msg"));
}

update

/**
 * 测试edit
*/
@Test
@DisplayName("test edit success")
void testEdit() throws Exception {
    EModel eModel = new EModel();
    eModel.setId(3L);
    eModel.setName("newName");

    HttpEntity<EModel> entity = new HttpEntity<>(eModel, headers);

    ResponseEntity<AjaxResult> response1 = restTemplate.exchange(
        new URL("http://localhost:" + port + "/edit").toString, HttpMethod.PUT, entity1, AjaxResult.class);

    System.out.println("msg: " + Objects.requireNonNull(response1.getBody()).get("msg"));
    assertEquals(200, Objects.requireNonNull(response1.getBody()).get("code"));
    assertEquals("操作成功", response1.getBody().get("msg"));
}

/**
 * 测试edit
 * json实例
*/
@Test
@DisplayName("test edit unique name success")
void editWithJson() throws Exception {
    List<String> list1 = new ArrayList<>();
    JSONObject json1 = new JSONObject();
    json1.put("sfId", 1);
    json1.put("weight", 50);
    list1.add(String.valueOf(json1));
    json1.put("sfId", 2);
    json1.put("weight", 50);
    list1.add(String.valueOf(json1));

    String urlTemplate1 = UriComponentsBuilder.fromHttpUrl(
            new URL("http://localhost:" + port + "/edit?id={id}&name={name}&jsonStr={jsonStr}").toString, port))
        .build()
        .toString();

    String name1 = "nameAdd3";
    ResponseEntity<AjaxResult> response1 = restTemplate.exchange(
        urlTemplate1, HttpMethod.PUT, requestHeader, AjaxResult.class, 12, name1, list1.toString());

    System.out.println("msg: " + Objects.requireNonNull(response1.getBody()).get("msg"));
    assertEquals(200, Objects.requireNonNull(response1.getBody()).get("code"));
    assertEquals("操作成功", response1.getBody().get("msg"));
}

delete

/**
 * 测试删除
 */
@Test
@DisplayName("test del success")
void testDel() throws Exception {
    HttpHeaders headers = new HttpHeaders();
    headers.setBearerAuth(token);
    HttpEntity<String> requestHeader = new HttpEntity<>(headers);

    String idsStr = "15,16";
    String urlTemplate = UriComponentsBuilder.fromHttpUrl(new URL("http://localhost:" + port + "/{ids}").toString, port))
        .build()
        .toString();

    ResponseEntity<AjaxResult> response = restTemplate.exchange(
        urlTemplate, HttpMethod.DELETE, requestHeader, AjaxResult.class, idsStr);

    assertEquals(200, Objects.requireNonNull(response.getBody()).get("code"));
    assertEquals("操作成功", response.getBody().get("msg"));
}

advantages

• test all changed codes work well one time

• 本文作者： Linking
• 本文链接： https://linking.fun/2022/09/17/unit-test-in-SpringBoot-with-login-token/
• 版权声明： 版权所有，转载请注明出处！