def _create_content_and_test(self, name, workspace, *args, **kwargs) -> Content: """ All extra parameters (*args, **kwargs) are for Content init :return: Created Content instance """ content = Content(*args, **kwargs) content.label = name content.workspace = workspace DBSession.add(content) DBSession.flush() eq_(1, ContentApi.get_canonical_query().filter(Content.label == name).count()) return ContentApi.get_canonical_query().filter(Content.label == name).one()
def test_children(self): admin = DBSession.query(User).filter(User.email == '*****@*****.**').one() self._create_thread_and_test( workspace_name='workspace_1', folder_name='folder_1', thread_name='thread_1', user=admin ) workspace = DBSession.query(Workspace).filter(Workspace.label == 'workspace_1').one() folder = ContentApi.get_canonical_query().filter(Content.label == 'folder_1').one() eq_([folder, ], list(workspace.get_valid_children()))