Fix tests

statisticsnorway · Jul 24, 2024 · 663a439 · 663a439
1 parent db3aefb
commit 663a439
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/src/datadoc/backend/core.py b/src/datadoc/backend/core.py
@@ -124,6 +124,8 @@ def _extract_metadata_from_files(self) -> None:
         - The 'contains_personal_data' attribute is set to False if not specified.
         - A lookup dictionary for variables is created based on their short names.
         """
+        extracted_metadata: model.DatadocMetadata | None = None
+        existing_metadata: model.DatadocMetadata | None = None
         if self.metadata_document is not None and self.metadata_document.exists():
             existing_metadata = self._extract_metadata_from_existing_document(
                 self.metadata_document,

diff --git a/src/datadoc/backend/utils.py b/src/datadoc/backend/utils.py
@@ -20,6 +20,7 @@
 )
 from datadoc.enums import Assessment
 from datadoc.enums import DataSetState
+from datadoc.enums import VariableRole
 
 
 def normalize_path(path: str) -> pathlib.Path | CloudPath:
@@ -87,6 +88,8 @@ def set_default_values_variables(variables: list) -> None:
             v.id = uuid.uuid4()
         if v.is_personal_data is None:
             v.is_personal_data = model.IsPersonalData.NOT_PERSONAL_DATA
+        if v.variable_role is None:
+            v.variable_role = VariableRole.MEASURE
 
 
 def set_default_values_dataset(dataset: model.Dataset) -> None: