mirror of
https://gitlab.ub.uni-bielefeld.de/sfb1288inf/nopaque.git
synced 2024-11-15 09:15:41 +00:00
68 lines
2.0 KiB
Python
68 lines
2.0 KiB
Python
from app import create_app, db
|
|
from app.models import User, Role, Permission
|
|
from tests import TestConfig
|
|
import time
|
|
import unittest
|
|
|
|
|
|
class UserModelTestCase(unittest.TestCase):
|
|
def setUp(self):
|
|
self.app = create_app(TestConfig)
|
|
self.app_context = self.app.app_context()
|
|
self.app_context.push()
|
|
db.create_all()
|
|
Role.insert_roles()
|
|
|
|
def tearDown(self):
|
|
db.session.remove()
|
|
db.drop_all()
|
|
self.app_context.pop()
|
|
|
|
def test_password_setter(self):
|
|
u = User(password='cat')
|
|
self.assertTrue(u.password_hash is not None)
|
|
|
|
def test_no_password_getter(self):
|
|
u = User(password='cat')
|
|
with self.assertRaises(AttributeError):
|
|
u.password
|
|
|
|
def test_password_verification(self):
|
|
u = User(password='cat')
|
|
self.assertTrue(u.verify_password('cat'))
|
|
self.assertFalse(u.verify_password('dog'))
|
|
|
|
def test_password_salts_are_random(self):
|
|
u = User(password='cat')
|
|
u2 = User(password='cat')
|
|
self.assertTrue(u.password_hash != u2.password_hash)
|
|
|
|
def test_valid_confirmation_token(self):
|
|
u = User(password='cat')
|
|
db.session.add(u)
|
|
db.session.commit()
|
|
token = u.generate_confirmation_token()
|
|
self.assertTrue(u.confirm(token))
|
|
|
|
def test_invalid_confirmation_token(self):
|
|
u1 = User(password='cat')
|
|
u2 = User(password='dog')
|
|
db.session.add(u1)
|
|
db.session.add(u2)
|
|
db.session.commit()
|
|
token = u1.generate_confirmation_token()
|
|
self.assertFalse(u2.confirm(token))
|
|
|
|
def test_expired_confirmation_token(self):
|
|
u = User(password='cat')
|
|
db.session.add(u)
|
|
db.session.commit()
|
|
token = u.generate_confirmation_token(1)
|
|
time.sleep(2)
|
|
self.assertFalse(u.confirm(token))
|
|
|
|
def test_user_role(self):
|
|
u = User(email='john@example.com', password='cat')
|
|
self.assertTrue(u.can(Permission.CREATE_JOB))
|
|
self.assertFalse(u.can(Permission.ADMIN))
|