| 123456789101112131415 |
- 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)
|
