PowerMock學習(三)之Mock局部變量
編寫powermock用例步驟:
- 類上面先寫這兩個註解@RunWith(PowerMockRunner.class)、@PrepareForTest(StudentService.class)
- 先模擬一個假對象即studentdao方法中的局部變量
- 用無參的方式new對象
- 再模擬這個對象被調用時,是否有返回,有返回值給出默認值,沒有用doNothing()
- 驗證有返回值使用assertEquals即可,無返回值使用Mockito.verify驗證
實際案例
接着上一篇文章中的代碼,修改下service中的代碼,這次我不通過構造器注入Dao,在方法中new一個StudentDao,創建一個名為StudentNewService的類。
具體示例代碼如下:
package com.rongrong.powermock.service; import com.rongrong.powermock.dao.StudentDao; /** * @author rongrong * @version 1.0 * @date 2019/11/17 21:13 */ public class StudentNewService { /** * 獲取學生個數 * @return返回學生總數 */ public int getTotal() { StudentDao studentDao = new StudentDao(); return studentDao.getTotal(); } /** * 創建學生 * @param student */ public void createStudent(Student student) { StudentDao studentDao = new StudentDao(); studentDao.createStudent(student); } }
針對上面修改部分代碼,進行單元測試,以下代碼有採用傳統方式測試和採用powermock方式進行測試,具體代碼如下:
package com.rongrong.powermock.service; import com.rongrong.powermock.dao.StudentDao; import org.junit.Assert; import org.junit.Test; import org.junit.runner.RunWith; import org.mockito.Mockito; import org.powermock.api.mockito.PowerMockito; import org.powermock.core.classloader.annotations.PrepareForTest; import org.powermock.modules.junit4.PowerMockRunner; import static org.junit.Assert.assertEquals; import static org.junit.Assert.fail; /** * @author rongrong * @version 1.0 * @date 2019/11/20 21:42 */ @RunWith(PowerMockRunner.class) @PrepareForTest(StudentNewService.class) public class TestNewStudentService { /** * 傳統方式測試 */ @Test public void testGetStudentTotal() { StudentNewService studentNewService = new StudentNewService(); int total = studentNewService.getTotal(); assertEquals(total, 10); } /** * @desc測試有返回值類型 採用powermock進行測試獲取學生個數 */ @Test public void testGetStudentTotalWithPowerMock() { //先模擬一個假對象即studentdao方法中的局部變量 StudentDao studentDao = PowerMockito.mock(StudentDao.class); try { //這句話我按照英文理解就是,我用無參的方式new了一個StudentDao對象 PowerMockito.whenNew(StudentDao.class).withNoArguments().thenReturn(studentDao); //再模擬這個對象被調用時,我們默認假定返回10個證明調用成功 PowerMockito.when(studentDao.getTotal()).thenReturn(10); //這裏就是service就不用再說了 StudentNewService studentNewService = new StudentNewService(); int total = studentNewService.getTotal(); assertEquals(total, 10); } catch (Exception e) { fail("測試失敗了!!!"); e.printStackTrace(); } } /** * @desc測試的無返回值類型 採用powermock進行測試創建學生 */ @Test public void testCreateStudentWithPowerMock() { //先模擬一個假對象即studentdao方法中的局部變量 StudentDao studentDao = PowerMockito.mock(StudentDao.class); try { //這句話我按照英文理解就是,我用無參的方式new了一個StudentDao對象 PowerMockito.whenNew(StudentDao.class).withNoArguments().thenReturn(studentDao); Student student = new Student(); //這句話註釋與否都能運行通過,也就是我只能判斷他是否被調用 //PowerMockito.doNothing().when(studentDao).createStudent(student); //這裏就是service就不用再說了 StudentNewService studentNewService = new StudentNewService(); studentNewService.createStudent(student); Mockito.verify(studentDao).createStudent(student); } catch (Exception e) { fail("測試失敗了!!!"); e.printStackTrace(); } } }
運行上面的測試用例,會發現第一個失敗,後面兩個都運行成功,即有返回值和無返回值類型的測試(void類型)。
注意:對於無返回值類型的測試,只能驗證其是否被調用,這裏還請注意。
本站聲明:網站內容來源於博客園,如有侵權,請聯繫我們,我們將及時處理
【其他文章推薦】
※台北網頁設計公司這麼多,該如何挑選?? 網頁設計報價省錢懶人包”嚨底家”
※網頁設計公司推薦更多不同的設計風格,搶佔消費者視覺第一線
※想知道購買電動車哪裡補助最多?台中電動車補助資訊懶人包彙整
※小三通海運與一般國際貿易有何不同?
※小三通快遞通關作業有哪些?