mock private 成员变量

Question

环境

1. 测试用组件 - mockito

2. 测试框架 - junit4

问题点

如下类所示：

public class SomeService {
  @Autowire
  private SomeDAO dao;

  public void func（） {
    dao.addRecord();
    ...
  }
  ...
}

现在希望单元测试SomeService类，并不连接真实的数据库，而是对SomeDAO进行mock。

此时该如何将mock后的SomeDAO设置到SomeService对象中。

Answer 1

你需要创建一个构造器，或者一个setter，便于你的mock对象注入。

假设你有了一个构造器注入的接口，测试代码可以这样写

@RunWith(MockitoJUnitRunner.class)
public class SomeServiceTest {
  @Mock
  private SomeDao someDao;

  private SomeService someService;

  @Before
  public void setUp() throws Exception {
    someService = new SomeService(someDao);
  }

  @Test
  public void verify_someDao() throws Exception {
    ...
    verify(someDao).addRecord();
    ...
  }
}

Answer 2

@InjectMocks
使用这个会自动的把@Mock标记的类注入到里面去