From fd5956bde4277ae2b5fff575c4413eabd5e937c7 Mon Sep 17 00:00:00 2001
From: Yoann Schneider <yschneider@teklia.com>
Date: Tue, 23 Aug 2022 17:25:15 +0200
Subject: [PATCH] provide slim output as kwargs
---
arkindex_worker/worker/element.py | 1 +
tests/test_elements_worker/test_elements.py | 23 ++++++++++++---------
2 files changed, 14 insertions(+), 10 deletions(-)
diff --git a/arkindex_worker/worker/element.py b/arkindex_worker/worker/element.py
index df225282..a22dbd0e 100644
--- a/arkindex_worker/worker/element.py
+++ b/arkindex_worker/worker/element.py
@@ -104,6 +104,7 @@ 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 e3ad26f7..7ff16855 100644
--- a/tests/test_elements_worker/test_elements.py
+++ b/tests/test_elements_worker/test_elements.py
@@ -447,7 +447,7 @@ def test_create_sub_element_api_error(responses, mock_elements_worker):
)
responses.add(
responses.POST,
- "http://testserver/api/v1/elements/create/",
+ "http://testserver/api/v1/elements/create/?slim_output=True",
status=500,
)
@@ -464,11 +464,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/"),
- ("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/?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"),
]
@@ -488,7 +488,7 @@ def test_create_sub_element(responses, mock_elements_worker, slim_output):
}
responses.add(
responses.POST,
- "http://testserver/api/v1/elements/create/",
+ f"http://testserver/api/v1/elements/create/?slim_output={slim_output}",
status=200,
json=child_elt,
)
@@ -505,7 +505,10 @@ def test_create_sub_element(responses, mock_elements_worker, slim_output):
assert [
(call.request.method, call.request.url) for call in responses.calls
] == BASE_API_CALLS + [
- ("POST", "http://testserver/api/v1/elements/create/"),
+ (
+ "POST",
+ f"http://testserver/api/v1/elements/create/?slim_output={slim_output}",
+ ),
]
assert json.loads(responses.calls[-1].request.body) == {
"type": "something",
@@ -533,7 +536,7 @@ def test_create_sub_element_confidence(responses, mock_elements_worker):
)
responses.add(
responses.POST,
- "http://testserver/api/v1/elements/create/",
+ "http://testserver/api/v1/elements/create/?slim_output=True",
status=200,
json={"id": "12345678-1234-1234-1234-123456789123"},
)
@@ -550,7 +553,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/"),
+ ("POST", "http://testserver/api/v1/elements/create/?slim_output=True"),
]
assert json.loads(responses.calls[-1].request.body) == {
"type": "something",
--
GitLab