def test_has_role_user_with_role(self):
role_names = TestUserRolesDatabase.add_roles_to_db(3)
user = User('username', 'password')
user.set_roles(role_names)
db.session.commit()
for role in role_names:
self.assertTrue(user.has_role(role))
def test_has_role_user_without_role(self):
role_names = TestUserRolesDatabase.add_roles_to_db(3)
user = User('username', 'password')
user.set_roles(role_names)
self.assertFalse(user.has_role('invalid'))
def test_has_role_user_with_no_roles(self):
user = User('username', 'password')
self.assertFalse(user.has_role('role3'))