Merge pull request #855 from typelevel/set-constraints

completion for `set constraints`
typelevel · Apr 26, 2023 · be3b06c · be3b06c
2 parents d43df5c + a680c60
commit be3b06c
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 0 deletions.
diff --git a/modules/core/shared/src/main/scala/data/Completion.scala b/modules/core/shared/src/main/scala/data/Completion.scala
@@ -59,6 +59,7 @@ object Completion {
   case object CreateTrigger             extends Completion
   case object AlterTrigger              extends Completion
   case object DropTrigger               extends Completion
+  case object SetConstraints            extends Completion
   // more ...
 
   /**

diff --git a/modules/core/shared/src/main/scala/net/message/CommandComplete.scala b/modules/core/shared/src/main/scala/net/message/CommandComplete.scala
@@ -97,6 +97,7 @@ object CommandComplete {
     case Patterns.Update(s)           => apply(Completion.Update(s.toInt))
     case Patterns.Insert(s)           => apply(Completion.Insert(s.drop(2).toInt))
     case Patterns.Copy(s)             => apply(Completion.Copy(s.toInt))
+    case "SET CONSTRAINTS"            => apply(Completion.SetConstraints)
     // more .. fill in as we hit them
 
     case s                  => apply(Completion.Unknown(s))

diff --git a/modules/tests/shared/src/test/scala/CommandTest.scala b/modules/tests/shared/src/test/scala/CommandTest.scala
@@ -430,6 +430,15 @@ class CommandTest extends SkunkTest {
     } yield "ok"
   }
 
+  sessionTest("set constraints") { s =>
+    s.transaction.use { _ =>
+      for {
+        c <- s.execute(sql"set constraints all deferred".command)
+        _ <- assert("completion", c == Completion.SetConstraints)
+      } yield "ok"
+    } >> s.assertHealthy
+  }
+
   sessionTest("insert, update and delete record") { s =>
     for {
       c <- s.execute(insertCity, Garin)