Small change to template database initialisation

ClearcodeHQ · Mar 11, 2024 · 4b4aa95 · 4b4aa95
1 parent 68a3504
commit 4b4aa95
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/pytest_postgresql/janitor.py b/pytest_postgresql/janitor.py
@@ -67,8 +67,7 @@ def init(self) -> None:
         """Create database in postgresql."""
         with self.cursor() as cur:
             if self.is_template():
-                cur.execute(f'CREATE DATABASE "{self.template_dbname}";')
-                cur.execute(f'ALTER DATABASE "{self.template_dbname}" with is_template true;')
+                cur.execute(f'CREATE DATABASE "{self.template_dbname}" WITH is_template = true;')
             elif self.template_dbname is None:
                 cur.execute(f'CREATE DATABASE "{self.dbname}";')
             else:
@@ -90,8 +89,6 @@ def drop(self) -> None:
         with self.cursor() as cur:
             self._dont_datallowconn(cur, db_to_drop)
             self._terminate_connection(cur, db_to_drop)
-            if self.is_template():
-                cur.execute(f'ALTER DATABASE "{db_to_drop}" with is_template false;')
             cur.execute(f'DROP DATABASE IF EXISTS "{db_to_drop}";')
 
     @staticmethod