arkindex-base-worker 0.3.7rc10__py3-none-any.whl → 0.4.0a1__py3-none-any.whl

tests/test_elements_worker/test_elements.py

@@ -17,6 +17,7 @@ from arkindex_worker.cache import (
 from arkindex_worker.models import Element
 from arkindex_worker.worker import ElementsWorker
 from arkindex_worker.worker.element import MissingTypeError
+from tests import CORPUS_ID
 
 from . import BASE_API_CALLS
 
@@ -32,12 +33,11 @@ def test_check_required_types_argument_types(mock_elements_worker):
 
 
 def test_check_required_types(responses, mock_elements_worker):
-    corpus_id = "11111111-1111-1111-1111-111111111111"
     responses.add(
         responses.GET,
-        f"http://testserver/api/v1/corpus/{corpus_id}/",
+        f"http://testserver/api/v1/corpus/{CORPUS_ID}/",
         json={
-            "id": corpus_id,
+            "id": CORPUS_ID,
             "name": "Some Corpus",
             "types": [{"slug": "folder"}, {"slug": "page"}],
         },
@@ -57,13 +57,11 @@ def test_check_required_types(responses, mock_elements_worker):
 
 
 def test_create_missing_types(responses, mock_elements_worker):
-    corpus_id = "11111111-1111-1111-1111-111111111111"
-
     responses.add(
         responses.GET,
-        f"http://testserver/api/v1/corpus/{corpus_id}/",
+        f"http://testserver/api/v1/corpus/{CORPUS_ID}/",
         json={
-            "id": corpus_id,
+            "id": CORPUS_ID,
             "name": "Some Corpus",
             "types": [{"slug": "folder"}, {"slug": "page"}],
         },
@@ -77,7 +75,7 @@ def test_create_missing_types(responses, mock_elements_worker):
                     "slug": "text_line",
                     "display_name": "text_line",
                     "folder": False,
-                    "corpus": corpus_id,
+                    "corpus": CORPUS_ID,
                 }
             )
         ],
@@ -91,7 +89,7 @@ def test_create_missing_types(responses, mock_elements_worker):
                     "slug": "act",
                     "display_name": "act",
                     "folder": False,
-                    "corpus": corpus_id,
+                    "corpus": CORPUS_ID,
                 }
             )
         ],
@@ -135,8 +133,7 @@ def test_list_elements_elements_list_arg_missing_id(
     monkeypatch, tmp_path, mock_elements_worker
 ):
     elements_path = tmp_path / "elements.json"
-    with elements_path.open("w") as f:
-        json.dump([{"type": "volume"}], f)
+    elements_path.write_text(json.dumps([{"type": "volume"}]))
 
     monkeypatch.setenv("TASK_ELEMENTS", str(elements_path))
     worker = ElementsWorker()
