Merge branch 'series/0.6.x'

typelevel · Oct 27, 2023 · 6cd0a53 · 6cd0a53
2 parents 53b79c2 + 314a319
commit 6cd0a53
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 4 deletions.
diff --git a/modules/core/shared/src/main/scala/data/Completion.scala b/modules/core/shared/src/main/scala/data/Completion.scala
@@ -64,6 +64,7 @@ object Completion {
   case object Explain                   extends Completion
   case object Grant                     extends Completion
   case object Revoke                    extends Completion
+  case object AlterIndex                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
@@ -102,6 +102,7 @@ object CommandComplete {
     case "EXPLAIN"                    => apply(Completion.Explain)
     case "GRANT"                      => apply(Completion.Grant)
     case "REVOKE"                     => apply(Completion.Revoke)
+    case "ALTER INDEX"                => apply(Completion.AlterIndex)
     // 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
@@ -143,9 +143,14 @@ class CommandTest extends SkunkTest {
       )
       """.command
 
+  val alterIndex: Command[Void] =
+    sql"""
+      ALTER INDEX IF EXISTS id_index RENAME TO pk_index
+      """.command
+
   val dropIndex: Command[Void] =
     sql"""
-      DROP INDEX id_index
+      DROP INDEX pk_index
       """.command
 
   val createView: Command[Void] =
@@ -329,16 +334,18 @@ class CommandTest extends SkunkTest {
         FROM skunk_role
        """.command
 
-  sessionTest("create table, create index, drop index, alter table and drop table") { s =>
+  sessionTest("create table, create index, alter table, alter index, drop index and drop table") { s =>
     for {
       c <- s.execute(createTable)
       _ <- assert("completion",  c == Completion.CreateTable)
       c <- s.execute(createIndex)
       _ <- assert("completion",  c == Completion.CreateIndex)
-      c <- s.execute(dropIndex)
-      _ <- assert("completion",  c == Completion.DropIndex)
       c <- s.execute(alterTable)
       _ <- assert("completion",  c == Completion.AlterTable)
+      c <- s.execute(alterIndex)
+      _ <- assert("completion", c == Completion.AlterIndex)
+      c <- s.execute(dropIndex)
+      _ <- assert("completion",  c == Completion.DropIndex)
       c <- s.execute(dropTable)
       _ <- assert("completion",  c == Completion.DropTable)
       _ <- s.assertHealthy