PrivacyScanData/Target/Azure/test_adx_5.py

def test_conn_method_aad_creds(self, mock_init):
        mock_init.return_value = None
        db.merge_conn(
            Connection(
                conn_id=ADX_TEST_CONN_ID,
                conn_type='azure_data_explorer',
                login='client_id',
                password='client secret',
                host='https://help.kusto.windows.net',
                extra=json.dumps(
                    {
                        'extra__azure_data_explorer__tenant': 'tenant',
                        'extra__azure_data_explorer__auth_method': 'AAD_CREDS',
                    }
                ),
            )
        )
        AzureDataExplorerHook(azure_data_explorer_conn_id=ADX_TEST_CONN_ID)
        assert mock_init.called_with(
            KustoConnectionStringBuilder.with_aad_user_password_authentication(
                'https://help.kusto.windows.net', 'client_id', 'client secret', 'tenant'
            )
        )