JUnit Mockito测试异常

Question

我应该如何测试异常？我可以mock connector并且我可以赋予它抛出异常的行为，但是我不明白下一步该怎么做。

public void removeStudentFromCourse(Integer studentId, Integer courseId) {
    try (Connection connection = connector.getConnection();
         PreparedStatement statement = connection.prepareStatement(
                 DELETE_STUDENT_FROM_ONE_COURSE_QUERY)) {
        statement.setInt(1, studentId);
        statement.setInt(2, courseId);
        statement.executeUpdate();
    } catch (SQLException e) {
        throw new DBException("Can`t remove student with id: " + studentId +
                " from course with id:" + courseId, e);
    }
}

Answer 1

类似于（未编译/未测试的代码，使用Mockito+AssertJ):

when(connector.getConnection())
    .thenThrow(new SQLException());
assertThatCode(() -> removeStudentFromCourse(...))
    .throwsExceptionOfType(DBException.class);