def fix_submission_number_does_nothing_if_method_is_not_hepcrawl(): schema = load_schema('hep') subschema = schema['properties']['acquisition_source'] data = { 'acquisition_source': { 'method': 'submitter', 'submission_number': '869215', }, } extra_data = {} assert validate(data['acquisition_source'], subschema) is None obj = MockObj(data, extra_data) eng = MockEng() fix_submission_number(obj, eng) expected = { 'method': 'submitter', 'submission_number': '869215', } result = obj.data['acquisition_source'] assert validate(result, subschema) is None assert expected == result
def test_fix_submission_number(): schema = load_schema('hep') subschema = schema['properties']['acquisition_source'] data = { 'acquisition_source': { 'method': 'hepcrawl', 'submission_number': '751e374a017311e896d6fa163ec92c6a', }, } extra_data = {} assert validate(data['acquisition_source'], subschema) is None obj = MockObj(data, extra_data) eng = MockEng() fix_submission_number(obj, eng) expected = { 'method': 'hepcrawl', 'submission_number': '1', } result = obj.data['acquisition_source'] assert validate(result, subschema) is None assert expected == result