atsh

quill-finagle-postgres/src/test/scala/io/getquill/TestSkunk.scala

@@ -0,0 +1,79 @@
+package io.getquill
+
+import cats.effect.{IO, Resource}
+import com.twitter.finagle.{Postgres, Status}
+import com.twitter.finagle.postgres.{
+  OK,
+  Param,
+  PostgresClient,
+  PostgresClientImpl,
+  QueryResponse,
+  Row
+}
+import com.twitter.util.Future
+import com.typesafe.config.Config
+import weaver.*
+import skunk.*
+import skunk.codec.all.*
+import natchez.Trace.Implicits.noop
+
+import scala.util.Try
+
+object UtilsSkunk {
+  def getClient: Option[PostgresClient] =
+    for {
+      host <- sys.env.get("PG_HOST")
+      port = sys.env
+        .get("PG_PORT")
+        .flatMap { x =>
+          Try { x.toInt }.toOption
+        }
+        .getOrElse(5432)
+      user <- sys.env.get("PG_USER")
+      password = sys.env.get("PG_PASSWORD")
+      dbname <- sys.env.get("PG_DBNAME")
+    } yield {
+      new finagle_postgres.skunk.PostgresClientImpl(
+        Session.single(
+          host = host,
+          port = port,
+          user = user,
+          database = dbname,
+          password = password
+        )
+      )
+    }
+}
+
+object TestSkunk extends MutableTwitterFutureSuite {
+
+  override type Res = FinaglePostgresContext[SnakeCase.type]
+  override def sharedResource: Resource[IO, Res] =
+    Resource.make {
+      IO.fromOption(UtilsSkunk.getClient)(new Exception()).map { client =>
+        val ctx = new FinaglePostgresContext[SnakeCase.type](SnakeCase, client)
+        val _ = {
+          import ctx.*
+          ctx.run(sql"DROP TABLE IF EXISTS table_test".as[Insert[Unit]])
+          ctx.run(
+            sql"CREATE TABLE table_test (id integer, name text)"
+              .as[Insert[Unit]]
+          )
+        }
+        ctx
+      }
+    } { client =>
+      IO.blocking(client.close()).void
+    }
+
+  future("Can make insert and fetch") { ctx =>
+    import ctx.*
+    val results = ctx.transaction {
+      for {
+        _ <- ctx.run(query[TableTest].insertValue(TableTest(0, "hola")))
+        result <- ctx.run(query[TableTest].filter(_.id == 0))
+      } yield result
+    }
+    results.map(res => expect(res.head == TableTest(0, "hola")))
+  }
+}