回顾

增加Service层代码

首先定义一个接口，在我们简单项目中是否定义接口其实并不重要，但这依然是个好习惯。

public interface IDemoService {
    List<Demo> findAll();
}

增加DemoService实现层代码

@Service
public class DemoService implements IDemoService{
    @Override
    public List<Demo> findAll() {
        return null;
    }
}

Controller层调用业务逻辑

接下来我们要在Controller层调用demoService

@Controller
@RequestMapping("demos")
public class DemoController {

    private IDemoService demoService;

        @Autowired
        public DemoController(IDemoService demoService) {
                this.demoService = demoService;
        }

    @GetMapping
    public ResponseEntity<List> searchDemo() {
        List<Demo> demoList = demoService.findAll();
        return new ResponseEntity<>(demoList, HttpStatus.OK);
    }
}

修改单元测试

修改完业务代码之后直接运行原来的测试代码，会发现抛出NullPointException

这是因为测试的searchDemo()方法需要调用demoService对象，而在我们的测试类中没有创建demoService对象。

@RunWith(MockitoJUnitRunner.class)
public class MockDemoControllerTest {

    private MockMvc mockMvc;

    @InjectMocks
    private DemoController demoController;
    @Mock // 创建一个模拟的demoService对象
    private DemoService demoService;

​

    @Before
    public void setUp() throws Exception {
        mockMvc = MockMvcBuilders.standaloneSetup(demoController).build();
    }

    @Test
    public void should_get_demos() throws Exception {
        mockMvc.perform(get("/demos"))
                .andExpect(status().isOk());
    }

}

修改的测试代码很简单，只增加了@Mock private DemoService demoService。
执行修改后的测试代码，将会看到我们的测试结果再次变绿。

发生了什么

相比较与前一个版本，发生了什么改变呢？
首先我们测试的类DemoController多了一个依赖的对象demoService，并且在测试的方法中还需要调用demoService的findAll()方法。但在原来的测试中，demoController对象中并没有找到所的demoService对象。
为什么增加@Mock private DemoService demoService之后就可以正常运行了？
关键要看使用@InjectMocks与@Mock这两个注解后发生了什么。

@InjectMocks与@Mock

private DemoService demoService1 = new DemoService();

/**
* 可以简单的理解为 增加这个注解后
* 执行了 private DemoService demoService2 = new DemoService();
*/
@Mock 
private DemoService demoService2;

/**
* @InjectMocks注解可以理解为
* 执行了 private DemoController demoController = new DemoController(demoService2);
*/
@InjectMocks
private DemoController demoController;

也就是说，MockitoJUnitRunner执行器在运行前会检测到使用@InjectMocks和@Mock的成员变量，并自动为其创建对象。

总结

本文在创建REST接口的基础上增加了业务逻辑，更加贴近实际生产环境。针对代码结构改变，相应改变了测试代码，解决待测试代码的依赖问题。
解释了测试代码中用到的两个注解作用，但没有对其深入展开。
下一篇我将继续介绍如何测试Controller代码，并介绍一下其他的测试用法。