RM-95 use quoted schema and table

bluelabsio · Nov 6, 2023 · 4404d75 · 4404d75
1 parent d528fea
commit 4404d75
Showing 1 changed file with 7 additions and 5 deletions.
diff --git a/tests/integration/records/records_database_fixture.py b/tests/integration/records/records_database_fixture.py
@@ -32,9 +32,11 @@ def tear_down(self):
 
     @bigquery_retry()
     def bring_up(self):
+        quoted_schema_and_table = self.quote_schema_and_table(self.schema_name,
+                                                              self.table_name)
         if self.engine.name == 'redshift':
             create_tables = f"""
-              CREATE TABLE {self.schema_name}.{self.table_name} AS
+              CREATE TABLE {quoted_schema_and_table} AS
               SELECT 123 AS num,
                      '123' AS numstr,
                      'foo' AS str,
@@ -49,7 +51,7 @@ def bring_up(self):
 """  # noqa
         elif self.engine.name == 'vertica':
             create_tables = f"""
-              CREATE TABLE {self.schema_name}.{self.table_name} AS
+              CREATE TABLE {quoted_schema_and_table} AS
               SELECT 123 AS num,
                     '123' AS numstr,
                     'foo' AS str,
@@ -64,7 +66,7 @@ def bring_up(self):
 """  # noqa
         elif self.engine.name == 'bigquery':
             create_tables = f"""
-              CREATE TABLE {self.schema_name}.{self.table_name} AS
+              CREATE TABLE {quoted_schema_and_table} AS
               SELECT 123 AS num,
                     '123' AS numstr,
                     'foo' AS str,
@@ -79,7 +81,7 @@ def bring_up(self):
 """  # noqa
         elif self.engine.name == 'postgresql':
             create_tables = f"""
-              CREATE TABLE {self.schema_name}.{self.table_name} AS
+              CREATE TABLE {quoted_schema_and_table} AS
               SELECT 123 AS num,
                      '123' AS numstr,
                      'foo' AS str,
@@ -94,7 +96,7 @@ def bring_up(self):
 """  # noqa
         elif self.engine.name == 'mysql':
             create_tables = f"""
-              CREATE TABLE {self.schema_name}.{self.table_name} AS
+              CREATE TABLE {quoted_schema_and_table} AS
               SELECT 123 AS num,
                      '123' AS numstr,
                      'foo' AS str,