def test_has_roles(self):
r1 = Role(name='admin')
r2 = Role(name='poweruser')
u1 = User(username='******', email='*****@*****.**')
u1.roles.append(r1)
u1.roles.append(r2)
u2 = User(username='******', email='*****@*****.**')
u2.roles.append(r2)
u3 = User(username='******', email='*****@*****.**')
db.session.add(r1)
db.session.add(r2)
db.session.add(u1)
db.session.add(u2)
db.session.add(u3)
self.assertTrue(u1.has_roles('admin'))
self.assertTrue(u1.has_roles('poweruser'))
self.assertTrue(u1.has_roles('admin', 'poweruser'))
self.assertTrue(u1.has_roles(['admin','poweruser']))
self.assertFalse(u2.has_roles('admin'))
self.assertTrue(u2.has_roles('poweruser'))
self.assertFalse(u2.has_roles('admin', 'poweruser'))
self.assertTrue(u2.has_roles(['admin', 'poweruser']))
self.assertFalse(u3.has_roles('admin'))
self.assertFalse(u3.has_roles('poweruser'))
self.assertFalse(u3.has_roles('admin', 'poweruser'))
self.assertFalse(u3.has_roles(['admin', 'poweruser']))
