Esempio n. 1
0
    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