@@ -147,18 +144,43 @@ def test_list_elements_elements_list_arg_missing_id(
     assert elt_list == []
 
 
-def test_list_elements_elements_list_arg(monkeypatch, tmp_path, mock_elements_worker):
+def test_list_elements_elements_list_arg_not_uuid(
+    monkeypatch, tmp_path, mock_elements_worker
+):
     elements_path = tmp_path / "elements.json"
-    with elements_path.open("w") as f:
-        json.dump(
+    elements_path.write_text(
+        json.dumps(
             [
                 {"id": "volumeid", "type": "volume"},
                 {"id": "pageid", "type": "page"},
                 {"id": "actid", "type": "act"},
                 {"id": "surfaceid", "type": "surface"},
-            ],
-            f,
+            ]
+        )
+    )
+
+    monkeypatch.setenv("TASK_ELEMENTS", str(elements_path))
+    worker = ElementsWorker()
+    worker.configure()
+
+    with pytest.raises(
+        Exception,
+        match="These element IDs are invalid: volumeid, pageid, actid, surfaceid",
+    ):
+        worker.list_elements()
+
+
+def test_list_elements_elements_list_arg(monkeypatch, tmp_path, mock_elements_worker):
+    elements_path = tmp_path / "elements.json"
+    elements_path.write_text(
+        json.dumps(
+            [
+                {"id": "11111111-1111-1111-1111-111111111111", "type": "volume"},
+                {"id": "22222222-2222-2222-2222-222222222222", "type": "page"},
+                {"id": "33333333-3333-3333-3333-333333333333", "type": "act"},
+            ]
         )
+    )
 
     monkeypatch.setenv("TASK_ELEMENTS", str(elements_path))
     worker = ElementsWorker()
@@ -166,10 +188,14 @@ def test_list_elements_elements_list_arg(monkeypatch, tmp_path, mock_elements_wo
 
     elt_list = worker.list_elements()
 
-    assert elt_list == ["volumeid", "pageid", "actid", "surfaceid"]
+    assert elt_list == [
+        "11111111-1111-1111-1111-111111111111",
+        "22222222-2222-2222-2222-222222222222",
+        "33333333-3333-3333-3333-333333333333",
+    ]
 
 
-def test_list_elements_element_arg(mocker, mock_elements_worker):
+def test_list_elements_element_arg_not_uuid(mocker, mock_elements_worker):
     mocker.patch(
         "arkindex_worker.worker.base.argparse.ArgumentParser.parse_args",
         return_value=Namespace(
@@ -184,23 +210,50 @@ def test_list_elements_element_arg(mocker, mock_elements_worker):
     worker = ElementsWorker()
     worker.configure()
 
+    with pytest.raises(
+        Exception, match="These element IDs are invalid: volumeid, pageid"
+    ):
+        worker.list_elements()
+
+
+def test_list_elements_element_arg(mocker, mock_elements_worker):
+    mocker.patch(
+        "arkindex_worker.worker.base.argparse.ArgumentParser.parse_args",
+        return_value=Namespace(
+            element=[
+                "11111111-1111-1111-1111-111111111111",
+                "22222222-2222-2222-2222-222222222222",
+            ],
+            verbose=False,
+            elements_list=None,
+            database=None,
+            dev=False,
+        ),
+    )
+
+    worker = ElementsWorker()
+    worker.configure()
+
     elt_list = worker.list_elements()
 
-    assert elt_list == ["volumeid", "pageid"]
+    assert elt_list == [
+        "11111111-1111-1111-1111-111111111111",
+        "22222222-2222-2222-2222-222222222222",
+    ]
 
 
 def test_list_elements_both_args_error(mocker, mock_elements_worker, tmp_path):
     elements_path = tmp_path / "elements.json"
-    with elements_path.open("w") as f:
-        json.dump(
+    elements_path.write_text(
+        json.dumps(
             [
                 {"id": "volumeid", "type": "volume"},
                 {"id": "pageid", "type": "page"},
                 {"id": "actid", "type": "act"},
                 {"id": "surfaceid", "type": "surface"},
-            ],
-            f,
+            ]
         )
+    )
     mocker.patch(
         "arkindex_worker.worker.base.argparse.ArgumentParser.parse_args",
         return_value=Namespace(
@@ -272,8 +325,8 @@ def test_database_arg_cache_missing_version_table(
 def test_load_corpus_classes_api_error(responses, mock_elements_worker):
     responses.add(
         responses.GET,
-        "http://testserver/api/v1/corpus/11111111-1111-1111-1111-111111111111/classes/",
-        status=500,
+        f"http://testserver/api/v1/corpus/{CORPUS_ID}/classes/",
+        status=418,
     )
 
     assert not mock_elements_worker.classes
@@ -289,23 +342,23 @@ def test_load_corpus_classes_api_error(responses, mock_elements_worker):
         # We do 5 retries
         (
             "GET",
-            "http://testserver/api/v1/corpus/11111111-1111-1111-1111-111111111111/classes/",
+            f"http://testserver/api/v1/corpus/{CORPUS_ID}/classes/",
         ),
         (
             "GET",
-            "http://testserver/api/v1/corpus/11111111-1111-1111-1111-111111111111/classes/",
+            f"http://testserver/api/v1/corpus/{CORPUS_ID}/classes/",
         ),
         (
             "GET",
-            "http://testserver/api/v1/corpus/11111111-1111-1111-1111-111111111111/classes/",
+            f"http://testserver/api/v1/corpus/{CORPUS_ID}/classes/",
         ),
         (
             "GET",
-            "http://testserver/api/v1/corpus/11111111-1111-1111-1111-111111111111/classes/",
+            f"http://testserver/api/v1/corpus/{CORPUS_ID}/classes/",
         ),
         (
             "GET",
-            "http://testserver/api/v1/corpus/11111111-1111-1111-1111-111111111111/classes/",
+            f"http://testserver/api/v1/corpus/{CORPUS_ID}/classes/",
         ),
     ]
     assert not mock_elements_worker.classes
@@ -314,7 +367,7 @@ def test_load_corpus_classes_api_error(responses, mock_elements_worker):
 def test_load_corpus_classes(responses, mock_elements_worker):
     responses.add(
         responses.GET,
-        "http://testserver/api/v1/corpus/11111111-1111-1111-1111-111111111111/classes/",
+        f"http://testserver/api/v1/corpus/{CORPUS_ID}/classes/",
         status=200,
         json={
             "count": 3,
@@ -345,7 +398,7 @@ def test_load_corpus_classes(responses, mock_elements_worker):
     ] == BASE_API_CALLS + [
         (
             "GET",
-            "http://testserver/api/v1/corpus/11111111-1111-1111-1111-111111111111/classes/",
+            f"http://testserver/api/v1/corpus/{CORPUS_ID}/classes/",
         ),
     ]
     assert mock_elements_worker.classes == {
@@ -532,14 +585,14 @@ def test_create_sub_element_api_error(responses, mock_elements_worker):
     elt = Element(
         {
             "id": "12341234-1234-1234-1234-123412341234",
-            "corpus": {"id": "11111111-1111-1111-1111-111111111111"},
+            "corpus": {"id": CORPUS_ID},
             "zone": {"image": {"id": "22222222-2222-2222-2222-222222222222"}},
         }
     )
     responses.add(
         responses.POST,
         "http://testserver/api/v1/elements/create/",
-        status=500,
+        status=418,
     )
 
     with pytest.raises(ErrorResponse):
@@ -550,17 +603,10 @@ def test_create_sub_element_api_error(responses, mock_elements_worker):
             polygon=[[1, 1], [2, 2], [2, 1], [1, 2]],
         )
 
-    assert len(responses.calls) == len(BASE_API_CALLS) + 5
+    assert len(responses.calls) == len(BASE_API_CALLS) + 1
     assert [
         (call.request.method, call.request.url) for call in responses.calls
-    ] == BASE_API_CALLS + [
-        # We retry 5 times the API call
-        ("POST", "http://testserver/api/v1/elements/create/"),
-        ("POST", "http://testserver/api/v1/elements/create/"),
-        ("POST", "http://testserver/api/v1/elements/create/"),
-        ("POST", "http://testserver/api/v1/elements/create/"),
-        ("POST", "http://testserver/api/v1/elements/create/"),
-    ]
+    ] == BASE_API_CALLS + [("POST", "http://testserver/api/v1/elements/create/")]
 
 
 @pytest.mark.parametrize("slim_output", [True, False])
@@ -568,13 +614,13 @@ def test_create_sub_element(responses, mock_elements_worker, slim_output):
     elt = Element(
         {
             "id": "12341234-1234-1234-1234-123412341234",
-            "corpus": {"id": "11111111-1111-1111-1111-111111111111"},
+            "corpus": {"id": CORPUS_ID},
             "zone": {"image": {"id": "22222222-2222-2222-2222-222222222222"}},
         }
     )
     child_elt = {
         "id": "12345678-1234-1234-1234-123456789123",
-        "corpus": {"id": "11111111-1111-1111-1111-111111111111"},
+        "corpus": {"id": CORPUS_ID},
         "zone": {"image": {"id": "22222222-2222-2222-2222-222222222222"}},
     }
     responses.add(
@@ -605,7 +651,7 @@ def test_create_sub_element(responses, mock_elements_worker, slim_output):
         "type": "something",
         "name": "0",
         "image": None,
-        "corpus": "11111111-1111-1111-1111-111111111111",
+        "corpus": CORPUS_ID,
         "polygon": [[1, 1], [2, 2], [2, 1], [1, 2]],
         "parent": "12341234-1234-1234-1234-123412341234",
         "worker_run_id": "56785678-5678-5678-5678-567856785678",
@@ -621,7 +667,7 @@ def test_create_sub_element_confidence(responses, mock_elements_worker):
     elt = Element(
         {
             "id": "12341234-1234-1234-1234-123412341234",
-            "corpus": {"id": "11111111-1111-1111-1111-111111111111"},
+            "corpus": {"id": CORPUS_ID},
             "zone": {"image": {"id": "22222222-2222-2222-2222-222222222222"}},
         }
     )
@@ -650,7 +696,7 @@ def test_create_sub_element_confidence(responses, mock_elements_worker):
         "type": "something",
         "name": "0",
         "image": None,
-        "corpus": "11111111-1111-1111-1111-111111111111",
+        "corpus": CORPUS_ID,
         "polygon": [[1, 1], [2, 2], [2, 1], [1, 2]],
         "parent": "12341234-1234-1234-1234-123412341234",
         "worker_run_id": "56785678-5678-5678-5678-567856785678",
@@ -931,7 +977,7 @@ def test_create_elements_api_error(responses, mock_elements_worker):
     responses.add(
         responses.POST,
         "http://testserver/api/v1/element/12341234-1234-1234-1234-123412341234/children/bulk/",
-        status=500,
+        status=418,
     )
 
     with pytest.raises(ErrorResponse):
@@ -946,31 +992,14 @@ def test_create_elements_api_error(responses, mock_elements_worker):
             ],
         )
 
-    assert len(responses.calls) == len(BASE_API_CALLS) + 5
+    assert len(responses.calls) == len(BASE_API_CALLS) + 1
     assert [
         (call.request.method, call.request.url) for call in responses.calls
     ] == BASE_API_CALLS + [
-        # We retry 5 times the API call
         (
             "POST",
             "http://testserver/api/v1/element/12341234-1234-1234-1234-123412341234/children/bulk/",
-        ),
-        (
-            "POST",
-            "http://testserver/api/v1/element/12341234-1234-1234-1234-123412341234/children/bulk/",
-        ),
-        (
-            "POST",
-            "http://testserver/api/v1/element/12341234-1234-1234-1234-123412341234/children/bulk/",
-        ),
-        (
-            "POST",
-            "http://testserver/api/v1/element/12341234-1234-1234-1234-123412341234/children/bulk/",
-        ),
-        (
-            "POST",
-            "http://testserver/api/v1/element/12341234-1234-1234-1234-123412341234/children/bulk/",
-        ),
+        )
     ]
 
 
@@ -1275,7 +1304,7 @@ def test_create_element_parent_api_error(responses, mock_elements_worker):
     responses.add(
         responses.POST,
         "http://testserver/api/v1/element/497f6eca-6276-4993-bfeb-53cbbbba6f08/parent/12341234-1234-1234-1234-123412341234/",
-        status=500,
+        status=418,
     )
 
     with pytest.raises(ErrorResponse):
@@ -1284,16 +1313,15 @@ def test_create_element_parent_api_error(responses, mock_elements_worker):
             child=child,
         )
 
-    assert len(responses.calls) == len(BASE_API_CALLS) + 5
+    assert len(responses.calls) == len(BASE_API_CALLS) + 1
     assert [
         (call.request.method, call.request.url) for call in responses.calls
     ] == BASE_API_CALLS + [
-        # We retry 5 times the API call
         (
             "POST",
             "http://testserver/api/v1/element/497f6eca-6276-4993-bfeb-53cbbbba6f08/parent/12341234-1234-1234-1234-123412341234/",
-        ),
-    ] * 5
+        )
+    ]
 
 
 def test_create_element_parent(responses, mock_elements_worker):
@@ -1527,7 +1555,7 @@ def test_partial_update_element_api_error(responses, mock_elements_worker):
     responses.add(
         responses.PATCH,
         f"http://testserver/api/v1/element/{elt.id}/",
-        status=500,
+        status=418,
     )
 
     with pytest.raises(ErrorResponse):
@@ -1538,17 +1566,10 @@ def test_partial_update_element_api_error(responses, mock_elements_worker):
             polygon=[[1, 1], [2, 2], [2, 1], [1, 2]],
         )
 
