Mockito：注入的模拟类，而不是为其他类注入的

Question

所以我有一个简单的java类：

@Component
public class IWantToTestThisClass {

@Autowired
private ClassA classA;

@Autowired 
private ClassB classB;

public ObjectA doIt(...) {
   classB.returnObject();
}

ClassA包含：

@Component
public class ClassA { 

@Autowired 
private ClassB classB;

public ObjectA doIt() {
   classB.returnObject(...);
}

B类：

@Component
public class ClassB { 

public ObjectA returnObject(...) {
   ...
}

我的测试设置如下所示：

@InjectMocks
private IWantToTestThisClass iWantToTestThisClass;

@Mock
private ClassA classA;

@Mock
private ClassB classB;

@Before
public void setUp() {
    initMocks(this);

    when(classA.doSmt()).thenCallRealMethod();
    when(classB.doSmt()).thenCallRealMethod();
}

问题是ClassB不是为ClassA注入的。

然后通过classA调用returnObject（ClassB），它返回NullPointerException（ClassB不是模拟/注入的）。

如果通过IWantToTestThisClass调用returObject（class B），则可以。

这是同时为两个类注入ClassB的一种方法吗？我想调用真实的方法，而不是模拟的方法。

Answer 1

@Autowired可以被构造函数注入替换。那么，你的课程A将是：

@Component
public class ClassA { 

    private static ClassB classB;

    public ClassA(ClassB classB){
        this.classB=classB;
    }

    public ObjectA doIt() {
       classB.returnObject(...);
    }
}

通过这种方式，您可以稍后在@Front setUp方法中注入class B mock：

@Before
public void setUp() {
    initMocks(this);
    when(classA.doSmt()).thenCallRealMethod();
    when(classB.doSmt()).thenCallRealMethod();
    classB = new classB(classA);
}

Answer 2

你把mock注入我的课堂，而不是A课堂。所以A类中没有模拟，B类中没有注入。

尝试在类A之前添加@InjectMocks

或者你可以这样做：

ClassB mB = Mockito.mock(ClassB.class);
classA.setClassB(mB);

如果添加setter setClassB（…）去ClassA