如何在同一 class 中的另一個方法中模擬方法調用

Question

我正在嘗試測試方法methodB （如下面的代碼所示）。 我想直接從methodA返回而不進入methodA的實際代碼。 我使用when/thenReturn和doReturn/when但沒有取得任何成功。 測試用例進入methodA的真實代碼。 還嘗試將spy與 Class A實例一起使用，但沒有成功。

實際 Class

class A{
  
  fun methodA(String a): String{

    // do something
    throw new Exception("An error occured");
  }

  fun methodB(String b): String{

    try{
      methodA("test")
    } catch (e: Exception){
      println(e.message());
    }
  }

}

測試 Class

class ATest{

    private lateinit var a: A

    @Before
    fun setUp() {
        a= A() // I am initializing services in real scenario
    }

    @Test
    fun `when methodB is not valid then throw Exception`(){

        val aMock = mock(A)
        Mockito.when(aMock.methodA("test") )
            .thenThrow(UserException.INVALID_REQUEST())


        // When
        val exception: Exception = Assert.assertThrows(
            UserException::class.java
        ) {
            a.methodB("test b")
        }

        val expectedMessage = "INVALID"
        val actualMessage = exception.message

        // Then
        Assert.assertTrue(actualMessage!!.contains(expectedMessage))
    }

   
}

有人可以幫忙嗎？

Answer 1

val a = A()
val aSpy = Mockito.spy(a)
Mockito.when(aSpy.methodA(ArgumentMatchers.any())).thenThrow(UserException.INVALID_REQUEST())

您需要獲取 class A 的真實實例，然后將其包裝在作為間諜的可編程包裝器中。 when語句的參數匹配器是第二個失敗點，因為“Test”與“test b”字符串不是同一個實例。 所以你不需要驗證一些參數所以跳過它。

如果要驗證使用的參數，可以使用捕獲器而不是 ANY()。