multi-statement with extended protocol

NocturnalGlory · Jul 12, 2020 · be5cfcc · be5cfcc
1 parent 71e560d
commit be5cfcc
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/modules/tests/src/test/scala/MultipleStatementsTest.scala b/modules/tests/src/test/scala/MultipleStatementsTest.scala
@@ -9,6 +9,8 @@ import skunk._
 import skunk.implicits._
 import skunk.codec.all._
 import skunk.exception.SkunkException
+import cats.effect.IO
+import skunk.exception.PostgresErrorException
 
 object MultipleStatementsTest extends SkunkTest {
 
@@ -24,4 +26,16 @@ object MultipleStatementsTest extends SkunkTest {
     sessionTest(s"command: ${c.sql}") { s =>  s.execute(c).assertFailsWith[SkunkException] *> s.assertHealthy }
   }
 
+  sessionTest("extended query (postgres raises an error here)") { s =>
+    s.prepare(sql"select 1;commit".query(int4))
+      .use(_ => IO.unit)
+      .assertFailsWith[PostgresErrorException] *> s.assertHealthy
+  }
+
+  sessionTest("extended command (postgres raises an error here)") { s =>
+    s.prepare(sql"select 1;commit".command)
+      .use(_ => IO.unit)
+      .assertFailsWith[PostgresErrorException] *> s.assertHealthy
+  }
+
 }