import pytest from delphi_es_utils.errors.exceptions import IndexMappingFilenameError @pytest.fixture(autouse=True) def mock_elasticsearch(mocker): # flake8: noqa mocker.patch('elasticsearch.Elasticsearch', autospec=True) @pytest.fixture def mock_index_manager(mocker): """:rtype: delphi_es_utils.services.index_manager.IndexManager""" mock_manager = mocker.patch( 'delphi_es_utils.services.index_manager.IndexManager', autospec=True ) mock_manager.index_name = 'linkfire-link' return mock_manager @pytest.fixture def index_manager(mock_elasticsearch): """:rtype: delphi_es_utils.services.index_manager.IndexManager""" from delphi_es_utils.services import IndexManager return IndexManager('linkfire-link') class TestIndexManager: def test_mapping_filename(self, index_manager): assert index_manager.mapping_filename == 'linkfire-link.json' index_manager.index_alias = 'foo' with pytest.raises(IndexMappingFilenameError): result = index_manager.mapping_filename def test_mapping(self, index_manager): result = index_manager.mapping assert isinstance(result, dict) def test_payload(self, index_manager): assert index_manager.reindex_payload['dest']['op_type'] == 'create'