From 21108bb15370f1a566c7009b9f97ef5535388b78 Mon Sep 17 00:00:00 2001
From: Yoann Schneider <yschneider@teklia.com>
Date: Mon, 6 Feb 2023 06:37:13 +0000
Subject: [PATCH] Blast slim output usage in API requests

---
 arkindex_worker/worker/element.py           |  1 -
 tests/test_elements_worker/test_elements.py | 20 ++++++++++----------
 2 files changed, 10 insertions(+), 11 deletions(-)

diff --git a/arkindex_worker/worker/element.py b/arkindex_worker/worker/element.py
index adba4b3a..68aeeb56 100644
--- a/arkindex_worker/worker/element.py
+++ b/arkindex_worker/worker/element.py
@@ -130,7 +130,6 @@ class ElementMixin(object):
 
         sub_element = self.request(
             "CreateElement",
-            slim_output=slim_output,
             body={
                 "type": type,
                 "name": name,
diff --git a/tests/test_elements_worker/test_elements.py b/tests/test_elements_worker/test_elements.py
index bceef1cb..502a20cb 100644
--- a/tests/test_elements_worker/test_elements.py
+++ b/tests/test_elements_worker/test_elements.py
@@ -503,7 +503,7 @@ def test_create_sub_element_api_error(responses, mock_elements_worker):
     )
     responses.add(
         responses.POST,
-        "http://testserver/api/v1/elements/create/?slim_output=True",
+        "http://testserver/api/v1/elements/create/",
         status=500,
     )
 
@@ -520,11 +520,11 @@ def test_create_sub_element_api_error(responses, mock_elements_worker):
         (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/?slim_output=True"),
-        ("POST", "http://testserver/api/v1/elements/create/?slim_output=True"),
-        ("POST", "http://testserver/api/v1/elements/create/?slim_output=True"),
-        ("POST", "http://testserver/api/v1/elements/create/?slim_output=True"),
-        ("POST", "http://testserver/api/v1/elements/create/?slim_output=True"),
+        ("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/"),
     ]
 
 
@@ -544,7 +544,7 @@ def test_create_sub_element(responses, mock_elements_worker, slim_output):
     }
     responses.add(
         responses.POST,
-        f"http://testserver/api/v1/elements/create/?slim_output={slim_output}",
+        "http://testserver/api/v1/elements/create/",
         status=200,
         json=child_elt,
     )
@@ -563,7 +563,7 @@ def test_create_sub_element(responses, mock_elements_worker, slim_output):
     ] == BASE_API_CALLS + [
         (
             "POST",
-            f"http://testserver/api/v1/elements/create/?slim_output={slim_output}",
+            "http://testserver/api/v1/elements/create/",
         ),
     ]
     assert json.loads(responses.calls[-1].request.body) == {
@@ -592,7 +592,7 @@ def test_create_sub_element_confidence(responses, mock_elements_worker):
     )
     responses.add(
         responses.POST,
-        "http://testserver/api/v1/elements/create/?slim_output=True",
+        "http://testserver/api/v1/elements/create/",
         status=200,
         json={"id": "12345678-1234-1234-1234-123456789123"},
     )
@@ -609,7 +609,7 @@ def test_create_sub_element_confidence(responses, mock_elements_worker):
     assert [
         (call.request.method, call.request.url) for call in responses.calls
     ] == BASE_API_CALLS + [
-        ("POST", "http://testserver/api/v1/elements/create/?slim_output=True"),
+        ("POST", "http://testserver/api/v1/elements/create/"),
     ]
     assert json.loads(responses.calls[-1].request.body) == {
         "type": "something",
-- 
GitLab