From 8afd87e44f7c89efe2f12a10879f42115a64e3d8 Mon Sep 17 00:00:00 2001
From: Matthias Bisping <matthias.bisping@iqser.com>
Date: Wed, 4 May 2022 15:18:15 +0200
Subject: [PATCH] added packing and bundling test

---
 test/unit_tests/rest/utils.py | 22 ++++++++++++++++++++++
 1 file changed, 22 insertions(+)
 create mode 100644 test/unit_tests/rest/utils.py

diff --git a/test/unit_tests/rest/utils.py b/test/unit_tests/rest/utils.py
new file mode 100644
index 0000000..79aeadf
--- /dev/null
+++ b/test/unit_tests/rest/utils.py
@@ -0,0 +1,22 @@
+from funcy import compose, lzip
+
+from pyinfra.server.packer.packers.identity import bundle
+from pyinfra.server.rest import unpack, pack
+from pyinfra.utils.func import lstarlift
+from test.utils.server import bytes_to_string
+
+
+def test_pack(input_data_items, metadata):
+    assert lstarlift(pack)(zip(input_data_items, metadata)) == [
+        {"data": bytes_to_string(d), "metadata": md} for d, md in zip(input_data_items, metadata)
+    ]
+
+
+def test_pack_unpack(input_data_items, metadata):
+    assert lstarlift(compose(unpack, pack))(zip(input_data_items, metadata)) == lzip(input_data_items, metadata)
+
+
+def test_bundle(input_data_items, metadata):
+    assert lstarlift(bundle)(zip(input_data_items, metadata)) == [
+        {"data": d, "metadata": md} for d, md in zip(input_data_items, metadata)
+    ]