def test_get_admin_roles(self):
roles = get_user_roles(current_user=self.admin)
self.assertIsInstance(roles, dict)
self.assertEqual(len(roles), 1)
self.assertIn('info', roles)
self.assertEqual(len(roles['info']), 4)
self.assertIn('user_id', roles['info'])
self.assertEqual(roles['info']['user_id'], self.admin.id)
self.assertIn('is_recruiter', roles['info'])
self.assertIn('is_senior_recruiter', roles['info'])
self.assertIn('is_admin', roles['info'])
self.assertFalse(roles['info']['is_recruiter'])
self.assertFalse(roles['info']['is_senior_recruiter'])
self.assertTrue(roles['info']['is_admin'])
def api_get_user_roles():
"""
Gets a list of all roles for the logged in user.
Returns:
response (dict)
Example:
response = {
'info': {
is_recruiter: bool,
is_senior_recruiter: bool,
is_admin: bool,
user_id: int,
}
}
"""
return jsonify(get_user_roles(current_user=current_user))