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')