def test_elasticsearch_builder_called_with_no_creds(mocker):
from elasticsearch import Elasticsearch
from Elasticsearch_v2 import elasticsearch_builder
es_mock = mocker.patch.object(Elasticsearch, '__init__', return_value=None)
elasticsearch_builder()
assert es_mock.call_args[1].get('http_auth') is None
assert es_mock.call_args[1].get('api_key') is None
def test_elasticsearch_builder_called_with_username_password(mocker):
from elasticsearch import Elasticsearch
from Elasticsearch_v2 import elasticsearch_builder
es_mock = mocker.patch.object(Elasticsearch, '__init__', return_value=None)
elasticsearch_builder()
assert es_mock.call_args[1].get('http_auth') == ('mock', 'demisto')
assert es_mock.call_args[1].get('api_key') is None
def test_elasticsearch_builder_called_with_username_password(params, mocker):
mocker.patch.object(demisto, 'params', return_value=params)
importlib.reload(
Elasticsearch_v2
) # To reset the Elasticsearch client with the OpenSearch library
from Elasticsearch_v2 import Elasticsearch, elasticsearch_builder
es_mock = mocker.patch.object(Elasticsearch, '__init__', return_value=None)
elasticsearch_builder(None)
assert es_mock.call_args[1].get('http_auth') == ('mock', 'demisto')
assert es_mock.call_args[1].get('api_key') is None
def test_elasticsearch_builder_called_with_no_creds(params, mocker):
mocker.patch.object(demisto, 'params', return_value=params)
importlib.reload(
Elasticsearch_v2
) # To reset the Elasticsearch client with the OpenSearch library
mocker.patch('Elasticsearch_v2.USERNAME', None)
mocker.patch('Elasticsearch_v2.PASSWORD', None)
from Elasticsearch_v2 import Elasticsearch, elasticsearch_builder
es_mock = mocker.patch.object(Elasticsearch, '__init__', return_value=None)
elasticsearch_builder(None)
assert es_mock.call_args[1].get('http_auth') is None
assert es_mock.call_args[1].get('api_key') is None
