def test_query_by_admin(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]) self.assertIsInstance(GroupAdmin.query_by_admin(a), BaseQuery) self.assertEqual(GroupAdmin.query_by_admin(a).count(), 1) self.assertEqual(GroupAdmin.query_by_admin(g).count(), 0)
def test_group_admin_query_by_admin(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]) assert isinstance(GroupAdmin.query_by_admin(a), BaseQuery) assert GroupAdmin.query_by_admin(a).count() == 1 assert GroupAdmin.query_by_admin(g).count() == 0