Avoid multistatements in test

cloud-utils · Dec 29, 2023 · 8615b64 · 8615b64
1 parent e5e1557
commit 8615b64
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/test/test.py b/test/test.py
@@ -25,18 +25,18 @@ def setUpClass(cls):
         cls.db_name = os.environ.get("AURORA_DB_NAME", __name__)
         with aurora_data_api.connect(database=cls.db_name) as conn, conn.cursor() as cur:
             try:
+                cur.execute('CREATE EXTENSION IF NOT EXISTS "uuid-ossp"')
+                cur.execute('DROP TABLE IF EXISTS aurora_data_api_test')
                 cur.execute(
                     """
-                    CREATE EXTENSION IF NOT EXISTS "uuid-ossp";
-                    DROP TABLE IF EXISTS aurora_data_api_test;
                     CREATE TABLE aurora_data_api_test (
                         id SERIAL,
                         name TEXT,
                         doc JSONB DEFAULT '{}',
                         num NUMERIC (10, 5) DEFAULT 0.0,
                         ts TIMESTAMP WITHOUT TIME ZONE
                     )
-                """
+                    """
                 )
                 cur.executemany(
                     """