如何用JUnit和Mockito测试所有分支均返回error的switch-case逻辑？

核心思路 #

要定位哪个switch分支触发了异常，核心是把分支标识和异常结果关联起来，要么在业务返回中携带分支信息，要么在测试环节通过Mock验证追踪调用链路。

方案1：修改业务代码（简单直接） #

在catch块返回的错误信息中加入当前分支的标识（即switch的入参str），测试时通过返回值就能直接定位分支：

try {
    switch (str) {
        case "a":
            return mapper.writeValueAsString(sampleService.sampleServiceMethod(samplePojo));
        // 其他case分支
    }
} catch (Exception e) {
    // 错误信息追加分支标识
    return DefaultMethodResult.error.name() + "-" + str;
}

测试时只需断言返回值包含对应分支的后缀，即可确认是该分支触发了异常。

方案2：仅通过Mockito验证（不修改业务代码） #

如果不想改动业务逻辑，可利用Mockito的调用验证能力，结合测试用例的分支入参定位问题：

1. 针对每个分支编写独立测试用例，以测试str="a"分支为例：

   @Test
   void testCaseAError() {
       // 模拟sampleService的目标方法抛出异常
       doThrow(new RuntimeException("mock error")).when(sampleService).sampleServiceMethod(any(SamplePojo.class));
       
       // 调用目标方法并传入分支参数"a"
       String result = targetMethod("a");
       
       // 断言返回错误结果
       assertEquals(DefaultMethodResult.error.name(), result);
       // 验证服务方法是否被调用——若被调用，说明异常来自case "a"分支
       verify(sampleService, times(1)).sampleServiceMethod(any(SamplePojo.class));
   }
   

2. 若某个分支对应的服务方法未被验证到调用，说明异常并非在该分支触发（比如分支匹配错误、前置逻辑抛出异常等）。

方案3：局部捕获异常并标记分支（折中方案） #

如果不想全局修改错误返回格式，但愿意在每个case分支做微小调整，可在case内局部捕获异常并携带分支标识后再抛出，最后在全局catch块返回带标识的错误：

try {
    switch (str) {
        case "a":
            try {
                return mapper.writeValueAsString(sampleService.sampleServiceMethod(samplePojo));
            } catch (Exception e) {
                throw new RuntimeException("branch-" + str, e);
            }
        // 其他case分支同理处理
    }
} catch (Exception e) {
    // 从异常信息中提取分支标识并返回
    String branch = e.getMessage().split("-")[1];
    return DefaultMethodResult.error.name() + "-" + branch;
}

这种方式既不破坏原有业务的错误返回逻辑（可按需调整），又能在测试时通过异常信息或返回值定位分支。

内容的提问来源于stack exchange，提问作者ProgrammerGnome