def test_publish_with_data_and_old_index(self) -> None:
"""
Test Publish functionality with data and with old_index in place
"""
mock_data = json.dumps({'KEY_DOESNOT_MATTER': 'NO_VALUE',
'KEY_DOESNOT_MATTER2': 'NO_VALUE2'})
self.mock_es_client.indices.get_alias.return_value = {'test_old_index': 'DOES_NOT_MATTER'}
with patch('builtins.open', mock_open(read_data=mock_data)) as mock_file:
publisher = ElasticsearchPublisher()
publisher.init(conf=Scoped.get_scoped_conf(conf=self.conf,
scope=publisher.get_scope()))
# assert mock was called with test_file_path and test_file_mode
mock_file.assert_called_once_with(self.test_file_path, self.test_file_mode)
publisher.publish()
# ensure indices create endpoint was called
default_mapping = ElasticsearchPublisher.DEFAULT_ELASTICSEARCH_INDEX_MAPPING
self.mock_es_client.indices.create.assert_called_once_with(index=self.test_es_new_index,
body=default_mapping)
# bulk endpoint called once
self.mock_es_client.bulk.assert_called_once_with(
[{'index': {'_index': self.test_es_new_index}},
{'KEY_DOESNOT_MATTER': 'NO_VALUE',
'KEY_DOESNOT_MATTER2': 'NO_VALUE2',
'resource_type': 'test_doc_type'}]
)
# update alias endpoint called once
self.mock_es_client.indices.update_aliases.assert_called_once_with(
{'actions': [{"add": {"index": self.test_es_new_index, "alias": self.test_es_alias}},
{"remove_index": {"index": 'test_old_index'}}]}
)
def test_publish_with_no_data(self) -> None:
"""
Test Publish functionality with no data
"""
with patch('builtins.open', mock_open(read_data='')) as mock_file:
publisher = ElasticsearchPublisher()
publisher.init(conf=Scoped.get_scoped_conf(conf=self.conf,
scope=publisher.get_scope()))
# assert mock was called with test_file_path and test_file_mode
mock_file.assert_called_with(self.test_file_path, self.test_file_mode)
publisher.publish()
# no calls should be made through elasticseach_client when there is no data
self.assertTrue(self.mock_es_client.call_count == 0)