Port init elements code

Closes #2 (closed)

tests/conftest.py

@@ -4,6 +4,7 @@ import pytest
 
 from arkindex.mock import MockApiClient
 from arkindex_worker.worker.base import BaseWorker
+from worker_init_elements.worker import InitElementWorker
 
 
 @pytest.fixture(autouse=True)
@@ -30,3 +31,22 @@ def _setup_environment(responses, monkeypatch) -> None:
         self.api_client = MockApiClient()
 
     monkeypatch.setattr(BaseWorker, "setup_api_client", mock_setup_api_client)
+
+
+@pytest.fixture()
+def mock_worker(tmp_path_factory) -> InitElementWorker:
+    worker = InitElementWorker()
+
+    # Default parameters
+    worker.chunks_number = 1
+    worker.use_cache = False
+
+    worker.work_dir = tmp_path_factory.mktemp("data")
+
+    worker.worker_run_id = os.environ["ARKINDEX_WORKER_RUN_ID"]
+    worker.process = {
+        "id": "process_id",
+        "corpus": os.environ["ARKINDEX_CORPUS_ID"],
+    }
+
+    return worker