def test_get_many_roles(self):
self.assertEqual(True, isinstance(Role.get_all(), dict))
self.role1.save()
self.role2.save()
self.role3.save()
self.assertEqual(True, isinstance(Role.get_all()[0], Role))
self.assertEqual(True, isinstance(Role.get_all()[1], Role))
self.assertEqual(3, len(Role.get_all()))
def test_add_and_remove_role_from_user(self):
self.user1.save()
self.role1.save()
self.role2.save()
user1 = User.get(id=1)
self.assertEqual(0, len(user1.roles))
user1.insert('roles', Role.get_all())
self.assertEqual(2, len(user1.roles))
user1.remove('roles', id=1)
self.assertEqual(1, len(user1.roles))
user1.remove('roles')
self.assertEqual(0, len(user1.roles))
def create_test_token():
"""
Create token for use in testing.
Seed roles before running this.
"""
user = User.get(email="*****@*****.**")
roles = Role.get_all()
user.insert('roles', roles)
user.save()
token = create_token("*****@*****.**")
environ['TEST_TOKEN'] = token
print('\nToken created:\n', token,
'\nThe token is saved in the environment.\n')
