def test_fetch_extra_data(app, example_cern):
"""Test extra data extraction."""
example_response, example_token, _ = example_cern
res = get_dict_from_response(example_response)
# Check that groups were hidden as required
extra_data = fetch_extra_data(res)
assert 'person_id' in extra_data
assert extra_data['person_id'] == "234567"
assert 'identity_class' in extra_data
assert extra_data['identity_class'] == "CERN Registered"
assert 'department' in extra_data
assert extra_data['department'] == "IT/CDA"
def test_fetch_extra_data_fields_missing(app, example_cern):
"""Test extra data extraction when fields are missing."""
example_response, example_token, _ = example_cern
res = get_dict_from_response(example_response)
del res['PersonID']
del res['IdentityClass']
del res['Department']
# Check that groups were hidden as required
extra_data = fetch_extra_data(res)
assert 'person_id' in extra_data
assert extra_data['person_id'] is None
assert 'identity_class' in extra_data
assert extra_data['identity_class'] is None
assert 'department' in extra_data
assert extra_data['department'] is None