在 Junit，Mockito 中模拟局部变量

Question

我正在为使用 Mockito 的代码编写测试用例。代码是这样的，

public class Example {
 
 @Autowired
 private HelperService hs; 

 public void someMethod() {
   List<String> names = new ArrayList<>();
   hs.addValues(names);
   if(names.size() < 5) {throw new RuntimeException("Invalid names list");}
   return;
 }

}

public class HelperService {

 public void addValues(List<String> names) {
  names.add("alex");
  names.add("harry");
  names.add("james");
  names.add("maria");
  names.add("bob");
 }

}

我知道这是一个人为的例子，但我有一个类似的用例，我无法修改现有代码。我想使用 Junit、Mockito 对 Example 类进行单元测试。如何测试names.size() 的异常情况

Answer 1

我假设您已经有一种方法可以在您的测试中设置HelperService 字段（无论是通过将其设置为@SpringBootTest、使用@InjectMocks 还是其他方式）。所以这只会提供一个选项，让你可以做你想做的事：

1. 扩展 HelperService 并注入：

// Inject this into your class under test
private HelperService hs = new LessThan5NameAddingHelperService();

class LessThan5NameAddingHelperService extends HelperService {
    @Override
    public void addNames( List<String> names ) {
        names.add( "Dave" );
    }
}

1. 提供一个mock HelperService，它的作用相同：

// Inject this into your class under test
private HelperService hs;

@Before
public void setupMockHelperService() {
    hs = mock( HelperService.class );

    doAnswer( invocation -> {
        List<String> names = ( List<String> ) invocation.getArguments()[0];
        names.add( "Dave" );
        return null;
    } ).when( hs ).addNames( any() );
}