From 26f4bd8f155299d16dd6b649d9f01dbc96e541ec Mon Sep 17 00:00:00 2001
From: Valentin Rigal <rigal@teklia.com>
Date: Mon, 26 Jul 2021 18:09:01 +0200
Subject: [PATCH] Simplify behavior to avoid confusion

---
 arkindex_worker/worker/__init__.py        | 2 +-
 tests/test_elements_worker/test_worker.py | 6 +-----
 2 files changed, 2 insertions(+), 6 deletions(-)

diff --git a/arkindex_worker/worker/__init__.py b/arkindex_worker/worker/__init__.py
index d4f709a8..2dcb7bff 100644
--- a/arkindex_worker/worker/__init__.py
+++ b/arkindex_worker/worker/__init__.py
@@ -225,4 +225,4 @@ class ElementsWorker(
             raise e
 
         logger.debug(f"Updated activity of element {element_id} to {state}")
-        return out
+        return True
diff --git a/tests/test_elements_worker/test_worker.py b/tests/test_elements_worker/test_worker.py
index e3c42036..962596e0 100644
--- a/tests/test_elements_worker/test_worker.py
+++ b/tests/test_elements_worker/test_worker.py
@@ -131,11 +131,7 @@ def test_update_call(responses, mock_elements_worker, mock_process_api):
     out = mock_elements_worker.update_activity("1234-deadbeef", ActivityState.Processed)
 
     # Check the response received by worker
-    assert out == {
-        "element_id": "1234-deadbeef",
-        "process_id": "aaaaaaaa-bbbb-cccc-dddd-eeeeeeeeffff",
-        "state": "processed",
-    }
+    assert out == True
 
     assert len(responses.calls) == len(BASE_API_CALLS) + 1
     assert [
-- 
GitLab