def test_execute_sql_no_results():
global executed_once
executed_once = False
spark_events = MagicMock()
sqlquery = SQLQuery("SHOW TABLES", "take", maxrows=-1, spark_events=spark_events)
sqlquery.to_command = MagicMock()
sqlquery.to_only_columns_query = MagicMock()
result1 = ""
result_data = pd.DataFrame([])
session = MagicMock()
sqlquery.to_command.return_value.execute.return_value = (True, result1)
session.kind = "spark"
result = sqlquery.execute(session)
assert_frame_equal(result, result_data)
sqlquery.to_command.return_value.execute.assert_called_once_with(session)
spark_events.emit_sql_execution_start_event.assert_called_once_with(
session.guid,
session.kind,
session.id,
sqlquery.guid,
sqlquery.samplemethod,
sqlquery.maxrows,
sqlquery.samplefraction,
)
spark_events.emit_sql_execution_end_event.assert_called_once_with(
session.guid, session.kind, session.id, sqlquery.guid, sqlquery.to_command.return_value.guid, True, "", ""
)
def test_execute_sql_no_results():
global executed_once
executed_once = False
spark_events = MagicMock()
sqlquery = SQLQuery("SHOW TABLES",
"take",
maxrows=-1,
spark_events=spark_events)
sqlquery.to_command = MagicMock()
sqlquery.to_only_columns_query = MagicMock()
result1 = ""
result_data = pd.DataFrame([])
session = MagicMock()
sqlquery.to_command.return_value.execute.return_value = (
True, result1, MIMETYPE_TEXT_PLAIN)
session.kind = "spark"
result = sqlquery.execute(session)
assert_frame_equal(result, result_data)
sqlquery.to_command.return_value.execute.assert_called_once_with(session)
spark_events.emit_sql_execution_start_event.assert_called_once_with(
session.guid, session.kind, session.id, sqlquery.guid,
sqlquery.samplemethod, sqlquery.maxrows, sqlquery.samplefraction)
spark_events.emit_sql_execution_end_event.assert_called_once_with(
session.guid, session.kind, session.id, sqlquery.guid,
sqlquery.to_command.return_value.guid, True, "", "")
