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