Errbit | fixing tests

OpenConceptLab · Jun 24, 2022 · 79aadeb · 79aadeb
1 parent 43a03a2
commit 79aadeb
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 12 deletions.
diff --git a/core/integration_tests/tests_concepts.py b/core/integration_tests/tests_concepts.py
@@ -1819,12 +1819,12 @@ def test_get_200(self):
         expansion3 = ExpansionFactory(collection_version=collection3)
         collection3.expansion_uri = expansion3.uri
         collection3.save()
-        expansion1.concepts.add(concept1.get_latest_version())
-        expansion2.concepts.add(concept1.get_latest_version())
-        expansion3.concepts.add(concept1.get_latest_version())
-        expansion1.concepts.add(concept2.get_latest_version())
-        expansion2.concepts.add(concept2.get_latest_version())
-        expansion3.concepts.add(concept2.get_latest_version())
+        expansion1.concepts.add(concept1)
+        expansion2.concepts.add(concept1)
+        expansion3.concepts.add(concept1)
+        expansion1.concepts.add(concept2)
+        expansion2.concepts.add(concept2)
+        expansion3.concepts.add(concept2)
 
         response = self.client.get(concept1.url + 'collection-versions/')
 

diff --git a/core/integration_tests/tests_mappings.py b/core/integration_tests/tests_mappings.py
@@ -842,12 +842,12 @@ def test_get_200(self):
         expansion3 = ExpansionFactory(collection_version=collection3)
         collection3.expansion_uri = expansion3.uri
         collection3.save()
-        expansion1.mappings.add(mapping1.get_latest_version())
-        expansion2.mappings.add(mapping1.get_latest_version())
-        expansion3.mappings.add(mapping1.get_latest_version())
-        expansion1.mappings.add(mapping2.get_latest_version())
-        expansion2.mappings.add(mapping2.get_latest_version())
-        expansion3.mappings.add(mapping2.get_latest_version())
+        expansion1.mappings.add(mapping1)
+        expansion2.mappings.add(mapping1)
+        expansion3.mappings.add(mapping1)
+        expansion1.mappings.add(mapping2)
+        expansion2.mappings.add(mapping2)
+        expansion3.mappings.add(mapping2)
 
         response = self.client.get(mapping1.url + 'collection-versions/')