]> git.ipfire.org Git - thirdparty/psycopg.git/commitdiff
test(enum): more idiomatic sql composition
authorDaniele Varrazzo <daniele.varrazzo@gmail.com>
Sun, 17 Apr 2022 22:44:43 +0000 (00:44 +0200)
committerDaniele Varrazzo <daniele.varrazzo@gmail.com>
Thu, 21 Apr 2022 14:05:46 +0000 (16:05 +0200)
tests/types/test_enum.py

index 78b9e0580ae8753375cbee802bad7aab7ebc87bb..128afa760fd10748c7d40e2ff87190733539bad8 100644 (file)
@@ -29,14 +29,14 @@ encodings = ["utf8", "latin1"]
 @pytest.fixture(scope="session", params=enum_cases)
 def testenum(request, svcconn):
     name, enum, labels = request.param
-    quoted_labels = [sql.quote(label) for label in labels]
-
     cur = svcconn.cursor()
     cur.execute(
-        f"""
-        drop type if exists {name} cascade;
-        create type {name} as enum({','.join(quoted_labels)});
-        """
+        sql.SQL(
+            """
+            drop type if exists {name} cascade;
+            create type {name} as enum ({labels});
+            """
+        ).format(name=sql.Identifier(name), labels=sql.SQL(",").join(labels))
     )
     return name, enum, labels