def tearDown(self):
"""Clean up created objects."""
from invenio_workflows.models import Workflow
from inspire.modules.workflows.receivers import precache_holdingpen_row
from invenio_workflows.receivers import index_holdingpen_record
from invenio_workflows.signals import workflow_halted, workflow_object_saved
workflow_halted.connect(precache_holdingpen_row)
workflow_object_saved.connect(index_holdingpen_record)
self.delete_objects(Workflow.get(Workflow.module_name == "unit_tests").all())
self.cleanup_registries()
def tearDown(self):
"""Clean up created objects."""
from invenio_knowledge.api import delete_kb
from invenio_workflows.models import Workflow
from inspire.modules.workflows.receivers import precache_holdingpen_row
from invenio_workflows.receivers import index_holdingpen_record
from invenio_workflows.signals import (
workflow_halted,
workflow_object_saved
)
workflow_halted.connect(precache_holdingpen_row)
workflow_object_saved.connect(index_holdingpen_record)
self.delete_objects(
Workflow.get(Workflow.module_name == 'unit_tests').all())
self.cleanup_registries()
delete_kb('harvesting_fixture_kb')
record.update(workflow.get_record(sender))
except Exception as err:
current_app.logger.exception(err)
try:
record.update(workflow.get_sort_data(sender))
except Exception as err:
current_app.logger.exception(err)
# Add collection to get correct mapping
record["_collections"] = get_record_collections(record)
# Depends on "_collections" being filled correctly for record
record_index = get_record_index(record) or current_app.config["SEARCH_ELASTIC_DEFAULT_INDEX"]
# Trigger any before_record_index receivers
before_record_index.send(sender.id, json=record, index=record_index)
if record_index:
index = current_app.config['WORKFLOWS_HOLDING_PEN_ES_PREFIX'] + record_index
es.index(
index=index,
doc_type=current_app.config["WORKFLOWS_HOLDING_PEN_DOC_TYPE"],
body=dict(record),
id=sender.id
)
workflow_halted.connect(precache_holdingpen_row)
workflow_halted.connect(continue_workflow)
listen(BibWorkflowObject, "after_delete", delete_from_index)
