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, "", "")