def test_run_query(self, mock_execute): mock_execute.return_value = None db.merge_conn( Connection( conn_id=ADX_TEST_CONN_ID, conn_type='azure_data_explorer', host='https://help.kusto.windows.net', extra=json.dumps({'extra__azure_data_explorer__auth_method': 'AAD_DEVICE'}), ) ) hook = AzureDataExplorerHook(azure_data_explorer_conn_id=ADX_TEST_CONN_ID) hook.run_query('Database', 'Logs | schema', options={'option1': 'option_value'}) properties = ClientRequestProperties() properties.set_option('option1', 'option_value') assert mock_execute.called_with('Database', 'Logs | schema', properties=properties)