diff --git a/inspire_schemas/builders/data.py b/inspire_schemas/builders/data.py index 6aa1b84e..7e1f9c0d 100644 --- a/inspire_schemas/builders/data.py +++ b/inspire_schemas/builders/data.py @@ -29,6 +29,7 @@ import warnings import idutils +from inspire_utils.date import normalize_date from six import python_2_unicode_compatible, text_type from inspire_schemas.builders.builder import RecordBuilder @@ -92,6 +93,15 @@ def add_abstract(self, abstract, source=None): ), ) + @filter_empty_parameters + def add_creation_date(self, date=None): + """ + Args: + date (str) + """ + if date is not None: + self.record['creation_date'] = normalize_date(date) + @filter_empty_parameters def add_accelerator_experiments_legacy_name(self, legacy_name): """Add legacy name in accelerator experiment. diff --git a/tests/integration/test_builders.py b/tests/integration/test_builders.py index fb6213a8..09adccf4 100644 --- a/tests/integration/test_builders.py +++ b/tests/integration/test_builders.py @@ -448,7 +448,6 @@ def test_data_builder(data_data): 'control_number': data_data['control_number'], 'deleted': data_data['deleted'], 'deleted_records': data_data['deleted_records'], - 'creation_date': data_data['creation_date'], 'legacy_version': data_data['legacy_version'], 'new_record': data_data['new_record'], 'self': data_data['self'], @@ -460,6 +459,10 @@ def test_data_builder(data_data): builder.add_doi(doi['value'], doi['source'], doi['material']) assert builder.record['dois'] == dois + creation_date = data_data['creation_date'] + builder.add_creation_date(creation_date) + assert builder.record['creation_date'] == creation_date + collaborations = data_data['collaborations'] for collaboration in collaborations: builder.add_collaboration(collaboration['value']) diff --git a/tests/unit/test_data_builder.py b/tests/unit/test_data_builder.py index 1c3bd8a3..fb39cb7f 100644 --- a/tests/unit/test_data_builder.py +++ b/tests/unit/test_data_builder.py @@ -72,6 +72,12 @@ def test_add_title(): ] assert result == expected +def test_add_creation_date(): + builder = DataBuilder() + creation_date = "2020-01-01" + builder.add_creation_date(creation_date) + result = builder.record.get("creation_date") + assert result == creation_date def test_add_doi(): builder = DataBuilder()