HOTFIX Add has_version None handler

NYPL · May 18, 2022 · 3a0f5b2 · 3a0f5b2
1 parent 330b72f
commit 3a0f5b2
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/model/postgres/record.py b/model/postgres/record.py
@@ -66,7 +66,9 @@ def has_version(self):
 
     @has_version.setter
     def has_version(self, versionNum):
-        if self.languages != [] and self.languages != None:
+        if versionNum is None:
+            self._has_version = versionNum
+        elif self.languages != [] and self.languages != None:
             editionNo = extract(versionNum, self.languages[0].split('|')[0])
             self._has_version = f'{versionNum}|{editionNo}'
         else:

diff --git a/tests/unit/test_model_record.py b/tests/unit/test_model_record.py
@@ -17,6 +17,11 @@ def test_no_language(self, testRecord):
         testRecord.has_version = 'first edition'
         assert testRecord.has_version == 'first edition|1'
 
+    # Test to assert that the setter properly handles null values
+    def test_no_edition(self, testRecord):
+        testRecord.has_version = None
+        assert testRecord.has_version is None
+
     # Failed Test for setter method
     def test_has_version_failure(self, testRecord):
         testRecord.has_version = 'other edition'