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

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

class A{

    @Autowired
    private B b;
    
    @Autowired
    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);

但是这里有一点需要注意:如果A中使用了动态代理的注解或者配置,那么会导致mock失败,那是因为调用A的时候不是调用的目标对象,而是代理对象,目标对象是mock部分成功了,但是代理对象不会。

发布了81 篇原创文章 · 获赞 16 · 访问量 20万+

猜你喜欢

转载自blog.csdn.net/mazhen1991/article/details/99694160
今日推荐