def test_not_has_role_admin(self): with self.app.test_request_context('/'): query = MyModel.query ac = HasRole(['admin']) query = ac.alter_query('read', query, self.u2, 'My', MyModel) items = query.values(MyModel.id) ids = [i[0] for i in items] self.assertNotIn(self.d1, ids) self.assertNotIn(self.d2, ids) self.assertNotIn(self.d3, ids) self.assertNotIn(self.d4, ids)
def test_has_role_anonymous(self): with self.app.test_request_context('/'): query = MyModel.query ac = HasRole([WBRoleModel.anonymous_role_name]) query = ac.alter_query('read', query, None, 'My', MyModel) items = query.values(MyModel.id) ids = [i[0] for i in items] self.assertIn(self.d1, ids) self.assertIn(self.d2, ids) self.assertIn(self.d3, ids) self.assertIn(self.d4, ids)