fix(sql): represent array literals correctly

diff --git a/psycopg/psycopg/sql.py b/psycopg/psycopg/sql.py
index f2b0ac1ccbce3538ecd9e1b634636f2b21cc0eca..9726af8b5f342808e145413ef7c9e722cfdeb32f 100644 (file)
--- a/psycopg/psycopg/sql.py
+++ b/psycopg/psycopg/sql.py
@@ -406,6 +406,8 @@ class Literal(Composable):
                 except KeyError:
                     type_name = ti.regtype.encode(tx.encoding)
                     self._names_cache[ti.regtype, tx.encoding] = type_name
+                if dumper.oid == ti.array_oid:
+                    type_name += b"[]"
                 rv = b"%s::%s" % (rv, type_name)
         return rv
 

diff --git a/tests/test_sql.py b/tests/test_sql.py
index d8ae837035f70de2cab5651daed62501ca2ae115..21d0018c5707ad2c1d297a683a1a8f21edf03bd6 100644 (file)
--- a/tests/test_sql.py
+++ b/tests/test_sql.py
@@ -339,6 +339,12 @@ class TestLiteral:
         with pytest.raises(ProgrammingError):
             sql.Literal(Foo()).as_string(conn)
 
+    def test_array(self, conn):
+        assert (
+            sql.Literal([dt.date(2000, 1, 1)]).as_string(conn)
+            == "'{2000-01-01}'::date[]"
+        )
+
     @pytest.mark.parametrize("name", ["a-b", f"{eur}", "order"])
     def test_invalid_name(self, conn, name):
         conn.execute(
@@ -368,6 +374,12 @@ class TestLiteral:
         cur = conn.execute(sql.SQL("select {}").format("hello"))
         assert cur.fetchone()[0] == "hello-inv"
 
+        assert (
+            sql.Literal(["hello"]).as_string(conn) == f"'{{hello-inv}}'::\"{name}\"[]"
+        )
+        cur = conn.execute(sql.SQL("select {}").format(["hello"]))
+        assert cur.fetchone()[0] == ["hello-inv"]
+
 
 class TestSQL:
     def test_class(self):