Tengo algo como esto:
public class Class1 { @Autowired Class2 class2; public Object class1Method { // Do something Object obj = class2.class2Method(arg); // Do something with obj return somthing; } }Estoy haciendo pruebas unitarias para Class1 y quiero simular class2Method. Traté de encontrar una solución y encontré esto:
Clase de prueba con una llamada new() con Mockito
Entendí que esta solución funciona si hago una nueva instancia de Class2 en class1Method, pero en mi caso uso Spring Dependency Injection y no puedo entender cómo hacerlo.
Dependiendo de cómo haya construido su DI con Spring, puede usar algo como esto si tiene @Autowired en el campo
@RunWith(SpringRunner.class) public Class1Test { @InjectMocks @Autwired private Class1 class1; @Mock Class2 class2; @BeforeMethod public void initMocks(){ MockitoAnnotations.initMocks(this); } @Test public Object testClass1Method { when(class2.class2Method(anyString()).thenReturn(expectedReturn); //Now test whatever you need with mocked class2.class2Method } } Es mucho más fácil y no necesita tener un contexto de primavera para las pruebas si tiene un constructor DI ( @Autowired en el constructor)
public Class1Test { private Class1 class1; private Class2 class2; @BeforeMethod public void initMocks(){ class2 = mock(Class2.class); class1 = new Class1(class2); } //then use the same approach to mock any methods that you need }deberías ir con
Mockito
, y haga que su clase JUnit incluya esto en la parte superior @RunWith(MockitoJUnitRunner.class) . Así que en tu caso puedes tener tu Junit así.
@RunWith(MockitoJUnitRunner.class) public Class1Test { @InjectMocks private Class1 class1 = new Class1(); @Mock Class2 class2; @Test public Object testClass1Method { // Now this will get the class2 mock instance when executed Object obj = class1.class1Method(arg); } }Así es como puede escribir una prueba unitaria para Class1 usando Mockito :
public Class1Test { @InjectMocks private Class1 class1; @Mock private Class2 class2; @BeforeMethod public void initMocks(){ MockitoAnnotations.initMocks(this); } @Test public void test_method1() { Object obj = //instantiate it here..., Mockito.doReturn(obj).when(class2.class2Method(anyString()); Assert.assertEquals(class1.class1Method(), something); } }