提问者:小点点

单元测试中的Spring启动数据源


我有一个简单的Spring启动Web应用程序,它从数据库中读取并返回JSON响应。我有以下测试配置:

@RunWith(SpringRunner.class)
@SpringBootTest(classes=MyApplication.class, properties={"spring.config.name=myapp"})
@AutoConfigureMockMvc
public class ControllerTests {
    @Autowired
    private MockMvc mvc;
    @MockBean
    private ProductRepository productRepo;
    @MockBean
    private MonitorRepository monitorRepo;

    @Before
    public void setupMock() {
        Mockito.when(productRepo.findProducts(anyString(), anyString()))
        .thenReturn(Arrays.asList(dummyProduct()));     
    }

    @Test
    public void expectBadRequestWhenNoParamters() throws Exception {    
        mvc.perform(get("/products"))
                .andExpect(status().is(400))
                .andExpect(jsonPath("$.advice.status", is("ERROR")));
    }

    //other tests
}

我有一个数据源 Bean,它是在应用程序的主配置中配置的。当我运行测试时,Spring 尝试加载上下文并失败,因为数据源取自 JNDI。一般来说,我想避免为此测试创建数据源,因为我有模拟的存储库。

是否可以在运行单元测试时跳过数据源的创建?

在内存中测试数据库不是一个选项,因为我的数据库创建脚本具有特定的结构,并且无法从classpath:schema轻松执行.sql

编辑 数据源在“我的应用程序”中定义.class

    @Bean
    DataSource dataSource(DatabaseProeprties databaseProps) throws NamingException {
       DataSource dataSource = null;
       JndiTemplate jndi = new JndiTemplate();
       setJndiEnvironment(databaseProps, jndi);
       try {
           dataSource = jndi.lookup(databaseProps.getName(), DataSource.class);
       } catch (NamingException e) {
           logger.error("Exception loading JNDI datasource", e);
           throw e;
       }
       return dataSource;
   }

共2个答案

匿名用户

由于您正在加载配置类 MyApplication.class数据源 Bean,尝试在另一个未在测试中使用的 Bean 中移动数据源,请确保为测试加载的所有类都不依赖于数据源。
或者在
测试中创建一个标有@TestConfiguration的配置类,并将其包含在Spring启动(类=TestConfig.class)中模拟数据源,例如

@Bean
public DataSource dataSource() {
    return Mockito.mock(DataSource.class);
}

但是这可能会失败,因为对这个模拟数据的方法调用连接将返回null,在这种情况下,您必须在内存中创建一个数据源,然后模拟jdbcTemplate和其余依赖项。

匿名用户

尝试将数据源添加为@MockBean

@MockBean
private DataSource dataSource

这样,Spring将为您完成替换逻辑,其优点是您的生产代码bean创建甚至不会执行(无JNDI查找)。