add tests

This commit is contained in:
Timi Fasubaa
2018-05-25 15:45:11 -07:00
parent d38315a307
commit a9d7fafd9f
6 changed files with 67 additions and 33 deletions

View File

@@ -186,11 +186,10 @@ def execute_sql(
query.user_id, start_dttm.strftime('%Y_%m_%d_%H_%M_%S'))
executed_sql = superset_query.as_create_table(query.tmp_table_name)
query.select_as_cta_used = True
elif (superset_query.is_select() and SQL_MAX_ROWS and
if (superset_query.is_select() and SQL_MAX_ROWS and
(not query.limit or query.limit > SQL_MAX_ROWS)):
query.limit = SQL_MAX_ROWS
executed_sql = database.apply_limit_to_sql(executed_sql, query.limit)
query.limit_used = True
# Hook to allow environment-specific mutation (usually comments) to the SQL
SQL_QUERY_MUTATOR = config.get('SQL_QUERY_MUTATOR')