HikariCP - 如何为单元测试配置模拟数据源

时间:2018-05-24 08:17:06

标签: java spring hikaricp

我有某种集成测试可以启动我的服务,但不会进行实际的数据库调用。

我得到了这样的例外:

[      main] 11:04:47,368 ERROR HikariPool:566 - HikariPool-1 - 
Exception during pool initialization.
com.mysql.cj.jdbc.exceptions.CommunicationsException: Communications link failure

我的测试没问题(它正在通过)。但是,我不希望在日志中出现该错误,这会使开发人员在测试失败时感到困惑 我正在使用弹簧配置。我可以配置某种模拟数据源吗?

1 个答案:

答案 0 :(得分:0)

我正在为此使用kotlin,spring boot和io.mockk:mockk。 在集成测试中,您可以通过将上下文配置放在顶部来指定上下文配置

@ContextConfiguration(classes = [DBConfiguration::class])

上述类是您的模拟数据库连接所在的位置。

您的DBConfiguration类看起来像这样

import com.zaxxer.hikari.HikariDataSource
import io.mockk.every
import io.mockk.mockk
import org.springframework.context.annotation.Bean
import org.springframework.context.annotation.Primary
import java.sql.Connection
import java.sql.DatabaseMetaData

/**
 * Custom configuration class used for testing and disables database connection and initial connection verification
 */
class DBConfiguration {
    @Primary
    @Bean
    fun hikariDataSource(): HikariDataSource {
        val connection: Connection = mockk()
        val databaseMetadata: DatabaseMetaData = mockk()
        val hikariDataSource: HikariDataSource = mockk()
        hikariDataSource
        every { hikariDataSource.connection }.returns(connection)
        every { hikariDataSource.connectionTestQuery }.returns("yeet")
        every { hikariDataSource.metricRegistry }.returns(Any())
        every { hikariDataSource.maximumPoolSize }.returns(2)
        every { hikariDataSource.minimumIdle }.returns(2)
        every { connection.metaData }.returns(databaseMetadata)
        return hikariDataSource
    }
}