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 test_update_redirect(db, version_pids, build_pid):
"""Test PIDNodeVersioning.update_redirect()."""
# Test update_redirect on a PID without any child
parent_pids = create_pids(1, prefix='parent', status=PIDStatus.RESERVED)
draft_pids = create_pids(2, prefix='draft', status=PIDStatus.RESERVED)
parent = PIDNodeVersioning(build_pid(parent_pids[0]))
parent.update_redirect()
assert parent_pids[0].status == PIDStatus.RESERVED
# Test that update_redirect remains reserved once it has a draft child
parent.insert_draft_child(draft_pids[0])
assert parent_pids[0].status == PIDStatus.RESERVED
h1 = PIDNodeVersioning(build_pid(version_pids[0]['parent']))
def test_redirect(expected_length, expected_redirect):
filtered = filter_pids(version_pids[0]['children'],
status=PIDStatus.REGISTERED)
assert len(filtered) == expected_length
assert h1.children.ordered('asc').all() == filtered
assert h1._resolved_pid.get_redirect() == expected_redirect
# Test update_redirect when it already points to the last version
last = h1.last_child
draft = h1.draft_child
h1.update_redirect()
test_redirect(3, last)
# Test update_redirect after publishing the draft
h1.draft_child.register()
h1.update_redirect()
test_redirect(4, draft)
# Test update_redirect after deleting the last version
h1.last_child.delete()
h1.update_redirect()
test_redirect(3, last)
# Test that if every version is deleted the HEAD pid is also deleted
for pid in filter_pids(version_pids[0]['children'],
status=PIDStatus.REGISTERED):
pid.delete()
h1.update_redirect()
test_redirect(0, last)
# Test that an exception is raised if unsupported PIDStatus are used.
version_pids[0]['children'][0].status = PIDStatus.NEW
with pytest.raises(PIDRelationConsistencyError):
h1.update_redirect()
def test_node_children(db, version_relation, version_pids, build_pid):
"""Test PIDNode.children()."""
# Test normal simple ordering of children
parent_node = PIDNode(build_pid(version_pids[0]['parent']),
version_relation)
assert parent_node.children.ordered('asc').all() == \
version_pids[0]['children']
# Check that status filtering works on children
assert parent_node.children.status(
PIDStatus.REGISTERED).ordered('asc').all() == filter_pids(
version_pids[0]['children'], PIDStatus.REGISTERED)
# Test children of a PID having no children
child_node = PIDNode(build_pid(version_pids[0]['children'][0]),
version_relation)
assert child_node.children.ordered('asc').all() == []
def test_versioning_children(db, version_pids, build_pid):
"""Test the children property of PIDNoneVersioning."""
parent_pid = build_pid(version_pids[0]['parent'])
h1 = PIDNodeVersioning(parent_pid)
assert h1.children.ordered('asc').all() == \
filter_pids(version_pids[0]['children'], PIDStatus.REGISTERED)
def test_redirect(expected_length, expected_redirect):
filtered = filter_pids(version_pids[0]['children'],
status=PIDStatus.REGISTERED)
assert len(filtered) == expected_length
assert h1.children.ordered('asc').all() == filtered
assert h1._resolved_pid.get_redirect() == expected_redirect