Helper for ListCorpusEntities

Closes #131 (closed)

Requires !209 (merged)

tests/test_elements_worker/test_classifications.py

@@ -15,7 +15,7 @@ def test_get_ml_class_id_load_classes(responses, mock_elements_worker):
     corpus_id = "11111111-1111-1111-1111-111111111111"
     responses.add(
         responses.GET,
-        f"http://testserver/api/v1/corpus/{mock_elements_worker.corpus_id}/classes/",
+        f"http://testserver/api/v1/corpus/{corpus_id}/classes/",
         status=200,
         json={
             "count": 1,
@@ -42,7 +42,7 @@ def test_get_ml_class_id_load_classes(responses, mock_elements_worker):
         ),
     ]
     assert mock_elements_worker.classes == {
-        "12341234-1234-1234-1234-123412341234": {"good": "0000"}
+        "11111111-1111-1111-1111-111111111111": {"good": "0000"}
     }
     assert ml_class_id == "0000"
 
@@ -51,19 +51,19 @@ def test_get_ml_class_id_inexistant_class(mock_elements_worker, responses):
     # A missing class is now created automatically
     corpus_id = "11111111-1111-1111-1111-111111111111"
     mock_elements_worker.classes = {
-        "12341234-1234-1234-1234-123412341234": {"good": "0000"}
+        "11111111-1111-1111-1111-111111111111": {"good": "0000"}
     }
 
     responses.add(
         responses.POST,
-        f"http://testserver/api/v1/corpus/{mock_elements_worker.corpus_id}/classes/",
+        f"http://testserver/api/v1/corpus/{corpus_id}/classes/",
         status=201,
         json={"id": "new-ml-class-1234"},
     )
 
     # Missing class at first
     assert mock_elements_worker.classes == {
-        "12341234-1234-1234-1234-123412341234": {"good": "0000"}
+        "11111111-1111-1111-1111-111111111111": {"good": "0000"}
     }
 
     ml_class_id = mock_elements_worker.get_ml_class_id("bad")
@@ -71,7 +71,7 @@ def test_get_ml_class_id_inexistant_class(mock_elements_worker, responses):
 
     # Now it's available
     assert mock_elements_worker.classes == {
-        "12341234-1234-1234-1234-123412341234": {
+        "11111111-1111-1111-1111-111111111111": {
             "good": "0000",
             "bad": "new-ml-class-1234",
         }
@@ -80,7 +80,7 @@ def test_get_ml_class_id_inexistant_class(mock_elements_worker, responses):
 
 def test_get_ml_class_id(mock_elements_worker):
     mock_elements_worker.classes = {
-        "12341234-1234-1234-1234-123412341234": {"good": "0000"}
+        "11111111-1111-1111-1111-111111111111": {"good": "0000"}
     }
 
     ml_class_id = mock_elements_worker.get_ml_class_id("good")
@@ -93,7 +93,7 @@ def test_get_ml_class_reload(responses, mock_elements_worker):
     # Add some initial classes
     responses.add(
         responses.GET,
-        f"http://testserver/api/v1/corpus/{mock_elements_worker.corpus_id}/classes/",
+        f"http://testserver/api/v1/corpus/{corpus_id}/classes/",
         json={
             "count": 1,
             "next": None,
@@ -109,7 +109,7 @@ def test_get_ml_class_reload(responses, mock_elements_worker):
     # Invalid response when trying to create class2
     responses.add(
         responses.POST,
-        f"http://testserver/api/v1/corpus/{mock_elements_worker.corpus_id}/classes/",
+        f"http://testserver/api/v1/corpus/{corpus_id}/classes/",
         status=400,
         json={"non_field_errors": "Already exists"},
     )
@@ -117,7 +117,7 @@ def test_get_ml_class_reload(responses, mock_elements_worker):
     # Add both classes (class2 is created by another process)
     responses.add(
         responses.GET,
-        f"http://testserver/api/v1/corpus/{mock_elements_worker.corpus_id}/classes/",
+        f"http://testserver/api/v1/corpus/{corpus_id}/classes/",
         json={
             "count": 2,
             "next": None,
@@ -139,7 +139,7 @@ def test_get_ml_class_reload(responses, mock_elements_worker):
 
     assert len(responses.calls) == len(BASE_API_CALLS) + 3
     assert mock_elements_worker.classes == {
-        mock_elements_worker.corpus_id: {
+        corpus_id: {
             "class1": "class1_id",
             "class2": "class2_id",
         }