从零学springboot—— 如何在junit测试中只mock一部分类

瘦欲@ 提交于 2019-11-27 16:06:27

假设有一个类A,我们要对其进行测试:

class A{
    
    private B b;
    
    private C c;

    @Value("${str}")
    private String str;
}

如果我们想要mock调对B和C的调用,那么我们在测试类中只要这样写即可:

@InjectMocks
private A a;

@Mock
private B b;

@Mock
private C c;

@Test
public void test(){

    Mockito.when(b.b的方法(Mockito.any()))
                .thenReturn(b方法的返回);
}

通过上面的代码,我们就能mock调B的方法,并按我们的要求返回结果,但是这么做,C也会被mock调,并且@Value的str也会是空值,如果我们想要只Mock调对B的调用,我们可以这么做:

@Autowired
@InjectMocks
private A a;

@Mock
private B b;

@Autowired
private C c;

@Before
public void setUp(){
    MockitoAnnotations.initMocks(this);
}

@Test
public void test(){

    Mockito.when(b.b的方法(Mockito.any()))
                .thenReturn(b方法的返回);
}

我们只需要在注入的测试类上面在加上一个@Autowired注解,并对无需mock的类使用@Autowired,关键的是需要加上MockitoAnnotations.initMocks(this);

标签
易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!