def test_index_record(inspire_app, celery_app_with_context,
celery_session_worker):
models_committed.disconnect(index_after_commit)
records = [
create_record_async("lit"),
create_record_async("aut"),
create_record_async("job"),
create_record_async("jou"),
create_record_async("exp"),
create_record_async("con"),
create_record_async("dat"),
create_record_async("ins"),
]
uuids = [record.id for record in records]
task = index_records.delay(uuids)
results = task.get(timeout=5)
uuids = [str(uuid) for uuid in uuids]
assert results == uuids
for record in records:
result = InspireSearch.get_record_data_from_es(record)
assert record["control_number"] == result["control_number"]
models_committed.connect(index_after_commit)
def test_index_record_deletes_a_deleted_record(inspire_app,
clean_celery_session):
record_to_delete = create_record_async("lit")
record_to_delete_control_number = record_to_delete["control_number"]
record_to_delete = InspireRecord.get_record_by_pid_value(
record_to_delete_control_number, "lit")
record_to_delete.delete()
db.session.commit()
uuids = [record_to_delete.id]
task = index_records.delay(uuids)
results = task.get(timeout=5)
uuids = [str(uuid) for uuid in uuids]
assert results == uuids
with pytest.raises(TransportError):
InspireSearch.get_record_data_from_es(record_to_delete)