/
tests.py
24 lines (20 loc) · 968 Bytes
/
tests.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
import unittest
from app.models import Users,Permissions,AnonymousUser
class TestPermissions(unittest.TestCase):
def test_role(self):
user=Users(email='alexmagwe@gmail.com',username='alex',password='random')
self.assertFalse(user.can(Permissions.ADDNOTES))
self.assertFalse(user.can(Permissions.ADMIN))
self.assertTrue(user.can(Permissions.MYNOTES))
def test_anonymous_user(self):
user=AnonymousUser()
self.assertFalse(user.can(Permissions.ADDNOTES))
self.assertFalse(user.can(Permissions.ADMIN))
self.assertFalse(user.can(Permissions.MYNOTES))
def test_admin(self):
user=Users(email='alexmagwe@gmail.com',username='admin',password='cat')
self.AssertTrue(user.can(Permissions.ADDNOTES))
self.AssertTrue(user.can(Permissions.ADMIN))
self.AssertTrue(user.can(Permissions.MYNOTES))
if __name__=='__main__':
unittest.main()