Junit 单元测试之错误和异常处理

测试大大怪

于 2024-07-04 15:41:35 发布

阅读量1.1k

点赞数 16

文章标签： junit 单元测试压力测试测试工具测试用例功能测试 postman

本文链接：https://blog.csdn.net/kk_lzvvkpj/article/details/140181253

版权

整体代码示例

首先，为了简化，我们让服务层就是简单的类，然后使用Id查找用户，这个和之前测试UserService接口不太一样哦：

　@Service
　　public class UserService {
　　
　　 @Autowired
　　 private UserRepository userRepository;
　　
　　 public User getUserById(Long id) {
　　 return userRepository.findById(id).orElse(null);
　　 }
　　}

现在，我们要模拟UserRepository的行为，使其在尝试获取用户时引发一个异常。这里我们使用Mockito进行模拟：

　@RunWith(SpringRunner.class)
　　@SpringBootTest
　　public class UserServiceTest {
　　
　　//之前我们是定义了一个UserService接口，现在简化成UserService类了哈
　　 @InjectMocks
　　 private UserService userService;
　　
　　 @Mock
　　 private UserRepository userRepository;
　　
　　 @Before
　　 public void setUp() throws Exception {
　　 MockitoAnnotations.initMocks(this);
　　 }
　　
　　//重点，后文详解！
　　 @Test(expected = DatabaseConnectionException.class)
　　 public void testGetUserByIdWithDbError() {
　　 when(userRepository.findById(anyLong())).thenThrow(new DatabaseConnectionException("Database connection failed!"));
　　
　　 userService.getUserById(1L);
　　 }
　　}
　　
　　//重点，后文详解！
　　class DatabaseConnectionException extends RuntimeException {
　　 public DatabaseConnectionException(String message) {
　　 super(message);
　　 }
　　}

在上述测试中，我们模拟了userRepository.findById()方法，使其抛出DatabaseConnectionException异常。然后，我们在测试方法上使用@Test(expected = DatabaseConnectionException.class)来表示我们期望该方法引发此异常。

这样，如果getUserById方法在遇到此异常时没有正确处理，测试将失败。这确保了即使在面对意外的数据库问题时，我们的代码仍能按预期的方式运行（在这种情况下，按预期抛出异常）。

到底在模拟什么？到底在测试什么？

下面，我们进一步说明：

1. 测试目标：这个测试的目标是确保当userRepository.findById()方法抛出DatabaseConnectionException异常时，userService.getUserById()方法也会抛出同样的异常。

2. 模拟异常：在这行代码中，我们指定了当userRepository.findById()被调用时，它应该抛出DatabaseConnectionException异常。

when(userRepository.findById(anyLong())).thenThrow(new DatabaseConnectionException("Database connection failed!"));

3. 调用Service方法：接下来，我们调用了userService.getUserById(1L)。我们期望它在内部调用userRepository.findById()（这在实际的UserService实现中应该是这样的）。因此，由于我们已经模拟了userRepository.findById()来抛出异常，所以userService.getUserById()也应该会抛出这个异常。

4. 验证异常：@Test(expected = DatabaseConnectionException.class)注解表示我们期望这个测试方法在执行时会抛出DatabaseConnectionException异常。如果这个方法执行完并没有抛出这个异常，那么测试将会失败。

5. 测试的目的：这个测试的目的并不是检查userRepository.findById()本身是否真的会抛出异常，而是检查当它抛出异常时，userService.getUserById()是否会正确地传递这个异常。这可以帮助我们确保UserService在处理异常时的行为是正确的。其实本质上来说，抛出异常和预期值的测试逻辑几乎是一样的，都是通过给定下层值，验证上层代码关系。

综上所述，这个测试确保了当底层UserRepository出现数据库连接错误时，上层的UserService可以正确地传递这个错误。这对于后续的异常处理很重要，例如：在Controller层将这个异常转化为一个友好的错误消息返回给用户。

什么时候测试失败？

在正常情况下，只要Service层确实调用了Repository的方法，并且Repository的方法抛出了RuntimeException（或其子类），那么Service层的调用方法也应该会收到并进一步抛出这个异常。

但是，以下几种情况可能导致测试不通过：

异常被吞没：如果Service层调用了Repository的方法，但内部捕获了该异常并没有重新抛出，那么测试就会失败。例如：

　public User getUserById(Long id) {
　　 try {
　　 return userRepository.findById(id);
　　 } catch (DatabaseConnectionException e) {
　　 // 异常被吞没了
　　 return null;
　　 }
　　}

调用的方法不正确：如果Service层没有调用预期的Repository方法，而是调用了其他方法，或者完全没有调用，那么模拟的异常就不会被触发，导致测试失败。

模拟的不正确：如果在测试中模拟的方法或参数与实际调用的方法或参数不匹配，那么模拟的异常也不会被触发。例如，如果Service实际上是这样调用的：userRepository.findById(2L)，但我们的模拟是这样的：when(userRepository.findById(1L))...，那么异常就不会被触发。

其他未预料到的异常：有时可能会有其他的未被预料到的异常被抛出，这也会导致测试失败。

因此，虽然大多数情况下，如果Repository层方法抛出了异常，Service层应该也会抛出，但还是存在一些情况导致测试不通过，这也是进行此类测试的原因。

Exception 异常类定义

　class DatabaseConnectionException extends RuntimeException {
　　 public DatabaseConnectionException(String message) {
　　 super(message);
　　 }
　　}

DatabaseConnectionException是一个自定义的异常类。在 Java中，异常是用来表示程序运行中的问题或异常情况的对象。当某些问题发生时，通常会抛出（throw）一个异常。

这里，我们定义了一个继承自RuntimeException的新异常类DatabaseConnectionException。RuntimeException是Java中所有非检查型异常的基类。所谓“非检查型”是指编译器不强制我们捕获或声明它。这与Exception（检查型异常）相对。

关于DatabaseConnectionException类的解释：

1. class DatabaseConnectionException extends RuntimeException - 这表示我们正在定义一个名为DatabaseConnectionException的新类，该类是RuntimeException的子类。这意味着DatabaseConnectionException继承了RuntimeException的所有特性。

2. public DatabaseConnectionException(String message) - 这是DatabaseConnectionException类的构造方法。当我们创建DatabaseConnectionException的新实例时，可以传递一个消息字符串给这个构造函数。

3. super(message); - 这行代码调用了父类（RuntimeException）的构造方法，并将message传递给它。这样，当异常被抛出并捕获时，我们可以获取并显示这个消息。

这种自定义异常，通常在我们希望为特定的错误情况定义更具描述性的异常名时使用，或者当我们想为特定的异常情况添加更多上下文信息时使用，信息越多，测试反馈的效果越好，所以一般使用自定义异常，继承RuntimeException！下面我们讨论一下，为什么建议使用RuntimeException？