arkindex-base-worker 0.4.0__py3-none-any.whl → 0.4.0a1__py3-none-any.whl

tests/test_merge.py

@@ -161,7 +161,7 @@ def test_merge_from_worker(
     """
     responses.add(
         responses.GET,
-        "http://testserver/api/v1/task/my_task/",
+        "http://testserver/api/v1/task/my_task/from-agent/",
         status=200,
         json={"parents": ["first", "second"]},
     )
@@ -181,6 +181,7 @@ def test_merge_from_worker(
     (tmp_path / "my_task").mkdir()
     mock_base_worker_with_cache.args = mock_base_worker_with_cache.parser.parse_args()
     mock_base_worker_with_cache.configure()
+    mock_base_worker_with_cache.configure_cache()
     # Store parent tasks IDs as attribute
     assert mock_base_worker_with_cache.task_parents == ["first", "second"]
 

tests/test_utils.py

@@ -1,18 +1,15 @@
-import logging
+from pathlib import Path
 
 import pytest
 
-from arkindex_worker.cache import unsupported_cache
 from arkindex_worker.utils import (
-    DEFAULT_BATCH_SIZE,
-    batch_publication,
     close_delete_file,
     extract_tar_zst_archive,
     parse_source_id,
 )
-from tests import FIXTURES_DIR
 
-ARCHIVE = FIXTURES_DIR / "archive.tar.zst"
+FIXTURES = Path(__file__).absolute().parent / "data"
+ARCHIVE = FIXTURES / "archive.tar.zst"
 
 
 @pytest.mark.parametrize(
@@ -58,63 +55,3 @@ def test_close_delete_file(tmp_path):
     close_delete_file(archive_fd, archive_path)
 
     assert not archive_path.exists()
-
-
-class TestMixin:
-    def __init__(self, use_cache: bool = False):
-        """
-        Args:
-            use_cache (bool, optional): To mock BaseWorker.use_cache attribute. Defaults to False.
-        """
-        self.use_cache = use_cache
-
-    @batch_publication
-    def custom_publication_in_batches(self, batch_size: int = DEFAULT_BATCH_SIZE):
-        return batch_size
-
-    @unsupported_cache
-    @batch_publication
-    def custom_publication_in_batches_without_cache(
-        self, batch_size: int = DEFAULT_BATCH_SIZE
-    ):
-        return batch_size
-
-
-def test_batch_publication_decorator_no_parameter():
-    assert TestMixin().custom_publication_in_batches() == DEFAULT_BATCH_SIZE
-
-
-@pytest.mark.parametrize("wrong_batch_size", [None, "not an int", 0])
-def test_batch_publication_decorator_wrong_parameter(wrong_batch_size):
-    with pytest.raises(
-        AssertionError,
-        match="batch_size shouldn't be null and should be a strictly positive integer",
-    ):
-        TestMixin().custom_publication_in_batches(batch_size=wrong_batch_size)
-
-
-@pytest.mark.parametrize("batch_size", [1, 10, DEFAULT_BATCH_SIZE])
-def test_batch_publication_decorator_right_parameter(batch_size):
-    assert (
-        TestMixin().custom_publication_in_batches(batch_size=batch_size) == batch_size
-    )
-
-
-def test_batch_publication_decorator_alongside_unsupported_cache(caplog):
-    # Capture log messages
-    caplog.clear()
-    with caplog.at_level(logging.WARNING):
-        # Call the helper
-        assert (
-            TestMixin(use_cache=True).custom_publication_in_batches_without_cache()
-            == DEFAULT_BATCH_SIZE
-        )
-
-    # Check logs
-    assert caplog.record_tuples == [
-        (
-            "arkindex_worker",
-            logging.WARNING,
-            "This API helper `custom_publication_in_batches_without_cache` did not update the cache database",
-        ),
-    ]

arkindex_base_worker-0.4.0.dist-info/RECORD

@@ -1,61 +0,0 @@
-arkindex_worker/__init__.py,sha256=OlgCtTC9MaWeejviY0a3iQpALcRQGMVArFVVYwTF6I8,162
-arkindex_worker/cache.py,sha256=qTblc_zKdYC47Wip6_O9Jf5qBkQW2ozQQrg-nsx1WuY,11221
-arkindex_worker/image.py,sha256=oEgVCrSHiGh3D5-UXfM6PvT17TttSxC0115irpvB3Dw,18581
-arkindex_worker/models.py,sha256=bPQzGZNs5a6z6DEcygsa8T33VOqPlMUbwKzHqlKzwbw,9923
-arkindex_worker/utils.py,sha256=MbbJT8oh8DMHHR-vidFeXdUH0TSXGWm7ZDGWzrRXoEY,9933
-arkindex_worker/worker/__init__.py,sha256=0_YHeOe31KR_8ynbnYMIMwnSQTVbKkkeLGmnlTMhFx0,16234
-arkindex_worker/worker/base.py,sha256=7Pmw-UQSxV-xkW8NO5cXsxJ8W8szzyppMaNjq_az81A,19844
-arkindex_worker/worker/classification.py,sha256=zECSNzGCZFzoPoDVZN4kuGYRNLzMQLBaRt3q1jnBSaA,10952
-arkindex_worker/worker/corpus.py,sha256=0TQFOwZ6Te-CZi6lgkZY1wzyJ5wO9LAmcVQtqHvZpPk,2291
-arkindex_worker/worker/dataset.py,sha256=LwzKwNFX4FqfLxh29LSvJydPwRw3VHaB1wjuFhUshsE,5267
-arkindex_worker/worker/element.py,sha256=Qvvq9kJnAHNATHW7zi96eIY1x-0MsR-T5rrSJg6e9Y4,45309
-arkindex_worker/worker/entity.py,sha256=DG8oVAdy-r18fliTjnzGI1j6l7SOFmyIBmE6JlE6A8g,14799
-arkindex_worker/worker/image.py,sha256=t_Az6IGnj0EZyvcA4XxfPikOUjn_pztgsyxTkFZhaXU,621
-arkindex_worker/worker/metadata.py,sha256=VRajtd2kaBvar9GercX4knvR6l1WFYjoCdJWU9ccKgk,7291
-arkindex_worker/worker/process.py,sha256=9TEHpMcBax1wc6PrWMMrdXe2uNfqyVj7n_dAYZRBGnY,1854
-arkindex_worker/worker/task.py,sha256=r1j7_qbdNu2Z8H8HbGzO3P3qdx-2N1pBbUPFDca0rqg,1519
-arkindex_worker/worker/training.py,sha256=H8FmCdzGcDW-WMMwcgvmZPlN5tPHwGo0BXn12qmzj8g,10875
-arkindex_worker/worker/transcription.py,sha256=52RY9kYsiR1sz9FxOigyo12Ker3VDbQ4U42gK9DpR3g,21146
-arkindex_worker/worker/version.py,sha256=JIT7OI3Mo7RPkNrjOB9hfqrsG-FYygz_zi4l8PbkuAo,1960
-hooks/pre_gen_project.py,sha256=xQJERv3vv9VzIqcBHI281eeWLWREXUF4mMw7PvJHHXM,269
-tests/__init__.py,sha256=DG--S6IpGl399rzSAjDdHL76CkOIeZIjajCcyUSDhOQ,241
-tests/conftest.py,sha256=Z9amrKmVtFltzTUUm07fGDrT4m540biaTpjedmplyzc,21536
-tests/test_base_worker.py,sha256=2EIYcd_3f9O0zB5WiGIQV0Cn9wndLvnEnSfcAE1qWWU,30607
-tests/test_cache.py,sha256=ii0gyr0DrG7ChEs7pmT8hMdSguAOAcCze4bRMiFQxuk,10640
-tests/test_dataset_worker.py,sha256=z8ydliUlwW2j-irgLAotJMacgJXkVvF5TgsWLyCn1Jo,22087
-tests/test_element.py,sha256=2G9M15TLxQRmvrWM9Kw2ucnElh4kSv_oF_5FYwwAxTY,13181
-tests/test_image.py,sha256=nFxf7fnh8DyYF6ZWhvDAtN5rEk96zdJEm03vMiyz7pA,25495
-tests/test_merge.py,sha256=TuOeUS0UCz66DPOQFFhc4NQBxIjZL9f5czi4XnvGrr4,8270
-tests/test_utils.py,sha256=nYL1s2ViZoLoMiNpLGDaWwxf8dJ1D8aT522AO-PVaEQ,3607
-tests/test_elements_worker/__init__.py,sha256=Fh4nkbbyJSMv_VtjQxnWrOqTnxXaaWI8S9WU0VrzCHs,179
-tests/test_elements_worker/test_classification.py,sha256=nya7veSPR_O9G41Enodp2-o6AifMBcaSTWJP2vXSSJ4,30133
-tests/test_elements_worker/test_cli.py,sha256=a23i1pUDbXi23MUtbWwGEcLLrmc_YlrbDgOG3h66wLM,2620
-tests/test_elements_worker/test_corpus.py,sha256=kscJyM8k1njYJJFGuvliVzn89lWh41mEyDCCawnp3W8,5483
-tests/test_elements_worker/test_dataset.py,sha256=00IlOZv9YFlZ23rGXyR-HLbKLQxGelZ1Bf9lEZYA0IY,11412
-tests/test_elements_worker/test_element.py,sha256=lb5tLjl0jsixX0OWVhBAaKLE9GKkBw79kFHDNGommaQ,12535
-tests/test_elements_worker/test_element_create_multiple.py,sha256=arYFGmxc0517ZUii6k__G_UQQatuNIASTC8MXvUrSwk,21887
-tests/test_elements_worker/test_element_create_single.py,sha256=Fa9zm12J2rQ3VrUe3yIlHAc7Vty_eQYb_YGnNPQB3IE,16697
-tests/test_elements_worker/test_element_list_children.py,sha256=2zH4h663w3EduqpzQr-7bf9zIDzO1x2WxdUYYHsIHkI,31358
-tests/test_elements_worker/test_element_list_parents.py,sha256=TXeGW-a3W-7GmB2QrhJH9mMnvxuybeAwQ4tL3iIxwXo,16734
-tests/test_elements_worker/test_entity_create.py,sha256=9Tjr9KA2yo44VFV283q_cs6XbbVguUMDNfCj-DILSJg,29353
-tests/test_elements_worker/test_entity_list_and_check.py,sha256=ENBLaqbXlRUDbHRvQla3080a0HJltrWAPYWNohUA9NU,4992
-tests/test_elements_worker/test_image.py,sha256=BljMNKgec_9a5bzNzFpYZIvSbuvwsWDfdqLHVJaTa7M,2079
-tests/test_elements_worker/test_metadata.py,sha256=Xfggy-vxw5DZ3hFKx3sB7OYb2d1tu1RiNK8fvKJIaBs,22294
-tests/test_elements_worker/test_process.py,sha256=y4RoVhPfyHzR795fw7-_FXElBcKo3fy4Ew_HI-kxJic,3088
-tests/test_elements_worker/test_task.py,sha256=wTUWqN9UhfKmJn3IcFY75EW4I1ulRhisflmY1kmP47s,5574
-tests/test_elements_worker/test_training.py,sha256=3W2LzpqxekvRiX42m_PvWcVel7ynQJmzO8gKcLmCMQI,8717
-tests/test_elements_worker/test_transcription_create.py,sha256=yznO9B_BVsOR0Z_VY5ZL8gJp0ZPCz_4sPUs5dXtixAg,29281
-tests/test_elements_worker/test_transcription_create_with_elements.py,sha256=tmcyglgssEqMnt1Mdy_u6X1m2wgLWTo_HdWst3GrK2k,33056
-tests/test_elements_worker/test_transcription_list.py,sha256=ikz7HYPCoQWTdTRCd382SB-y-T2BbigPLlIcx5Eow-I,15324
-tests/test_elements_worker/test_version.py,sha256=xqCgcgukTFJzkMgYfQG-8mTbu0o2fdYjWC07FktThfw,2125
-tests/test_elements_worker/test_worker.py,sha256=pLUgjyrrXrzVD6T-kdH1ppk5Yn_iDuI8JdFGweTEMXE,25156
-worker-demo/tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-worker-demo/tests/conftest.py,sha256=XzNMNeg6pmABUAH8jN6eZTlZSFGLYjS3-DTXjiRN6Yc,1002
-worker-demo/tests/test_worker.py,sha256=3DLd4NRK4bfyatG5P_PK4k9P9tJHx9XQq5_ryFEEFVg,304
-worker-demo/worker_demo/__init__.py,sha256=2BPomV8ZMNf3YXJgloatKeHQCE6QOkwmsHGkO6MkQuM,125
-worker-demo/worker_demo/worker.py,sha256=Rt-DjWa5iBP08k58NDZMfeyPuFbtNcbX6nc5jFX7GNo,440
-arkindex_base_worker-0.4.0.dist-info/LICENSE,sha256=NVshRi1efwVezMfW7xXYLrdDr2Li1AfwfGOd5WuH1kQ,1063
-arkindex_base_worker-0.4.0.dist-info/METADATA,sha256=OjqRHv4B7-BiEwyDeDgWt6HhBWJ7AkpBe6qCCHnG_W0,3334
-arkindex_base_worker-0.4.0.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
-arkindex_base_worker-0.4.0.dist-info/top_level.txt,sha256=58NuslgxQC2vT4DiqZEgO4JqJRrYa2yeNI9QvkbfGQU,40
-arkindex_base_worker-0.4.0.dist-info/RECORD,,

arkindex_worker/worker/process.py

@@ -1,92 +0,0 @@
-from collections.abc import Iterator
-from enum import Enum
-
-from arkindex_worker.cache import unsupported_cache
-
-# Increases the number of elements returned per page by the API
-PROCESS_ELEMENTS_PAGE_SIZE = 500
-
-
-class ActivityState(Enum):
-    """
-    Processing state of an element.
-    """
-
-    Queued = "queued"
-    """
-    The element has not yet been processed by a worker.
-    """
-
-    Started = "started"
-    """
-    The element is being processed by a worker.
-    """
-
-    Processed = "processed"
-    """
-    The element has been successfully processed by a worker.
-    """
-
-    Error = "error"
-    """
-    An error occurred while processing this element.
-    """
-
-
-class ProcessMode(Enum):
-    """
-    Mode of the process of the worker.
-    """
-
-    Files = "files"
-    """
-    Processes of files (images, PDFs, IIIF, ...) imports.
-    """
-
-    Workers = "workers"
-    """
-    Processes of worker executions.
-    """
-
-    Template = "template"
-    """
-    Process templates.
-    """
-
-    S3 = "s3"
-    """
-    Processes of imports from an S3-compatible storage.
-    """
-
-    Local = "local"
-    """
-    Local processes.
-    """
-
-    Dataset = "dataset"
-    """
-    Dataset processes.
-    """
-
-    Export = "export"
-    """
-    Export processes.
-    """
-
-
-class ProcessMixin:
-    @unsupported_cache
-    def list_process_elements(self, with_image: bool = False) -> Iterator[dict]:
-        """
-        List the elements of a process.
-
-        :param with_image: whether or not to include zone and image information in the elements response.
-        :returns: the process' elements.
-        """
-        return self.api_client.paginate(
-            "ListProcessElements",
-            id=self.process_information["id"],
-            with_image=with_image,
-            allow_missing_data=True,
-            page_size=PROCESS_ELEMENTS_PAGE_SIZE,
-        )

tests/test_elements_worker/test_element.py

@@ -1,427 +0,0 @@
-import json
-import re
-
-import pytest
-from responses import matchers
-
-from arkindex.exceptions import ErrorResponse
-from arkindex_worker.cache import (
-    CachedElement,
-    CachedImage,
-)
-from arkindex_worker.models import Element
-from arkindex_worker.worker.element import MissingTypeError
-from tests import CORPUS_ID
-
-from . import BASE_API_CALLS
-
-
-def test_list_corpus_types(responses, mock_elements_worker):
-    responses.add(
-        responses.GET,
-        f"http://testserver/api/v1/corpus/{CORPUS_ID}/",
-        json={
-            "id": CORPUS_ID,
-            "types": [{"slug": "folder"}, {"slug": "page"}],
-        },
-    )
-
-    mock_elements_worker.list_corpus_types()
-
-    assert mock_elements_worker.corpus_types == {
-        "folder": {"slug": "folder"},
-        "page": {"slug": "page"},
-    }
-
-
-def test_check_required_types_argument_types(mock_elements_worker):
-    with pytest.raises(
-        AssertionError, match="At least one element type slug is required."
-    ):
-        mock_elements_worker.check_required_types()
-
-    with pytest.raises(AssertionError, match="Element type slugs must be strings."):
-        mock_elements_worker.check_required_types("lol", 42)
-
-
-def test_check_required_types(mock_elements_worker):
-    mock_elements_worker.corpus_types = {
-        "folder": {"slug": "folder"},
-        "page": {"slug": "page"},
-    }
-
-    assert mock_elements_worker.check_required_types("page")
-    assert mock_elements_worker.check_required_types("page", "folder")
-
-    with pytest.raises(
-        MissingTypeError,
-        match=re.escape(
-            "Element types act, text_line were not found in corpus (11111111-1111-1111-1111-111111111111)."
-        ),
-    ):
-        assert mock_elements_worker.check_required_types("page", "text_line", "act")
-
-
-def test_check_required_types_create_missing(responses, mock_elements_worker):
-    mock_elements_worker.corpus_types = {
-        "folder": {"slug": "folder"},
-        "page": {"slug": "page"},
-    }
-
-    responses.add(
-        responses.POST,
-        "http://testserver/api/v1/elements/type/",
-        match=[
-            matchers.json_params_matcher(
-                {
-                    "slug": "text_line",
-                    "display_name": "text_line",
-                    "folder": False,
-                    "corpus": CORPUS_ID,
-                }
-            )
-        ],
-    )
-    responses.add(
-        responses.POST,
-        "http://testserver/api/v1/elements/type/",
-        match=[
-            matchers.json_params_matcher(
-                {
-                    "slug": "act",
-                    "display_name": "act",
-                    "folder": False,
-                    "corpus": CORPUS_ID,
-                }
-            )
-        ],
-    )
-
-    assert mock_elements_worker.check_required_types(
-        "page", "text_line", "act", create_missing=True
-    )
-
-
-@pytest.mark.parametrize(
-    ("payload", "error"),
-    [
-        # Element
-        (
-            {"element": None},
-            "element shouldn't be null and should be an Element or CachedElement",
-        ),
-        (
-            {"element": "not element type"},
-            "element shouldn't be null and should be an Element or CachedElement",
-        ),
-    ],
-)
-def test_partial_update_element_wrong_param_element(
-    mock_elements_worker, payload, error
-):
-    api_payload = {
-        "element": Element({"zone": None}),
-        **payload,
-    }
-
-    with pytest.raises(AssertionError, match=error):
-        mock_elements_worker.partial_update_element(
-            **api_payload,
-        )
-
-
-@pytest.mark.parametrize(
-    ("payload", "error"),
-    [
-        # Type
-        ({"type": 1234}, "type should be a str"),
-        ({"type": None}, "type should be a str"),
-    ],
-)
-def test_partial_update_element_wrong_param_type(mock_elements_worker, payload, error):
-    api_payload = {
-        "element": Element({"zone": None}),
-        **payload,
-    }
-
-    with pytest.raises(AssertionError, match=error):
-        mock_elements_worker.partial_update_element(
-            **api_payload,
-        )
-
-
-@pytest.mark.parametrize(
-    ("payload", "error"),
-    [
-        # Name
-        ({"name": 1234}, "name should be a str"),
-        ({"name": None}, "name should be a str"),
-    ],
-)
-def test_partial_update_element_wrong_param_name(mock_elements_worker, payload, error):
-    api_payload = {
-        "element": Element({"zone": None}),
-        **payload,
-    }
-
-    with pytest.raises(AssertionError, match=error):
-        mock_elements_worker.partial_update_element(
-            **api_payload,
-        )
-
-
-@pytest.mark.parametrize(
-    ("payload", "error"),
-    [
-        # Polygon
-        ({"polygon": "not a polygon"}, "polygon should be a list"),
-        ({"polygon": None}, "polygon should be a list"),
-        ({"polygon": [[1, 1], [2, 2]]}, "polygon should have at least three points"),
-        (
-            {"polygon": [[1, 1, 1], [2, 2, 1], [2, 1, 1], [1, 2, 1]]},
-            "polygon points should be lists of two items",
-        ),
-        (
-            {"polygon": [[1], [2], [2], [1]]},
-            "polygon points should be lists of two items",
-        ),
-        (
-            {"polygon": [["not a coord", 1], [2, 2], [2, 1], [1, 2]]},
-            "polygon points should be lists of two numbers",
-        ),
-    ],
-)
-def test_partial_update_element_wrong_param_polygon(
-    mock_elements_worker, payload, error
-):
-    api_payload = {
-        "element": Element({"zone": None}),
-        **payload,
-    }
-
-    with pytest.raises(AssertionError, match=error):
-        mock_elements_worker.partial_update_element(
-            **api_payload,
-        )
-
-
-@pytest.mark.parametrize(
-    ("payload", "error"),
-    [
-        # Confidence
-        ({"confidence": "lol"}, "confidence should be None or a float in [0..1] range"),
-        ({"confidence": "0.2"}, "confidence should be None or a float in [0..1] range"),
-        ({"confidence": -1.0}, "confidence should be None or a float in [0..1] range"),
-        ({"confidence": 1.42}, "confidence should be None or a float in [0..1] range"),
-        (
-            {"confidence": float("inf")},
-            "confidence should be None or a float in [0..1] range",
-        ),
-    ],
-)
-def test_partial_update_element_wrong_param_conf(mock_elements_worker, payload, error):
-    api_payload = {
-        "element": Element({"zone": None}),
-        **payload,
-    }
-
-    with pytest.raises(AssertionError, match=re.escape(error)):
-        mock_elements_worker.partial_update_element(
-            **api_payload,
-        )
-
-
-@pytest.mark.parametrize(
-    ("payload", "error"),
-    [
-        # Rotation angle
-        ({"rotation_angle": "lol"}, "rotation_angle should be a positive integer"),
-        ({"rotation_angle": -1}, "rotation_angle should be a positive integer"),
-        ({"rotation_angle": 0.5}, "rotation_angle should be a positive integer"),
-        ({"rotation_angle": None}, "rotation_angle should be a positive integer"),
-    ],
-)
-def test_partial_update_element_wrong_param_rota(mock_elements_worker, payload, error):
-    api_payload = {
-        "element": Element({"zone": None}),
-        **payload,
-    }
-
-    with pytest.raises(AssertionError, match=error):
-        mock_elements_worker.partial_update_element(
-            **api_payload,
-        )
-
-
-@pytest.mark.parametrize(
-    ("payload", "error"),
-    [
-        # Mirrored
-        ({"mirrored": "lol"}, "mirrored should be a boolean"),
-        ({"mirrored": 1234}, "mirrored should be a boolean"),
-        ({"mirrored": None}, "mirrored should be a boolean"),
-    ],
-)
-def test_partial_update_element_wrong_param_mir(mock_elements_worker, payload, error):
-    api_payload = {
-        "element": Element({"zone": None}),
-        **payload,
-    }
-
-    with pytest.raises(AssertionError, match=error):
-        mock_elements_worker.partial_update_element(
-            **api_payload,
-        )
-
-
-@pytest.mark.parametrize(
-    ("payload", "error"),
-    [
-        # Image
-        ({"image": "lol"}, "image should be a UUID"),
-        ({"image": 1234}, "image should be a UUID"),
-        ({"image": None}, "image should be a UUID"),
-    ],
-)
-def test_partial_update_element_wrong_param_image(mock_elements_worker, payload, error):
-    api_payload = {
-        "element": Element({"zone": None}),
-        **payload,
-    }
-
-    with pytest.raises(AssertionError, match=error):
-        mock_elements_worker.partial_update_element(
-            **api_payload,
-        )
-
-
-def test_partial_update_element_api_error(responses, mock_elements_worker):
-    elt = Element({"id": "12341234-1234-1234-1234-123412341234"})
-    responses.add(
-        responses.PATCH,
-        f"http://testserver/api/v1/element/{elt.id}/",
-        status=418,
-    )
-
-    with pytest.raises(ErrorResponse):
-        mock_elements_worker.partial_update_element(
-            element=elt,
-            type="something",
-            name="0",
-            polygon=[[1, 1], [2, 2], [2, 1], [1, 2]],
-        )
-
-    assert len(responses.calls) == len(BASE_API_CALLS) + 1
-    assert [
-        (call.request.method, call.request.url) for call in responses.calls
-    ] == BASE_API_CALLS + [("PATCH", f"http://testserver/api/v1/element/{elt.id}/")]
-
-
-@pytest.mark.usefixtures("_mock_cached_elements", "_mock_cached_images")
-@pytest.mark.parametrize(
-    "payload",
-    [
-        (
-            {
-                "polygon": [[10, 10], [20, 20], [20, 10], [10, 20]],
-                "confidence": None,
-            }
-        ),
-        (
-            {
-                "rotation_angle": 45,
-                "mirrored": False,
-            }
-        ),
-        (
-            {
-                "polygon": [[10, 10], [20, 20], [20, 10], [10, 20]],
-                "confidence": None,
-                "rotation_angle": 45,
-                "mirrored": False,
-            }
-        ),
-    ],
-)
-def test_partial_update_element(responses, mock_elements_worker_with_cache, payload):
-    elt = CachedElement.select().first()
-    new_image = CachedImage.select().first()
-
-    elt_response = {
-        "image": str(new_image.id),
-        **payload,
-    }
-    responses.add(
-        responses.PATCH,
-        f"http://testserver/api/v1/element/{elt.id}/",
-        status=200,
-        # UUID not allowed in JSON
-        json=elt_response,
-    )
-
-    element_update_response = mock_elements_worker_with_cache.partial_update_element(
-        element=elt,
-        **{**elt_response, "image": new_image.id},
-    )
-
-    assert len(responses.calls) == len(BASE_API_CALLS) + 1
-    assert [
-        (call.request.method, call.request.url) for call in responses.calls
-    ] == BASE_API_CALLS + [
-        (
-            "PATCH",
-            f"http://testserver/api/v1/element/{elt.id}/",
-        ),
-    ]
-    assert json.loads(responses.calls[-1].request.body) == elt_response
-    assert element_update_response == elt_response
-
-    cached_element = CachedElement.get(CachedElement.id == elt.id)
-    # Always present in payload
-    assert str(cached_element.image_id) == elt_response["image"]
-    # Optional params
-    if "polygon" in payload:
-        # Cast to string as this is the only difference compared to model
-        elt_response["polygon"] = str(elt_response["polygon"])
-
-    for param in payload:
-        assert getattr(cached_element, param) == elt_response[param]
-
-
-@pytest.mark.usefixtures("_mock_cached_elements")
-@pytest.mark.parametrize("confidence", [None, 0.42])
-def test_partial_update_element_confidence(
-    responses, mock_elements_worker_with_cache, confidence
-):
-    elt = CachedElement.select().first()
-    elt_response = {
-        "polygon": [[10, 10], [20, 20], [20, 10], [10, 20]],
-        "confidence": confidence,
-    }
-    responses.add(
-        responses.PATCH,
-        f"http://testserver/api/v1/element/{elt.id}/",
-        status=200,
-        json=elt_response,
-    )
-
-    element_update_response = mock_elements_worker_with_cache.partial_update_element(
-        element=elt,
-        **elt_response,
-    )
-
-    assert len(responses.calls) == len(BASE_API_CALLS) + 1
-    assert [
-        (call.request.method, call.request.url) for call in responses.calls
-    ] == BASE_API_CALLS + [
-        (
-            "PATCH",
-            f"http://testserver/api/v1/element/{elt.id}/",
-        ),
-    ]
-    assert json.loads(responses.calls[-1].request.body) == elt_response
-    assert element_update_response == elt_response
-
-    cached_element = CachedElement.get(CachedElement.id == elt.id)
-    assert cached_element.polygon == str(elt_response["polygon"])
-    assert cached_element.confidence == confidence