generated from navikt/helse-sparkel-sykepengeperioder
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Co-authored-by: Håkon Arneng Holmstedt <hakon.arneng.holmstedt@nav.no>
- Loading branch information
1 parent
f9f986f
commit 0188fd2
Showing
3 changed files
with
16 additions
and
134 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -1,126 +1,6 @@ | ||
| package no.nav.helse.spokelse | ||
|
|
||
| import com.zaxxer.hikari.HikariConfig | ||
| import com.zaxxer.hikari.HikariDataSource | ||
| import kotliquery.queryOf | ||
| import kotliquery.sessionOf | ||
| import org.flywaydb.core.Flyway | ||
| import org.intellij.lang.annotations.Language | ||
| import org.testcontainers.containers.PostgreSQLContainer | ||
| import java.time.Duration | ||
| import javax.sql.DataSource | ||
| import com.github.navikt.tbd_libs.test_support.CleanupStrategy | ||
| import com.github.navikt.tbd_libs.test_support.DatabaseContainers | ||
|
|
||
| internal object PgDb { | ||
|
|
||
| private var state: DBState = NotStarted | ||
| private val postgres = PostgreSQLContainer<Nothing>("postgres:14") | ||
| private lateinit var dataSource: DataSource | ||
|
|
||
| fun start(): PgDb { | ||
| state.start(this) | ||
| return this | ||
| } | ||
|
|
||
| fun reset() { | ||
| state.reset(this) | ||
| } | ||
|
|
||
| fun hardReset() { | ||
| state.hardReset(this) | ||
| } | ||
|
|
||
| fun config() = state.config(this) | ||
| fun connection() = state.connection(this) | ||
| private val DataSource.flyway get() = Flyway.configure().dataSource(this).cleanDisabled(false).load() | ||
|
|
||
| private fun stop(): PgDb { | ||
| state.stop(this) | ||
| return this | ||
| } | ||
|
|
||
| private fun startDatbase() { | ||
| postgres.start() | ||
| dataSource = HikariDataSource(config()) | ||
| createSchema(connection()) | ||
| Runtime.getRuntime().addShutdownHook(Thread(this::stop)) | ||
| } | ||
|
|
||
| private fun createSchema(dataSource: DataSource) { | ||
| dataSource.flyway.migrate() | ||
| sessionOf(dataSource).use { it.run(queryOf(truncateTablesSql).asExecute) } | ||
| } | ||
|
|
||
| private fun resetSchema() { | ||
| sessionOf(connection()).use { it.run(queryOf("SELECT truncate_tables();").asExecute) } | ||
| } | ||
|
|
||
| private fun stopDatabase() { | ||
| postgres.stop() | ||
| } | ||
|
|
||
| private interface DBState { | ||
| fun config(db: PgDb): HikariConfig { | ||
| throw IllegalStateException("Cannot create config in state ${this::class.simpleName}") | ||
| } | ||
| fun connection(db: PgDb): DataSource { | ||
| throw IllegalStateException("Cannot create connection in state ${this::class.simpleName}") | ||
| } | ||
| fun start(db: PgDb) {} | ||
| fun stop(db: PgDb) {} | ||
| fun reset(db: PgDb) {} | ||
| fun hardReset(db: PgDb) {} | ||
| } | ||
|
|
||
| private object NotStarted : DBState { | ||
| override fun start(db: PgDb) { | ||
| state = Started | ||
| db.startDatbase() | ||
| } | ||
| } | ||
|
|
||
| private object Started : DBState { | ||
| override fun stop(db: PgDb) { | ||
| db.state = NotStarted | ||
| db.stopDatabase() | ||
| } | ||
|
|
||
| override fun config(db: PgDb) = HikariConfig().apply { | ||
| jdbcUrl = db.postgres.jdbcUrl | ||
| username = db.postgres.username | ||
| password = db.postgres.password | ||
| maximumPoolSize = 2 | ||
| connectionTimeout = Duration.ofSeconds(5).toMillis() | ||
| initializationFailTimeout = Duration.ofMinutes(1).toMillis() | ||
| } | ||
|
|
||
| override fun connection(db: PgDb): DataSource { | ||
| return db.dataSource | ||
| } | ||
|
|
||
| override fun reset(db: PgDb) { | ||
| db.resetSchema() | ||
| } | ||
|
|
||
| override fun hardReset(db: PgDb) { | ||
| val ds = connection() | ||
| ds.flyway.clean() | ||
| createSchema(ds) | ||
| } | ||
|
|
||
| } | ||
|
|
||
| @Language("PostgreSQL") | ||
| private val truncateTablesSql = """ | ||
| CREATE OR REPLACE FUNCTION truncate_tables() RETURNS void AS ${'$'}${'$'} | ||
| DECLARE | ||
| statements CURSOR FOR | ||
| SELECT tablename FROM pg_tables | ||
| WHERE schemaname = 'public' AND tablename NOT LIKE 'flyway%'; | ||
| BEGIN | ||
| FOR stmt IN statements LOOP | ||
| EXECUTE 'TRUNCATE TABLE ' || quote_ident(stmt.tablename) || ' CASCADE;'; | ||
| END LOOP; | ||
| END; | ||
| ${'$'}${'$'} LANGUAGE plpgsql; | ||
| """ | ||
| } | ||
| val databaseContainer = DatabaseContainers.container("spokelse", CleanupStrategy.tables("alle_annulleringer, annullering, gamle_utbetalinger, hendelse, old_utbetaling, old_vedtak, oppdrag, tbdutbetaling_melding, tbdutbetaling_utbetaling, tbdutbetaling_utbetalingslinje, utbetaling, vedtak, vedtak_utbetalingsref")) |