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