From 4fa8181ff3955e46fe5426b2cf1d07c78fb5d21b Mon Sep 17 00:00:00 2001
From: Adam Hendel <adam@tembo.io>
Date: Mon, 21 Aug 2023 16:36:02 -0500
Subject: [PATCH] remove erroneous semicolon

---
 core/src/query.rs | 14 ++++++++++++--
 1 file changed, 12 insertions(+), 2 deletions(-)

diff --git a/core/src/query.rs b/core/src/query.rs
index 33607e15..f5c8087f 100644
--- a/core/src/query.rs
+++ b/core/src/query.rs
@@ -336,7 +336,7 @@ pub fn assign_queue(name: CheckedName<'_>) -> Result<String, PgmqError> {
 }
 
 pub fn assign_archive(name: CheckedName<'_>) -> Result<String, PgmqError> {
-    Ok(assign(&format!("{name}_archive; ")))
+    Ok(assign(&format!("{name}_archive")))
 }
 
 pub fn unassign_queue(name: CheckedName<'_>) -> Result<String, PgmqError> {
@@ -362,7 +362,11 @@ pub fn assign(table_name: &str) -> String {
             SELECT 1
             FROM pg_depend
             WHERE refobjid = (SELECT oid FROM pg_extension WHERE extname = 'pgmq')
-            AND objid = (SELECT oid FROM pg_class WHERE relname = '{TABLE_PREFIX}_{table_name}')
+            AND objid = (
+                SELECT oid
+                FROM pg_class
+                WHERE relname = '{TABLE_PREFIX}_{table_name}'
+            )
         ) THEN
             EXECUTE 'ALTER EXTENSION pgmq ADD TABLE {PGMQ_SCHEMA}.{TABLE_PREFIX}_{table_name}';
         END IF;
@@ -402,6 +406,12 @@ pub fn check_input(input: &str) -> Result<(), PgmqError> {
 mod tests {
     use super::*;
 
+    #[test]
+    fn test_assign() {
+        let query = assign("my_queue_archive");
+        assert!(query.contains("WHERE relname = 'pgmq_my_queue_archive'"));
+    }
+
     #[test]
     fn test_create() {
         let queue_name = CheckedName::new("yolo").unwrap();