如何模拟 @Autowire 对象的依赖项？

Question

我有一个 @Autowire 对象，其中包含需要模拟其方法调用的字段。

在主类中：

@Component
public class Pizza {
     private Tomato tomato;
     private Cheese cheese;

     @Autowired
     private Pizza(Tomato tomato, Cheese cheese) {
        this.tomato = tomato;
        this.cheese = cheese;
     }
     
     public String arrangePizza(tomato, cheese) {
        Sauce sauce = tomato.createSauce();
        combine(sauce, cheese);
        return "Pizza created!"
     }
}

在测试类中：

@RunWith(SpringRunner.class)
public class TestPizza {
     @Autowire
     private Pizza pizza;

     //probably create instances of cheese and tomato here?


     private void testCreatePizza {
        //here I want to mock tomato.createSauce()
        pizza.arrangePizza(tomato, cheese);
     }
}

我正在尝试使用 Mockito 或 EasyMock 在 testCreatePizza 中模拟方法tomato.createSauce()，但鉴于 Pizza 是自动装配的，我不确定如何执行此操作。我是否必须在测试类中创建 tomato 和 cheese 的 Autowire 实例？ spring 会自动知道将构造函数设置为那些实例吗？

Answer 1

Mockito 提供 @Mock 注释来模拟对象，并提供 @InjectMocks 注释来将这些模拟注入到自动装配的字段中。

@RunWith(SpringRunner.class)
@ExtendWith(MockitoExtension.class)
public class TestPizza {

     @Mock
     private Tomato tomato;

     @Mock
     private Cheese cheese;        

     @InjectMocks
     private Pizza pizza;

     @BeforeEach
     void init() {
         MockitoAnnotations.initMocks(this);
         when(tomato.createSauce()).thenReturn("Spicy Sauce");
     }

     @Test
     private void testCreatePizza {
        pizza.createPizza(tomato, cheese);
     }
}

Answer 2

由于这是标记为 spring-boot，因此还值得指出 @MockBean annotation。 （相关引用：“在上下文中定义的任何现有的相同类型的单个 bean 都将被模拟替换。如果没有定义现有的 bean，将添加一个新的。”）

这意味着，在你的测试课中，你可以这样做

@Autowired
private Pizza pizza;

@MockBean
private Tomato tomato;

然后像平常一样使用 Mockito 的 when 等等。与其他答案相比，这种方式可以为您节省一两个注释（如果您正在模拟多个事物），以及对 initMocks() 的调用。