-    assert len(responses.calls) == len(BASE_API_CALLS) + 5
+    assert len(responses.calls) == len(BASE_API_CALLS) + 1
     assert [
         (call.request.method, call.request.url) for call in responses.calls
-    ] == BASE_API_CALLS + [
-        # We retry 5 times the API call
-        ("PATCH", f"http://testserver/api/v1/element/{elt.id}/"),
-        ("PATCH", f"http://testserver/api/v1/element/{elt.id}/"),
-        ("PATCH", f"http://testserver/api/v1/element/{elt.id}/"),
-        ("PATCH", f"http://testserver/api/v1/element/{elt.id}/"),
-        ("PATCH", f"http://testserver/api/v1/element/{elt.id}/"),
-    ]
+    ] == BASE_API_CALLS + [("PATCH", f"http://testserver/api/v1/element/{elt.id}/")]
 
 
 @pytest.mark.usefixtures("_mock_cached_elements", "_mock_cached_images")
@@ -1860,7 +1881,7 @@ def test_list_element_children_api_error(responses, mock_elements_worker):
     responses.add(
         responses.GET,
         "http://testserver/api/v1/elements/12341234-1234-1234-1234-123412341234/children/",
-        status=500,
+        status=418,
     )
 
     with pytest.raises(
@@ -2392,7 +2413,7 @@ def test_list_element_parents_api_error(responses, mock_elements_worker):
     responses.add(
         responses.GET,
         "http://testserver/api/v1/elements/12341234-1234-1234-1234-123412341234/parents/",
-        status=500,
+        status=418,
     )
 
     with pytest.raises(

tests/test_elements_worker/test_entities.py

@@ -15,6 +15,7 @@ from arkindex_worker.cache import (
 from arkindex_worker.models import Transcription
 from arkindex_worker.worker.entity import MissingEntityType
 from arkindex_worker.worker.transcription import TextOrientation
+from tests import CORPUS_ID
 
 from . import BASE_API_CALLS
 
@@ -90,7 +91,7 @@ def test_create_entity_api_error(responses, mock_elements_worker):
     responses.add(
         responses.POST,
         "http://testserver/api/v1/entity/",
-        status=500,
+        status=418,
     )
 
     with pytest.raises(ErrorResponse):
@@ -99,17 +100,10 @@ def test_create_entity_api_error(responses, mock_elements_worker):
             type="person",
         )
 
-    assert len(responses.calls) == len(BASE_API_CALLS) + 5
+    assert len(responses.calls) == len(BASE_API_CALLS) + 1
     assert [
         (call.request.method, call.request.url) for call in responses.calls
-    ] == BASE_API_CALLS + [
-        # We retry 5 times the API call
-        ("POST", "http://testserver/api/v1/entity/"),
-        ("POST", "http://testserver/api/v1/entity/"),
-        ("POST", "http://testserver/api/v1/entity/"),
-        ("POST", "http://testserver/api/v1/entity/"),
-        ("POST", "http://testserver/api/v1/entity/"),
-    ]
+    ] == BASE_API_CALLS + [("POST", "http://testserver/api/v1/entity/")]
 
 
 def test_create_entity(responses, mock_elements_worker):
@@ -139,7 +133,7 @@ def test_create_entity(responses, mock_elements_worker):
         "type_id": "person-entity-type-id",
         "metas": {},
         "validated": None,
-        "corpus": "11111111-1111-1111-1111-111111111111",
+        "corpus": CORPUS_ID,
         "worker_run_id": "56785678-5678-5678-5678-567856785678",
     }
     assert entity_id == "12345678-1234-1234-1234-123456789123"
