diff --git a/tests/test_elements_worker/test_elements.py b/tests/test_elements_worker/test_elements.py index 0d5d18ccb91bf47883bd41d36024d0275fda7af6..46a606306bd0be3004c69e863903f824c5b995ae 100644 --- a/tests/test_elements_worker/test_elements.py +++ b/tests/test_elements_worker/test_elements.py @@ -271,10 +271,9 @@ def test_database_arg_cache_missing_version_table( def test_load_corpus_classes_api_error(responses, mock_elements_worker): - mock_elements_worker.corpus_id = "12341234-1234-1234-1234-123412341234" responses.add( responses.GET, - f"http://testserver/api/v1/corpus/{mock_elements_worker.corpus_id}/classes/", + "http://testserver/api/v1/corpus/11111111-1111-1111-1111-111111111111/classes/", status=500, ) @@ -291,33 +290,32 @@ def test_load_corpus_classes_api_error(responses, mock_elements_worker): # We do 5 retries ( "GET", - f"http://testserver/api/v1/corpus/{mock_elements_worker.corpus_id}/classes/", + "http://testserver/api/v1/corpus/11111111-1111-1111-1111-111111111111/classes/", ), ( "GET", - f"http://testserver/api/v1/corpus/{mock_elements_worker.corpus_id}/classes/", + "http://testserver/api/v1/corpus/11111111-1111-1111-1111-111111111111/classes/", ), ( "GET", - f"http://testserver/api/v1/corpus/{mock_elements_worker.corpus_id}/classes/", + "http://testserver/api/v1/corpus/11111111-1111-1111-1111-111111111111/classes/", ), ( "GET", - f"http://testserver/api/v1/corpus/{mock_elements_worker.corpus_id}/classes/", + "http://testserver/api/v1/corpus/11111111-1111-1111-1111-111111111111/classes/", ), ( "GET", - f"http://testserver/api/v1/corpus/{mock_elements_worker.corpus_id}/classes/", + "http://testserver/api/v1/corpus/11111111-1111-1111-1111-111111111111/classes/", ), ] assert not mock_elements_worker.classes def test_load_corpus_classes(responses, mock_elements_worker): - mock_elements_worker.corpus_id = "12341234-1234-1234-1234-123412341234" responses.add( responses.GET, - f"http://testserver/api/v1/corpus/{mock_elements_worker.corpus_id}/classes/", + "http://testserver/api/v1/corpus/11111111-1111-1111-1111-111111111111/classes/", status=200, json={ "count": 3, @@ -348,11 +346,11 @@ def test_load_corpus_classes(responses, mock_elements_worker): ] == BASE_API_CALLS + [ ( "GET", - f"http://testserver/api/v1/corpus/{mock_elements_worker.corpus_id}/classes/", + "http://testserver/api/v1/corpus/11111111-1111-1111-1111-111111111111/classes/", ), ] assert mock_elements_worker.classes == { - "12341234-1234-1234-1234-123412341234": { + "11111111-1111-1111-1111-111111111111": { "good": "0000", "average": "1111", "bad": "2222", diff --git a/tests/test_elements_worker/test_entities.py b/tests/test_elements_worker/test_entities.py index 41e53644834137a116f1248c60418a72421a30cb..84963d796675ff02a2090c3f99fdbe0c2eed7937 100644 --- a/tests/test_elements_worker/test_entities.py +++ b/tests/test_elements_worker/test_entities.py @@ -94,6 +94,7 @@ def test_create_entity_wrong_corpus(monkeypatch, mock_elements_worker): elt = Element({"id": "12341234-1234-1234-1234-123412341234"}) # Triggering an error on metas param, not giving corpus should work since + # ARKINDEX_CORPUS_ID environment variable is set on mock_elements_worker with pytest.raises(AssertionError) as e: mock_elements_worker.create_entity( element=elt,