單元測試（一）——SpringBoot創建單元測試

時間 2019-11-07

原文原文鏈接

在Spring項目中，對於controller、service、dao各層都須要創建單元測試項。對應不一樣的分層，咱們能夠使用junit和mock不一樣的方式。然而有些狀況會須要啓動spring容器來測試業務邏輯在容器內可否正常運行，針對此狀況可參考以下單元測試方式。java

SpringBoot增長依賴

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

創建測試基類

新建JUnitBaseTest.java

package com.lucas.device;

import org.junit.After;
import org.junit.AfterClass;
import org.junit.Before;
import org.junit.BeforeClass;
import org.junit.runner.RunWith;
import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.test.context.junit4.SpringRunner;
import org.springframework.test.context.web.WebAppConfiguration;
import org.springframework.util.StopWatch;

import lombok.extern.slf4j.Slf4j;

/**
 * <Description> <br>
 * 
 * @author xubin<br>
 * @version 1.0<br>
 * @taskId <br>
 * @CreateDate 2018年9月27日 <br>
 */

@Slf4j
@RunWith(SpringRunner.class)
@SpringBootTest
@WebAppConfiguration
public class JUnitBaseTest {

    /**
     * sw 計時器<br>
     */
    public static StopWatch sw = null;

    /**
     * Description: <br>
     * 
     * @author xubin<br>
     * @taskId <br>
     * @throws java.lang.Exception <br>
     */
    @BeforeClass
    public static void setUpBeforeClass() throws Exception {
        sw = new StopWatch();
    }

    /**
     * Description: <br>
     * 
     * @author xubin<br>
     * @taskId <br>
     * @throws java.lang.Exception <br>
     */
    @AfterClass
    public static void tearDownAfterClass() throws Exception {
        log.info("**************************************************************");
        log.info("單元測試計時統計：{}", sw.prettyPrint());
        log.info("**************************************************************");
    }

    /**
     * Description: <br>
     * 
     * @author xubin<br>
     * @taskId <br>
     * @throws java.lang.Exception <br>
     */
    @Before
    public void setUp() throws Exception {
        log.info("開始測試-----------------");
    }

    /**
     * Description: <br>
     * 
     * @author xubin<br>
     * @taskId <br>
     * @throws java.lang.Exception <br>
     */
    @After
    public void tearDown() throws Exception {
        sw.stop();
        log.info("測試結束-----------------");
    }
}

@BeforeClass 基類執行前觸發
@AfterClass 基類執行後觸發
@Before 每一個@Test方法調用前觸發
@After 每一個@Test方法調用後觸發

測試Case

service層單元測試示例 DeviceInfoTest.java

package com.lucas.device.service;

import static org.junit.Assert.fail;

import java.util.Date;
import java.util.List;

import javax.annotation.Resource;

import org.junit.Assert;
import org.junit.Ignore;
import org.junit.Test;
import org.springframework.data.redis.core.StringRedisTemplate;

import com.lucas.core.common.PageInfo;
import com.lucas.device.JUnitBaseTest;
import com.lucas.device.bo.DeviceInfoBO;
import com.lucas.device.entity.DeviceInfoEntity;
import com.lucas.device.entity.cache.DeviceCache;
import com.phlico.common.framework.tool.unique.IdWorkerUtil;
import com.phlico.common.framework.web.util.JsonUtil;

import lombok.extern.slf4j.Slf4j;

/**
 * <Description> <br>
 * 
 * @author xubin<br>
 * @version 1.0<br>
 * @taskId <br>
 * @CreateDate Nov 4, 2018 <br>
 */

@Slf4j
public class DeviceInfoTest extends JUnitBaseTest {

    /**
     * deviceInfoService <br>
     */
    @Resource(name = "deviceInfoServiceImpl2")
    private DeviceInfoService deviceInfoService;

    /**
     * deviceRedisTemplate <br>
     */
    @Resource
    private StringRedisTemplate stringRedisTemplate;

    /**
     * appKeyArr <br>
     */
    public static final String[] appKeyArr = {
            "2e2b380a56e7464aa678294c2c345545", "e2c85a1d245844fd9bdb14f0d0fc868a", "fa58f5306eb64ce094fe65fec6261587"
    };

