def test_push_metadata_to_elasticsearch_index_exception():
with pytest.raises(Exception):
metadata_file_name = "tests/data/metadata_output_raw.json"
bucket_name = "bucket_name"
data = open(metadata_file_name, 'r').read()
metadata = json.loads(data)
elasticsearch_client = Mock()
metadata_obj = HandleBucketEvent()
metadata_obj.es_client = elasticsearch_client
metadata_obj.push_metadata_to_elasticsearch(bucket_name, metadata)
def test_push_metadata_to_elasticsearch():
elasticsearch_endpoint = None
if os.environ.get("SUBMISSIONS_BUCKET_NAME", None):
elasticsearch_endpoint = os.environ.pop("ELASTICSEARCH_ENDPOINT")
metadata_file_name = "tests/data/metadata_output_curated.json"
bucket_name = "bucket_name"
data = open(metadata_file_name, 'r').read()
metadata = json.loads(data)
metadata_obj = HandleBucketEvent()
try:
with pytest.raises(KeyError):
metadata_obj.push_metadata_to_elasticsearch(bucket_name, metadata)
pass
finally:
os.environ["ELASTICSEARCH_ENDPOINT"] = elasticsearch_endpoint