@@ -152,7 +146,7 @@ def test_create_entity_missing_type(responses, mock_elements_worker):
     # Call to list entity types
     responses.add(
         responses.GET,
-        "http://testserver/api/v1/corpus/11111111-1111-1111-1111-111111111111/entity-types/",
+        f"http://testserver/api/v1/corpus/{CORPUS_ID}/entity-types/",
         status=200,
         json={
             "count": 1,
@@ -177,7 +171,7 @@ def test_create_entity_missing_type(responses, mock_elements_worker):
     ] == BASE_API_CALLS + [
         (
             "GET",
-            "http://testserver/api/v1/corpus/11111111-1111-1111-1111-111111111111/entity-types/",
+            f"http://testserver/api/v1/corpus/{CORPUS_ID}/entity-types/",
         ),
     ]
 
@@ -209,7 +203,7 @@ def test_create_entity_with_cache(responses, mock_elements_worker_with_cache):
         "type_id": "person-entity-type-id",
         "metas": {},
         "validated": None,
-        "corpus": "11111111-1111-1111-1111-111111111111",
+        "corpus": CORPUS_ID,
         "worker_run_id": "56785678-5678-5678-5678-567856785678",
     }
     assert entity_id == "12345678-1234-1234-1234-123456789123"
@@ -387,7 +381,7 @@ def test_create_transcription_entity_api_error(responses, mock_elements_worker):
     responses.add(
         responses.POST,
         "http://testserver/api/v1/transcription/11111111-1111-1111-1111-111111111111/entity/",
-        status=500,
+        status=418,
     )
 
     with pytest.raises(ErrorResponse):
@@ -403,31 +397,14 @@ def test_create_transcription_entity_api_error(responses, mock_elements_worker):
             length=10,
         )
 
-    assert len(responses.calls) == len(BASE_API_CALLS) + 5
+    assert len(responses.calls) == len(BASE_API_CALLS) + 1
     assert [
         (call.request.method, call.request.url) for call in responses.calls
     ] == BASE_API_CALLS + [
-        # We retry 5 times the API call
         (
             "POST",
             "http://testserver/api/v1/transcription/11111111-1111-1111-1111-111111111111/entity/",
-        ),
-        (
-            "POST",
-            "http://testserver/api/v1/transcription/11111111-1111-1111-1111-111111111111/entity/",
-        ),
-        (
-            "POST",
-            "http://testserver/api/v1/transcription/11111111-1111-1111-1111-111111111111/entity/",
-        ),
-        (
-            "POST",
-            "http://testserver/api/v1/transcription/11111111-1111-1111-1111-111111111111/entity/",
-        ),
-        (
-            "POST",
-            "http://testserver/api/v1/transcription/11111111-1111-1111-1111-111111111111/entity/",
-        ),
+        )
     ]
 
 
@@ -746,10 +723,9 @@ def test_list_transcription_entities(fake_dummy_worker):
 
 
 def test_list_corpus_entities(responses, mock_elements_worker):
-    corpus_id = "11111111-1111-1111-1111-111111111111"
     responses.add(
         responses.GET,
-        f"http://testserver/api/v1/corpus/{corpus_id}/entities/",
+        f"http://testserver/api/v1/corpus/{CORPUS_ID}/entities/",
         json={
             "count": 1,
             "next": None,
@@ -775,7 +751,7 @@ def test_list_corpus_entities(responses, mock_elements_worker):
     ] == BASE_API_CALLS + [
         (
             "GET",
-            f"http://testserver/api/v1/corpus/{corpus_id}/entities/",
+            f"http://testserver/api/v1/corpus/{CORPUS_ID}/entities/",
         ),
     ]
 
@@ -807,13 +783,13 @@ def test_check_required_entity_types(responses, mock_elements_worker):
             matchers.json_params_matcher(
                 {
                     "name": "new-entity",
-                    "corpus": "11111111-1111-1111-1111-111111111111",
+                    "corpus": CORPUS_ID,
                 }
             )
         ],
         json={
             "id": "new-entity-id",
-            "corpus": "11111111-1111-1111-1111-111111111111",
+            "corpus": CORPUS_ID,
             "name": "new-entity",
             "color": "ffd1b3",
         },

tests/test_elements_worker/test_image.py

@@ -0,0 +1,65 @@
+import json
+
+import pytest
+from apistar.exceptions import ErrorResponse
+
+from . import BASE_API_CALLS
+
+
+@pytest.mark.parametrize(
+    ("data", "error_message"),
+    [
+        (None, "url shouldn't be null and should be of type str"),
+        (1234, "url shouldn't be null and should be of type str"),
+    ],
+)
+def test_create_iiif_url_wrong_data(data, error_message, mock_elements_worker):
+    with pytest.raises(AssertionError, match=error_message):
+        mock_elements_worker.create_iiif_url(url=data)
+
+
+def test_create_iiif_url_api_error(responses, mock_elements_worker):
+    responses.add(
+        responses.POST,
+        "http://testserver/api/v1/image/iiif/url/",
+        status=418,
+    )
+
+    with pytest.raises(ErrorResponse):
+        mock_elements_worker.create_iiif_url("http://url/to/my/image")
+
+    assert len(responses.calls) == len(BASE_API_CALLS) + 1
+    assert [
+        (call.request.method, call.request.url) for call in responses.calls
+    ] == BASE_API_CALLS + [("POST", "http://testserver/api/v1/image/iiif/url/")]
+
+
+def test_create_iiif_url(responses, mock_elements_worker):
+    responses.add(
+        responses.POST,
+        "http://testserver/api/v1/image/iiif/url/",
+        status=201,
+        json={
+            "id": "cafecafe-cafe-cafe-cafe-cafecafecafe",
+            "url": "http://url/to/my/image",
+            "status": "checked",
+            "server": {
+                "id": 5,
+                "display_name": "My server",
+                "url": "http://url/to/my",
+                "max_width": 42,
+                "max_height": 42,
+            },
+        },
+    )
+
+    image = mock_elements_worker.create_iiif_url("http://url/to/my/image")
+
+    assert len(responses.calls) == len(BASE_API_CALLS) + 1
+    assert [
+        (call.request.method, call.request.url) for call in responses.calls
+    ] == BASE_API_CALLS + [("POST", "http://testserver/api/v1/image/iiif/url/")]
+    assert json.loads(responses.calls[-1].request.body) == {
+        "url": "http://url/to/my/image"
+    }
+    assert image.id == "cafecafe-cafe-cafe-cafe-cafecafecafe"