MM-57990: fix failing tests

mattermost · Apr 29, 2024 · 5d93afd · 5d93afd
1 parent d07cd08
commit 5d93afd
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/tests/utils/db/test_helpers.py b/tests/utils/db/test_helpers.py
@@ -108,7 +108,7 @@ def test_should_upload_file_to_table(mocker):
             call('TRUNCATE TABLE test-schema.a-table'),
             call('CREATE TEMPORARY STAGE IF NOT EXISTS test-schema.a-table'),
             call('PUT file:///tmp/upload.csv @test-schema.a-table OVERWRITE=TRUE'),
-            call('COPY INTO test-schema.a-table FROM @test-schema.a-table FILE_FORMAT = (TYPE = CSV SKIP_HEADER = 1)'),
+            call("COPY INTO test-schema.a-table FROM @test-schema.a-table FILE_FORMAT = (TYPE = CSV SKIP_HEADER = 1 FIELD_OPTIONALLY_ENCLOSED_BY = '\"')"),  # noqa: E501
         ]
     )
 

diff --git a/utils/db/helpers.py b/utils/db/helpers.py
@@ -69,7 +69,7 @@ def upload_csv_as_table(engine: Engine, file: str, schema: str, table: str) -> N
     """
     # Truncate table, upload file and replace table content's within a transaction.
     with engine.begin() as conn:
-        conn.execute(f"TRUNCATE TABLE IF EXISTS {schema}.{table}")
+        conn.execute(f"TRUNCATE TABLE {schema}.{table}")
         conn.execute(f"CREATE TEMPORARY STAGE IF NOT EXISTS {schema}.{table}")
         conn.execute(f"PUT file://{file} @{schema}.{table} OVERWRITE=TRUE")
         conn.execute(