fix[data builder]: Access doi value directly in literature

Signed-off-by: Lorenzo Vagliano
inspirehep · Jan 13, 2025 · 3db8ae7 · 3db8ae7
1 parent 0afd2ac
commit 3db8ae7
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/inspire_schemas/builders/data.py b/inspire_schemas/builders/data.py
@@ -311,6 +311,6 @@ def add_literature(self, doi=None, record=None):
         }
 
         if doi:
-            literature_dict['doi'] = doi
+            literature_dict['doi']["value"] = doi
 
         self._append_to('literature', literature_dict)
diff --git a/tests/unit/test_data_builder.py b/tests/unit/test_data_builder.py
@@ -112,7 +112,10 @@ def test_add_accelerator_experiment():
 def test_add_literature():
     builder = DataBuilder()
 
-    doi = "10.1234/example.doi"
+    doi = {
+        "source": "amet Ut",
+        "value": "10.912/W/K[Bz",
+    }
     record = {"$ref": "http://example.com/api/literature/123"}
 
     builder.add_literature(doi, record=record)