def test_principal():
__configuration__ = '''
jwt:
secret: JWT-SECRET
algorithm: HS256
max_age: 86400 # 24 Hours
refresh_token:
secret: JWT-REFRESH-SECRET
algorithm: HS256
max_age: 2678400 # 30 Days
'''
configure(force=True)
settings.merge(__configuration__)
principal = JwtPrincipal(
dict(email='*****@*****.**', id=1, sessionId=1, roles=['admin']))
assert principal.email == '*****@*****.**'
assert principal.id == 1
assert principal.session_id == 1
assert principal.roles == ['admin']
assert principal.is_in_roles('admin') is True
assert principal.is_in_roles('admin', 'god') is True
encoded = principal.dump()
principal = JwtPrincipal.load(encoded.decode())
assert principal.email == '*****@*****.**'
assert principal.id == 1
assert principal.session_id == 1
assert principal.roles == ['admin']
assert principal.is_in_roles('admin') is True
assert principal.is_in_roles('admin', 'god') is True
principal = JwtPrincipal.load(encoded.decode(), force=True)
assert principal.email == '*****@*****.**'
assert principal.id == 1
assert principal.session_id == 1
assert principal.roles == ['admin']
assert principal.is_in_roles('admin') is True
assert principal.is_in_roles('admin', 'god') is True
principal =\
JwtPrincipal.load((b'Bearer %s' % encoded).decode(), force=True)
assert principal.email == '*****@*****.**'
assert principal.id == 1
assert principal.session_id == 1
assert principal.roles == ['admin']
assert principal.is_in_roles('admin') is True
assert principal.is_in_roles('admin', 'god') is True
def test_principal(self):
principal = JwtPrincipal(
dict(email='*****@*****.**', id=1, sessionId=1, roles=['admin']))
self.assertEqual(principal.email, '*****@*****.**')
self.assertEqual(principal.id, 1)
self.assertEqual(principal.session_id, 1)
self.assertEqual(principal.roles, ['admin'])
self.assertTrue(principal.is_in_roles('admin'))
self.assertTrue(principal.is_in_roles('admin', 'god'))
encoded = principal.dump()
principal = JwtPrincipal.load(encoded.decode())
self.assertEqual(principal.email, '*****@*****.**')
self.assertEqual(principal.id, 1)
self.assertEqual(principal.session_id, 1)
self.assertEqual(principal.roles, ['admin'])
self.assertTrue(principal.is_in_roles('admin'))
self.assertTrue(principal.is_in_roles('admin', 'god'))
principal = JwtPrincipal.load(encoded.decode(), force=True)
self.assertEqual(principal.email, '*****@*****.**')
self.assertEqual(principal.id, 1)
self.assertEqual(principal.session_id, 1)
self.assertEqual(principal.roles, ['admin'])
self.assertTrue(principal.is_in_roles('admin'))
self.assertTrue(principal.is_in_roles('admin', 'god'))
principal = JwtPrincipal.load((b'Bearer %s' % encoded).decode(),
force=True)
self.assertEqual(principal.email, '*****@*****.**')
self.assertEqual(principal.id, 1)
self.assertEqual(principal.session_id, 1)
self.assertEqual(principal.roles, ['admin'])
self.assertTrue(principal.is_in_roles('admin'))
self.assertTrue(principal.is_in_roles('admin', 'god'))