How to mock an interface with two different implementation [duplicate]

Question

I have a class I need to test that has two instance of different class but same interface. This is how the codes looks like,

Class to be tested:

@Service
public class MainClass {
   @Resource(name = "aClass")
   private IClass instance1;
   @Resource(name = "bClass")
   private IClass instance2;
}

Other classes:

@Service("aClass")
public class A implements IClass {}

@Service("bClass")
public class B implements IClass {}

My Unit Test:

public MainClassTest {
   @InjectMocks
   private MainClass mainClass;

   @Mock
   private IClass instance1;
   @Mock
   private IClass instance2;

   @Test
   public void test() {...}
}

When I run the test, both instance1 and instance2 are null since they are not mocked. This doesn't happen when the interface has only one implementation.

Any idea how to handle this?

Thanks, Angelo

Answer 1

@InjectMocks is a short cut which tells Mockito to instance your MainClass and to try to inject mocks by one of constructor injection or setter injection or property injection (in order, IIRC).

I think the behaviour you are seeing may be known behaviour for Mockito; there's an open issue describe here realting to Mockito's handling of @InjectMocks with two instances of the same mocked type.

So, instead of relying on it, you could create the instance of MainClass the 'old fashioned' way e.g. in a @Before method. For example:

private MainClass mainClass;

@Before
public void setup() {
    IClass instance1 = mock(IClass.class); 
    IClass instance2 = mock(IClass.class);

    mainClass = new MainClass(instance1, instance2);
}