SlickZIOSPec.scala

package slickzio

import org.scalatest.{FlatSpec, Matchers}
import slick.basic.BasicBackend
import slick.dbio.DBIO
import slick.jdbc.{H2Profile, JdbcProfile}
import zio.internal.PlatformLive
import zio.{Runtime, ZIO}

class SlickZioSpec extends FlatSpec with Matchers {
  implicit val profile: JdbcProfile = H2Profile

  val action: DBIO[Int] = {
    import profile.api._
    sql"select 1".as[Int].head
  }

  type SlickZIO[T] = ZIO[BasicBackend#DatabaseDef, Throwable, T]

  object SlickZIO {
    def apply[T](action: DBIO[T]): SlickZIO[T] = {
      for {
        db <- ZIO.environment[BasicBackend#DatabaseDef]
        res <- ZIO.fromFuture(implicit executionContext => db.run(action))
      } yield res
    }
  }

  implicit def DBIO2SlickZIO[T](dbio: DBIO[T]): SlickZIO[T] = SlickZIO(dbio)

  "all manual: simple select using zio" should "also work" in {
    import profile.api._

    val dbRuntime =
      Runtime(Database.forDriver(org.h2.Driver.load(), "jdbc:h2:mem:"),
              PlatformLive.Default)
    dbRuntime.unsafeRun(action) shouldBe 1
  }
}