tech_blog_20240314_SingletonMySQLContainerSpec.scala

package example

import cats.effect.{IO, Resource}
import cats.effect.testing.scalatest.AsyncIOSpec
import com.dimafeng.testcontainers.MySQLContainer
import doobie.util.transactor
import example.db.Database
import org.scalatest.BeforeAndAfterAll
import org.scalatest.freespec.AsyncFreeSpec
import org.testcontainers.containers.JdbcDatabaseContainer
import org.testcontainers.ext.ScriptUtils
import org.testcontainers.jdbc.JdbcDatabaseDelegate

trait SingletonMySQLContainerSpec extends AsyncFreeSpec with AsyncIOSpec with BeforeAndAfterAll:

  // テスト開始時に実行するSQLファイルのリスト
  val fixturePathList: Seq[String]

  // シングルトンオブジェクトのコンテナを提供する
  final def mysqlContainer: MySQLContainer =
    SingletonMySQLContainer.mysqlContainer

  private val jdbcContainer: JdbcDatabaseContainer[Nothing] =
    mysqlContainer.container.asInstanceOf[JdbcDatabaseContainer[Nothing]]

  private val databaseDelegate: JdbcDatabaseDelegate =
    new JdbcDatabaseDelegate(jdbcContainer, "")

  final val customTransactor: Resource[IO, transactor.Transactor[IO]] =
    Database.transactor(
      "com.mysql.cj.jdbc.Driver",
      s"${jdbcContainer.getJdbcUrl}?useSSL=false",
      "root",
      ""
    )

  override def beforeAll(): Unit =
    super.beforeAll()
    // 各テストスイート実行前にDBを初期化する
    ScriptUtils.runInitScript(databaseDelegate, "fixtures/cleanup.sql")
    // テストデータを投入する
    fixturePathList.foreach { path =>
      ScriptUtils.runInitScript(databaseDelegate, path)
    }