def test_store_root_new_record(workflow_app):
config = {'FEATURE_FLAG_ENABLE_MERGER': True}
eng = MagicMock(workflow_definition=MagicMock(data_type='hep'))
with patch.dict(current_app.config, config):
head = TestRecordMetadata.create_from_kwargs(index=False,
has_pid=False)
head_uuid = head.record_metadata.id
record = head.record_metadata.json
obj = workflow_object_class.create(record)
root = {
'version': 'original',
'acquisition_source': {
'source': 'arXiv'
}
}
extra_data = {
'head_uuid': str(head_uuid),
'merger_root': root,
}
obj.extra_data = extra_data
store_root(obj, eng)
root_entry = read_wf_record_source(head_uuid, 'arxiv')
assert root_entry.json == root
def test_store_root_new_record(workflow_app):
config = {
'FEATURE_FLAG_ENABLE_MERGER': True
}
eng = MagicMock(workflow_definition=MagicMock(data_type='hep'))
with patch.dict(current_app.config, config):
head = TestRecordMetadata.create_from_kwargs(index=False, has_pid=False)
head_uuid = head.record_metadata.id
record = head.record_metadata.json
obj = workflow_object_class.create(record)
root = {
'version': 'original',
'acquisition_source': {'source': 'arXiv'}
}
extra_data = {
'head_uuid': str(head_uuid),
'merger_root': root,
}
obj.extra_data = extra_data
store_root(obj, eng)
root_entry = read_wf_record_source(head_uuid, 'arxiv')
assert root_entry.json == root