CTAS table delete (#797)

jaidisido · web-flow · commit 5813e7477c17 · 2021-07-06T16:49:18.000+01:00
* Deleting in correct database

* Upgrading dependencies
diff --git a/awswrangler/athena/_read.py b/awswrangler/athena/_read.py
@@ -558,7 +558,9 @@ def _resolve_query_without_cache(
                 boto3_session=boto3_session,
             )
         finally:
-            catalog.delete_table_if_exists(database=database, table=name, boto3_session=boto3_session)
+            catalog.delete_table_if_exists(
+                database=ctas_database_name or database, table=name, boto3_session=boto3_session
+            )
     return _resolve_query_without_cache_regular(
         sql=sql,
         database=database,
diff --git a/tests/test_athena.py b/tests/test_athena.py
@@ -112,7 +112,7 @@ def test_athena_ctas(path, path2, path3, glue_table, glue_table2, glue_database,
         ctas_temp_table_name=glue_table2,
         s3_output=path3,
     )
-    assert wr.catalog.does_table_exist(database=glue_ctas_database, table=glue_table2) is True
+    assert wr.catalog.does_table_exist(database=glue_ctas_database, table=glue_table2) is False
     assert len(wr.s3.list_objects(path=path3)) > 2
     assert len(wr.s3.list_objects(path=final_destination)) > 0
     for df in dfs:

Original file line number	Diff line number	Diff line change
`@@ -112,7 +112,7 @@ def test_athena_ctas(path, path2, path3, glue_table, glue_table2, glue_database,`
`112`	`112`	`ctas_temp_table_name=glue_table2,`
`113`	`113`	`s3_output=path3,`
`114`	`114`	`)`
`115`		`- assert wr.catalog.does_table_exist(database=glue_ctas_database, table=glue_table2) is True`
	`115`	`+ assert wr.catalog.does_table_exist(database=glue_ctas_database, table=glue_table2) is False`
`116`	`116`	`assert len(wr.s3.list_objects(path=path3)) > 2`
`117`	`117`	`assert len(wr.s3.list_objects(path=final_destination)) > 0`
`118`	`118`	`for df in dfs:`