def test_map_instance_var_to_claim_name_unmapped(self):
"""
Test that an instance variable that is not in the map is returned as the claim name.
Expected Result: The name of the instance variable is returned unchanged.
"""
instance_var = 'part_of_the_claim_set'
self.assertNotIn(instance_var,
EasyJWT._instance_var_claim_name_mapping)
self.assertEqual(instance_var,
EasyJWT._map_instance_var_to_claim_name(instance_var))
def test_map_instance_var_to_claim_name_mapped(self):
"""
Test that all instance variable names in the mapping return the corresponding claim name.
Expected Result: The claim name for each instance variable name is returned.
"""
for instance_var, claim_name in EasyJWT._instance_var_claim_name_mapping.items(
):
self.assertEqual(
claim_name,
EasyJWT._map_instance_var_to_claim_name(instance_var))
def test_registered_claim_subject(self):
"""
Test the registered claim ``sub``.
Expected Result: The field is an optional claim and correctly mapped to the claim name and vice versa.
"""
instance_var_name = 'subject'
claim_name = 'sub'
self.assertTrue(EasyJWT._is_claim(instance_var_name))
self.assertTrue(EasyJWT._is_optional_claim(claim_name))
self.assertEqual(instance_var_name,
EasyJWT._map_claim_name_to_instance_var(claim_name))
self.assertEqual(
claim_name,
EasyJWT._map_instance_var_to_claim_name(instance_var_name))
