Spring Boot：在 ApplicationRunner 中注入模拟

Question

我正在尝试进行集成测试，其中包括 ApplicationRunner 的执行。

我用的是@SpringBootTest，在上下文初始化的时候，ApplicationRunner自动启动，还可以。

我想要实现的是在 ApplicationRunner 中注入一个 Mock 并在 ApplicationRunner 执行之前配置该模拟。

如果我在测试中配置mock是不行的，因为到执行测试的时候，Spring上下文已经被初始化并且ApplicationRunner已经被执行了。

有没有办法配置模拟在 Spring 上下文已初始化，之前 ApplicationRunner 被执行？

Answer 1

如果您的 ApplicationRunner 类有 bean 依赖项，您可以按如下方式模拟它。

import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.boot.test.mock.mockito.MockBean;
import org.junit.Before;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.test.context.junit4.SpringRunner;

@RunWith(SpringRunner.class)
@SpringBootTest   
public class ApplicationRunnerTest {
    @MockBean
    private final Dependency dependency;
    @Autowired
    private final ApplicationRunner applicationRunner;

    @Before
    public void setUp() throws Exception {
    }

    @Test
    public void testMethod() {
    }
}

Answer 2

希望这能回答您的问题

当我们尝试执行一个测试用例时，这里的 webApplicationContext 会被初始化为 @Before 被启动，它调用了具有 webApplicationContext 初始化逻辑的 AbstractTest 类的 setUp()

@RunWith(SpringJUnit4ClassRunner.class)
@SpringBootTest(classes = Main.class)
@WebAppConfiguration
public abstract class AbstractTest {

   protected MockMvc mvc;
   @Autowired
   WebApplicationContext webApplicationContext;

   protected void setUp() {
      mvc = MockMvcBuilders.webAppContextSetup(webApplicationContext).build();
   }

   protected String mapToJson(Object obj) throws JsonProcessingException {
      ObjectMapper objectMapper = new ObjectMapper();
      return objectMapper.writeValueAsString(obj);
   }
   protected <T> T mapFromJson(String json, Class<T> clazz)
      throws JsonParseException, JsonMappingException, IOException {

      ObjectMapper objectMapper = new ObjectMapper();
      return objectMapper.readValue(json, clazz);
   }
}


public class UserControllerTest extends AbstractTest {
   @Override
   @Before
   public void setUp() {
      super.setUp();
   }
   @Test
   public void testGet() throws Exception {
      String uri = “/url”;
      MvcResult mvcResult = mvc.perform(MockMvcRequestBuilders.get(uri)
         .accept(MediaType.APPLICATION_JSON_VALUE)).andReturn();

      int status = mvcResult.getResponse().getStatus();
      assertEquals(200, status);
   }

}

主 Spring 引导类

@SpringBootApplication
public class Main extends SpringBootServletInitializer{

    private Logger LOGGER = (Logger) LoggerFactory.getLogger(FactsMain.class);

    @Value("${facts.trustCertPath}")
    private String trustCertPath;


    public static void main(String args[]) {
        SpringApplication.run(Main.class, args);
    }

    @Override
     protected SpringApplicationBuilder configure(SpringApplicationBuilder application) {
      return application.sources(FactsMain.class);
     }

    @PostConstruct
    public void setSSLPath() {

        LOGGER.info("trustCertPath - " + trustCertPath);
        System.setProperty("javax.net.ssl.trustStore", trustCertPath);
    }
}