def test_index_relations(app, db):
"""Test the index_relations method."""
data_v1 = {'body': u'test_body',
'title': u'test_title'}
data_v2 = {'body': u'test_body2',
'title': u'test_title2'}
# add first child to the relation
rec_v1 = Record.create(data_v1)
parent_pid = RecordIdProvider.create(object_type='rec',
object_uuid=None,
status=PIDStatus.REGISTERED).pid
data_v1['conceptrecid'] = parent_pid.pid_value
provider = RecordIdProvider.create('rec', rec_v1.id)
data_v1['recid'] = provider.pid.pid_value
versioning = PIDNodeVersioning(pid=parent_pid)
versioning.insert_child(child_pid=provider.pid)
db.session.commit()
output = index_relations(app, 'recid', record=rec_v1)
expected_output = \
{'relations': {
'version': [{
u'children': [{u'pid_type': u'recid',
u'pid_value': u'2'}],
u'index': 0,
u'is_child': True,
u'is_last': True,
u'is_parent': False,
u'next': None,
u'parent': {u'pid_type': u'recid',
u'pid_value': u'1'},
u'previous': None,
u'type': 'version'}]}}
assert compare_dictionaries(output, expected_output)
# add second child to the relation
rec_v2 = Record.create(data_v2)
data_v2['conceptrecid'] = parent_pid.pid_value
provider_v2 = RecordIdProvider.create('rec', rec_v2.id)
versioning.insert_child(child_pid=provider_v2.pid)
db.session.commit()
output = index_relations(app, 'recid', record=rec_v2)
expected_output = \
{'relations': {
'version': [{
u'children': [{u'pid_type': u'recid',
u'pid_value': u'2'},
{u'pid_type': u'recid',
u'pid_value': u'3'}],
u'index': 1,
u'is_child': True,
u'is_last': True,
u'is_parent': False,
u'next': None,
u'parent': {u'pid_type': u'recid',
u'pid_value': u'1'},
u'previous': {u'pid_type': u'recid',
u'pid_value': u'2'},
u'type': 'version'}]}}
assert compare_dictionaries(output, expected_output)
def test_versioning_insert_child(db, version_pids, build_pid):
"""Test PIDNodeVersioning.insert_child(...)."""
new_pids = create_pids(3)
parent_pid = build_pid(version_pids[0]['parent'])
h1 = PIDNodeVersioning(parent_pid)
# insert as first child
h1.insert_child(new_pids[0], 0)
version_pids[0]['children'].insert(0, new_pids[0])
assert h1.children.ordered('asc').all() == \
filter_pids(version_pids[0]['children'], PIDStatus.REGISTERED)
# insert as last child. This should insert just before the draft
version_pids[0]['children'].insert(h1.index(h1.draft_child), new_pids[1])
h1.insert_child(new_pids[1], -1)
# Check that the parent redirects to the added PID
assert (version_pids[0]['parent'].get_redirect() == new_pids[1])
# Register the draft so that it appears in the children
h1.draft_child.register()
h1.update_redirect()
assert h1.children.ordered('asc').all() == \
filter_pids(version_pids[0]['children'], PIDStatus.REGISTERED)
# insert again but without a draft child. It should be inserted at the end.
version_pids[0]['children'].append(new_pids[2])
h1.insert_child(new_pids[2], -1)
assert h1.children.ordered('asc').all() == \
filter_pids(version_pids[0]['children'], PIDStatus.REGISTERED)
reserved_pid = create_pids(1, status=PIDStatus.RESERVED)[0]
# Check the exception raised when trying to insert a RESERVED PID
with pytest.raises(PIDRelationConsistencyError):
h1.insert_child(reserved_pid)
def record_minter(record_uuid, data):
parent = data.get('conceptrecid')
if not parent:
parent_pid = RecordIdProvider.create(object_type='rec',
object_uuid=None,
status=PIDStatus.REGISTERED).pid
data['conceptrecid'] = parent_pid.pid_value
else:
parent_pid = PersistentIdentifier.get(
pid_type=RecordIdProvider.pid_type, pid_value=parent)
provider = RecordIdProvider.create('rec', record_uuid)
data['recid'] = provider.pid.pid_value
versioning = PIDNodeVersioning(pid=parent_pid)
versioning.insert_child(child_pid=provider.pid)
return provider.pid
def test_index_siblings(app, db, version_pids):
"""Test the index_siblings method."""
# Create a pid relation with 3 children
data_v1 = {'body': u'test_body',
'title': u'test_title'}
data_v2 = {'body': u'test_body2',
'title': u'test_title2'}
data_v3 = {'body': u'test_body3',
'title': u'test_title3'}
rec_v1 = Record.create(data_v1)
parent_pid = RecordIdProvider.create(object_type='rec',
object_uuid=None,
status=PIDStatus.REGISTERED).pid
data_v1['conceptrecid'] = parent_pid.pid_value
provider = RecordIdProvider.create('rec', rec_v1.id)
data_v1['recid'] = provider.pid.pid_value
versioning = PIDNodeVersioning(pid=parent_pid)
versioning.insert_child(child_pid=provider.pid)
rec_v2 = Record.create(data_v2)
data_v2['conceptrecid'] = parent_pid.pid_value
provider_v2 = RecordIdProvider.create('rec', rec_v2.id)
data_v2['recid'] = provider_v2.pid.pid_value
versioning.insert_child(child_pid=provider_v2.pid)
rec_v3 = Record.create(data_v3)
data_v3['conceptrecid'] = parent_pid.pid_value
provider_v3 = RecordIdProvider.create('rec', rec_v3.id)
data_v3['recid'] = provider.pid.pid_value
versioning.insert_child(child_pid=provider_v3.pid)
db.session.commit()
with patch('invenio_indexer.api.RecordIndexer.index_by_id') as mock:
index_siblings(provider.pid, include_pid=True, eager=True,
with_deposits=False)
mock.assert_any_call(str(provider.pid.object_uuid))
mock.assert_any_call(str(provider_v2.pid.object_uuid))
mock.assert_any_call(str(provider_v3.pid.object_uuid))