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