    /**
     * Description: <br>
     * 
     * @author xubin<br>
     * @taskId <br>
     *         <br>
     */
    @Ignore
    @Test
    public void saveTest() {
        log.info("設備保存");
        sw.start("設備保存");
        try {
            DeviceInfoBO deviceInfoBO = new DeviceInfoBO();
            deviceInfoBO.setAppKey("2e2b380a56e7464aa678294c2c345545");
            deviceInfoBO.setDeviceName("test2");
            deviceInfoBO.setDeviceCode("EE-FF-GG");
            deviceInfoBO.setDeviceDesc("測試設備1");
            deviceInfoBO.setDeviceType("bcb0ca4ef0eb463aa76a5bc0ccee1b85");
            deviceInfoBO.setDeviceState("b9cf9e01a5534487919a77f21c34c93c");
            deviceInfoBO.setDeviceIpv4("192.168.109.12");
            deviceInfoBO.setDevicePort(8088);
            deviceInfoBO.setDeviceMac("ED:AS:ZQ:QS");
            deviceInfoBO.setProtocol(0);
            deviceInfoService.save(deviceInfoBO);
        } catch (Exception e) {
            fail("設備保存異常");
        }
    }

    /**
     * Description: <br>
     * 
     * @author xubin<br>
     * @taskId <br>
     *         <br>
     */
    @Ignore
    @Test
    public void updateTest() {
        log.info("設備修改");
        sw.start("設備修改");
        try {
            DeviceInfoBO deviceInfoBO = new DeviceInfoBO();
            deviceInfoBO.setAppKey("2e2b380a56e7464aa678294c2c345545");
            deviceInfoBO.setDeviceName("測試設備-" + appKeyArr[(int) (Math.random() * appKeyArr.length)]);
            deviceInfoBO.setDeviceCode("EE-FF-GG");
            deviceInfoBO.setDeviceDesc("測試設備2");
            deviceInfoBO.setDeviceState("b9cf9e01a5534487919a77f21c34c93c");
            deviceInfoBO.setDeviceIpv4("192.168.109.122");
            deviceInfoBO.setDevicePort(8089);
            deviceInfoBO.setDeviceMac("ED:AS:ZQ:QS:DA");
            deviceInfoBO.setProtocol(0);
            deviceInfoService.update(deviceInfoBO);
        } catch (Exception e) {
            fail("設備修改異常");
        }
    }

    /**
     * Description: <br>
     * 
     * @author xubin<br>
     * @taskId <br>
     *         <br>
     */
    @Ignore
    @Test
    public void delDeviceByCodesTest() {
        log.info("設備刪除");
        sw.start("設備刪除");
        try {
            deviceInfoService.delDevicesByCode("2e2b380a56e7464aa678294c2c345545", "EE-FF-GG", null, null, null);
        } catch (Exception e) {
            fail("設備刪除異常");
        }
    }

    /**
     * Description: <br>
     * 
     * @author xubin<br>
     * @taskId <br>
     *         <br>
     */
    @Ignore
    @Test
    public void delDeviceByIdsTest() {
        log.info("按id刪除設備");
        sw.start("按id刪除設備");
        try {
            deviceInfoService.delete("2e2b380a56e7464aa678294c2c345545", "d99b77637c1e421db89ac89579d43eef,00d2105433d94106a88596f275e58d41", null,
                    null, new Date());
        } catch (Exception e) {
            fail("按id刪除設備異常");
        }
    }

    /**
     * Description: <br>
     * 
     * @author xubin<br>
     * @taskId <br>
     *         <br>
     */
    @Ignore
    @Test
    public void queryEntityListByPageTest() {
        log.info("分頁查詢設備");
        sw.start("分頁查詢設備");
        try {
            DeviceInfoBO bo = new DeviceInfoBO();
            bo.setAppKey("2e2b380a56e7464aa678294c2c345545");
            PageInfo<DeviceInfoBO> list = deviceInfoService.queryEntityListByPage(bo, 1, 15);
            log.info(JsonUtil.getSucc4data(list));
            Assert.assertNotEquals(null, list);
        } catch (Exception e) {
            fail("分頁查詢設備異常");
        }
    }

    /**
     * Description: <br>
     * 
     * @author xubin<br>
     * @taskId <br>
     *         <br>
     */
    @Ignore
    @Test
    public void queryDeviceInfoByCacheTest() {
        log.info("查詢緩存全部設備");
        sw.start("查詢緩存全部設備");
        try {
            String appKey = "fa58f5306eb64ce094fe65fec6261587";
            String deviceCodes = "577984734685110272";
            List<DeviceCache> list = deviceInfoService.queryDeviceInfoByCache(appKey, deviceCodes);
            log.info(JsonUtil.getSucc4data(list));
            Assert.assertNotEquals(null, list);
        } catch (Exception e) {
            fail("查詢緩存全部設備異常");
        }
    }
}