diff --git a/cartoframes/io/managers/context_manager.py b/cartoframes/io/managers/context_manager.py index 7343f03ed..4565f1c82 100644 --- a/cartoframes/io/managers/context_manager.py +++ b/cartoframes/io/managers/context_manager.py @@ -283,14 +283,14 @@ def _create_table_from_columns(self, table_name, schema, columns, cartodbfy): query = 'BEGIN; {create}; {cartodbfy}; COMMIT;'.format( create=_create_table_from_columns_query(table_name, columns), cartodbfy=_cartodbfy_query(table_name, schema) if cartodbfy else '') - self.execute_long_running_query(query) + self.execute_query(query) def _truncate_table(self, table_name, schema, cartodbfy): log.debug('TRUNCATE table "{}"'.format(table_name)) query = 'BEGIN; {truncate}; {cartodbfy}; COMMIT;'.format( truncate=_truncate_table_query(table_name), cartodbfy=_cartodbfy_query(table_name, schema) if cartodbfy else '') - self.execute_long_running_query(query) + self.execute_query(query) def _truncate_and_drop_add_columns(self, table_name, schema, df_columns, table_columns, cartodbfy): log.debug('TRUNCATE AND DROP + ADD columns table "{}"'.format(table_name)) diff --git a/tests/unit/io/managers/test_context_manager.py b/tests/unit/io/managers/test_context_manager.py index a8fdb2262..2b7e12f9d 100644 --- a/tests/unit/io/managers/test_context_manager.py +++ b/tests/unit/io/managers/test_context_manager.py @@ -62,7 +62,7 @@ def test_copy_from(self, mocker): mocker.patch('cartoframes.io.managers.context_manager._create_auth_client') mocker.patch.object(ContextManager, 'has_table', return_value=False) mocker.patch.object(ContextManager, 'get_schema', return_value='schema') - mock_create_table = mocker.patch.object(ContextManager, 'execute_long_running_query') + mock_create_table = mocker.patch.object(ContextManager, 'execute_query') mock = mocker.patch.object(ContextManager, '_copy_from') df = DataFrame({'A': [1]}) columns = [ColumnInfo('A', 'a', 'bigint', False)]