时间戳

首页 > 解决方案 > 模拟测试异常处理

java - 模拟测试异常处理

问题描述

描述：我想对实现接口的方法进行单元测试。方法从文件中检索数据，因此可能会引发异常。实际检索数据的代码是Files.readAllLines(Paths.get(fileName));，这块迫使我处理IOException。还有两个其他类使用接口FileLinesReader来检索数据。一切正常，代码符合预期。但是，当我使用无效的文件名（在任何位置都不存在）运行整个代码时，NoSuchFileException会抛出在我的理解中由Paths.get(fileName). 我想正确处理代码执行期间最有可能发生的异常。

代码：

public interface FileLinesReader {
    List<String> readAllLines(String fileName);
}

    public class DefaultFileLinesReader implements FileLinesReader {
    String defaultFile;

    public DefaultFileLinesReader(String defaultFile) {
        this.defaultFile = defaultFile;
    }

    @Override
    public List<String> readAllLines(String fileName) {
        List<String> linesOutput;

        try {
            linesOutput = Files.readAllLines(Paths.get(fileName));
        }
        catch(IOException e) {
            System.err.println("Provided file: \"" + fileName + "\" does not exist. Default file: " + defaultFile);
            e.printStackTrace();
            System.err.println("Do you want to use default file? [y/n]");

            try(Scanner scanner = new Scanner(System.in)) {
                while(true) {
                    String input = userInput(scanner);
                    if (input.equals("y")) {
                        try {
                            linesOutput = Files.readAllLines(Paths.get(defaultFile));
                            break;
                        }
                        catch (IOException ex) {
                            throw new RuntimeException("Default file not defined!", ex); //Should never be thrown. Default file must exist in MessageSender folder.
                        }
                    }
                    //Default file must exist in designated folder.
                    //If no file selected, execution is ceased.
                    if (input.equals("n")) {
                        throw new RuntimeException("No file selected."); 
                    }
                    System.out.println("Invalid command.");
                    System.out.println("Do you want to use default file? [y/n]");
                }
            }
        }
        return linesOutput;
    }

    //Created to define input in unit tests
    protected String userInput(Scanner scanner) {
        return scanner.next();
    }
}

问题：我想使用 Mockito 来测试这个类，但显然我对 mocking 的理解还不够。我想监视设置用户输入的方法，然后还监视 throw RuntimeException，因为我想停止代码执行。我得到的结果就像我的代码正常执行，而不是作为测试。try-catch块已执行，但我得到的消息是NoSuchFileException已被抛出。它达到了用户可以输入命令的那一刻，但是应该通过“n”作为输入的间谍方法肯定不起作用。这是我真正感到困惑的地方。我不确定我是否正确捕获了异常。我试图只为创建单独的方法，Paths.get(fileName)所以我也可以模拟它，但我觉得代码因此变得更加麻烦。我现在拥有的是这样的代码：

@ExtendWith(MockitoExtension.class)
class DefaultFileLinesReaderTest {

    private static final String FAKE_FILE = "fakeFile";

    private DefaultFileLinesReader defaultFileLinesReaderTest;
    private DefaultFileLinesReader defFileLinesReaderSpy;

    @BeforeEach
    void setUp() {
        defaultFileLinesReaderTest = new DefaultFileLinesReader(FAKE_FILE);
        defFileLinesReaderSpy = spy(defaultFileLinesReaderTest);
    }

    @Test
    void readAllLines_RuntimeException() {
        Scanner scanner = mock(Scanner.class);
        when(defFileLinesReaderSpy.userInput(scanner)).thenReturn("n");
        when(defFileLinesReaderSpy.readAllLines(FAKE_FILE)).thenThrow(new RuntimeException());

        assertThrows(RuntimeException.class, () -> defFileLinesReaderSpy.readAllLines(FAKE_FILE));
    }
}

运行测试时的输出是这样的：

Provided file: "fakeFile" does not exist. Default file: fakeFile
java.nio.file.NoSuchFileException: fakeFile
    \\stacktrace here
Do you want to use default file? [y/n]

问题：如何测试readAllLines方法中抛出的异常是否按预期处理？

标签： javaexceptioninterfacemockingmockito

解决方案

通常，您会抛出RuntimeException的子类，您可能会想到使用IllegalArgumetException。

对于单元测试，你应该有测试方法来测试你的类方法的正常行为和其他测试方法来测试抛出的异常情况。

要测试可能引发异常的情况，您可以考虑以下示例（JUnit 4）：

@Rule
public ExpectedException exceptionRule = ExpectedException.none();

@Test
public void shouldThrowException() {
    exceptionRule.expect(IllegalArgumentException.class);
    exceptionRule.expectMessage("Your exception message");
}

如果您只想检查抛出的异常类型，您可以简单地使用：

@Test(expected = IllegalArgumentException.class)
public void shouldthrowException() {
    // Call method
}

推荐阅读