def test_query_by_group(self): """.""" from invenio_groups.models import Group, GroupAdmin from flask.ext.sqlalchemy import BaseQuery a = Group.create(name="admin") g = Group.create(name="test", admins=[a]) g2 = Group.create(name="test2") self.assertIsInstance(GroupAdmin.query_by_group(g), BaseQuery) self.assertEqual(GroupAdmin.query_by_group(g).count(), 1) self.assertEqual(GroupAdmin.query_by_group(g2).count(), 0)
def test_group_admin_query_by_group(app): """.""" with app.app_context(): from invenio_groups.models import Group, GroupAdmin from flask.ext.sqlalchemy import BaseQuery a = Group.create(name="admin") g = Group.create(name="test", admins=[a]) g2 = Group.create(name="test2") assert isinstance(GroupAdmin.query_by_group(g), BaseQuery) assert GroupAdmin.query_by_group(g).count() == 1 assert GroupAdmin.query_by_group(g2).count() == 0