From 656bc7cd637a2bd1b42804908d56b417f3ec45bf Mon Sep 17 00:00:00 2001
From: Julius Unverfehrt <julius.unverfehrt@iqser.com>
Date: Tue, 26 Apr 2022 16:34:58 +0200
Subject: [PATCH] explore partial responses

---
 test/exploration_tests/partial_response_test.py | 8 ++------
 test/fixtures/input.py                          | 2 +-
 test/fixtures/server.py                         | 4 ++--
 3 files changed, 5 insertions(+), 9 deletions(-)

diff --git a/test/exploration_tests/partial_response_test.py b/test/exploration_tests/partial_response_test.py
index 6b84f4e..9f51f95 100644
--- a/test/exploration_tests/partial_response_test.py
+++ b/test/exploration_tests/partial_response_test.py
@@ -11,18 +11,14 @@ from test.utils.server import bytes_to_string, string_to_bytes
 def test_sending_partial_request(url, data_items, metadata):
     def pack(metadata: dict, data: bytes):
         package = {"data": bytes_to_string(data), "metadata": metadata}
-        package = string_to_bytes(json.dumps(package))
+        package = json.dumps(package).encode()
         return package
 
     packages = starmap(pack, zip(repeat(metadata), data_items))
 
-    def f():
-        yield "a"
-        yield "b"
-
     requests.post(f"{url}/process", data=packages, stream=True)
 
 
 @pytest.fixture
 def metadata():
-    return {"idx": [0, 3, 9]}
+    return {"idx": [1, 100, 101], "path": "asd/asd"}
diff --git a/test/fixtures/input.py b/test/fixtures/input.py
index 3d7a59f..cadc5d6 100644
--- a/test/fixtures/input.py
+++ b/test/fixtures/input.py
@@ -12,4 +12,4 @@ def data(data_type, pdf):
 @pytest.fixture
 def data_items(item_type):
     if item_type == "string":
-        return iter([b"content"] * 5)
+        return iter([b"content"] * 4)
diff --git a/test/fixtures/server.py b/test/fixtures/server.py
index 89d87d9..b33d8e6 100644
--- a/test/fixtures/server.py
+++ b/test/fixtures/server.py
@@ -7,6 +7,7 @@ from funcy import retry
 from waitress import serve
 
 from test.server import set_up_processing_server
+from test.utils.server import bytes_to_string
 
 
 @pytest.fixture
@@ -43,8 +44,7 @@ def processor_fn(item_type, data_items):
 
 def make_string_processor(data_items):
     def processor_fn(payload):
-        print(111111111111111111111111111111111111)
-        print(payload)
+        print(payload.stream.read())
         # def parse(package)
         # payload = json.load(payload)
         # data = string_to_bytes(payload["data"])