隨著專案的功能不斷增多啟動專案需要的時間就越來越長，如果每個功能都從Controller入口進行測試，不僅費時而且費力。
那麼這時候單元測試就體現出非常大的優勢了，對所有的介面進行測試，可以非常快速的發現問題，而且方便除錯。而且通過編寫單元測試來發現程式碼結構的不合理性，通常來說程式碼結構越合理，單元測試寫的越方便。
假設有這個場景，我們的專案啟動需要30s，共有A、B、C、D這四個service，A依賴B、C這兩個service，D Service為需要自動注入屬性值的Service。這裡使用執行緒睡眠的方式模擬啟動延遲：

@SpringBootApplication
public class Main {
    public static void main(String[] args) {
        SpringApplication.run(Main.class, args);
    }

    @Bean
    public ApplicationRunner applicationRunner() throws InterruptedException {
        System.out.println("===================模擬延遲--------------------");
        Thread.sleep(30000);
        return args -> {
            System.out.println("===================模擬延遲啟動--------------------");
        };
    }
}
 

如果使用SpringBootTest，那麼得要等到專案啟動後才執行測試方法。
故此添加了MainTest單元測試：

@RunWith(SpringRunner.class)
@SpringBootTest(classes = Main.class)
public class MainTest {
    @Autowired
    private AService aService;

    @Before
    public void before() {
        System.out.println("start=============");
    }

    @Test
    public void testStartUp() {
        aService.doSmthing();
        System.out.println("--------------測試----------");
    }
}

 

從控制檯可以看出，要等到專案30s後才執行到我們的測試方法，要是有多個單元測試單單啟動就浪費了大量的時間：

拆分單元測試

例如我們只需要測試A服務，那麼可以使用ContextConfiguration指定需要啟動的類和依賴類，由於A依賴B、C兩個服務，因此ContextConfiguration注入了A、B、C三個類，如：

@RunWith(SpringRunner.class)
@ContextConfiguration(classes = {
        AService.class,
        BService.class,
        CService.class
})
public class AServiceTest {
    @Autowired
    private AService aService;


    @Test
    public void test() {
        aService.doSmthing();
    }

}

 

執行後如下，不到1s就執行完了。

使用Mock服務

比如我們需要測試A的服務方法，B、C服務並不是我們關注的重點，這個時候可以將BC服務mock出來。

@RunWith(SpringRunner.class)
@ContextConfiguration(classes = {
        AService.class
})
public class AServiceMockTest {
    @Autowired
    private AService aService;
    @MockBean
    private BService bService;
    @MockBean
    private CService cService;

    @Before
    public void setUp() {
        Mockito.when(bService.doSmthing()).thenReturn("mockBServce");
        Mockito.when(cService.doSmthing()).thenReturn("mockCServce");
    }

    @Test
    public void test() {
        aService.doSmthing();
    }

}

佔位符測試

通常我們有配置檔案，如果使用配置中的屬性可以使用@Value("${key}")進行注入，如D Service：

@Service
public class DService {
    @Value("${config.username}")
    private String username;

    public String doSmthing() {
        return username+" doSmthing";
    }
}

@RunWith(SpringRunner.class)
@ContextConfiguration(classes = {
        DService.class,
        PropertySourcesPlaceholderConfigurer.class
})
@TestPropertySource(properties = {"config.username=myUsername"})
public class DServiceTest {
    @Autowired
    private DService dService;

    @Test
    public void test() {
        String result = dService.doSmthing();
        System.out.println(result);
    }
}

執行後輸出如下：

myUsername doSmthing

如果屬性多，那麼使用註解注入就不太方便了，這時可以使用屬性檔案進行配置，
@TestPropertySource(“classpath:application.properties”)或者在ContextConfiguration中新增initializers=ConfigFileApplicationContextInitializer.class進行自動配置，預設會配置resoruces下的application.properties屬性檔案

@RunWith(SpringRunner.class)
@ContextConfiguration(classes = {
        DService.class,
        PropertySourcesPlaceholderConfigurer.class
}, initializers = ConfigFileApplicationContextInitializer.class)
public class DServicePropertiesTest {
    @Autowired
    private DService dService;

    @Test
    public void test() {
        String result = dService.doSmthing();
        System.out.println(result);
    }
}

總結

以上例子說明了@SpringBootTest、@TestPropertySource、PropertySourcesPlaceholderConfigurer、ConfigFileApplicationContextInitializer等使用。已經可以覆蓋到大部分常用的單元測試情況了，更多的使用方式可以參考Spring官網和SpringBoot中的單元測試這章節：

以上程式碼已經提交到工程：LearnTest