单元测试中如何验证 if 语句的执行情况？

2024-03-17 22:39:40

在单元测试中验证 if 语句的执行

问题

有时，我们需要在单元测试中验证代码中 if 语句的执行情况。例如，考虑以下 Kotlin 代码：

internal class ClassA {
  internal fun doA() {
    var file: File? = null

    try {
      // do something
      file = fileCreator.create()
    } catch (e: IOException) {
      throw e
    } finally {
      if (file.exists()) {
        file.delete()
      }
    }
  }
}

我们如何在单元测试中验证 file.delete() 行是否执行？

解决方案

为了验证 file.delete() 行的执行，我们可以使用 Mockito 框架的 verify() 方法。它允许我们验证方法是否被调用。

测试代码

以下是测试代码的示例：

import org.junit.Test
import org.junit.runner.RunWith
import org.mockito.ArgumentCaptor
import org.mockito.Captor
import org.mockito.InjectMocks
import org.mockito.Mock
import org.mockito.Mockito.*
import org.mockito.junit.MockitoJUnitRunner

@RunWith(MockitoJUnitRunner::class)
class ClassATest {

    @Mock
    private lateinit var fileCreator: FileCreator

    @InjectMocks
    private lateinit var classA: ClassA

    @Captor
    private val fileCaptor: ArgumentCaptor<File> = ArgumentCaptor.forClass(File::class.java)

    @Test
    fun `should delete file if it exists`() {
        // Arrange
        val file = mock(File::class.java)
        `when`(fileCreator.create()).thenReturn(file)
        `when`(file.exists()).thenReturn(true)

        // Act
        classA.doA()

        // Assert
        verify(file, times(1)).delete()
        verify(fileCreator, times(1)).create()
        verify(file, times(1)).exists()
    }

    @Test
    fun `should not delete file if it does not exist`() {
        // Arrange
        val file = mock(File::class.java)
        `when`(fileCreator.create()).thenReturn(file)
        `when`(file.exists()).thenReturn(false)

        // Act
        classA.doA()

        // Assert
        verify(file, times(0)).delete()
        verify(fileCreator, times(1)).create()
        verify(file, times(1)).exists()
    }
}