java - 使用 PowerMockito 测试框架进行 junit 测试时如何将接口(interface)引用注入(inject) Spring Controller？

Question

我有 Spring Controller ，其中一个接口(interface)是 Autowiring 的，如下面的类所示。

@Controller
public class EmployeeController
{
    @Autowired
    private EmployeeManager employeeManager;

    public int saveEmployee(Employee employee) {
        int empId = employeeManager.saveEmployee(employee);
        return empId;
    }
}

现在，我创建了一个 Junit 测试类，如下所示。

public class EmployeeControllerTest {

    private EmployeeController employeeController ; 
    private EmployeeManager employeeManager ;

    @Before
    public void setUp() throws Exception {
        employeeController = PowerMockito.spy(new EmployeeController());
        employeeManager = PowerMockito.mock(EmployeeManager.class);
    }

    @Test
    public void shouldGetCountOfEmployees()
    { 
        Employee emp = new Employee("Peter");
        int empId = employeeController.saveEmployee(emp);   
        Assert.assertEquals(1, empId);
    }

现在执行测试类后，我在 Controller 的 saveEmployee 方法内的 int empId = employeeManager.saveEmployee(employee); 行收到 nullPointerException。

我也尝试过@Mock注释，@Spy注释但无法执行测试用例。因此，请在使用 PowerMockito 执行测试用例时帮助在 Controller 中注入(inject) employeeManager 实例。

谢谢

Answer 1

您必须使用 Whitebox.setInternalState 将模拟注入(inject)到 EmployeeController 中。

示例:

Whitebox.setInternalState(EmployeeController, "employeeManager", employeeManager) ;