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_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)
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)
