def add_admin():
"""Adds admin to database"""
admin = User.query.filter_by(username="******").first()
if admin is None:
new_admin = User("admin", generate_password_hash("supertajnehaslo"), "*****@*****.**", social_id=None)
new_admin.role = USER.ADMIN
new_admin.confirmed = True
db.session.add(new_admin)
db.session.commit()
def test_user_role(self):
user = User(username='******', password=generate_password_hash('a'), email='*****@*****.**')
self.assertEqual(user.role, USER.USER)
self.assertEqual(user.get_role(), USER.ROLE[USER.USER])
self.assertFalse(user.is_admin())
user.role = USER.ADMIN
self.assertEqual(user.get_role(), USER.ROLE[USER.ADMIN])
self.assertTrue(user.is_admin())
