java - 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();
    ...
  }
}