diff --git a/test/unit_tests/rest/packer_test.py b/test/unit_tests/rest/packer_test.py new file mode 100644 index 0000000..8c4705f --- /dev/null +++ b/test/unit_tests/rest/packer_test.py @@ -0,0 +1,22 @@ +import pytest + +from pyinfra.server.packer.packers.identity import IdentityPacker, bundle +from pyinfra.server.packer.packers.rest import RestPacker +from pyinfra.server.rest import pack +from pyinfra.utils.func import lstarlift + + +def test_identity_packer(data, metadata): + packer = IdentityPacker() + assert list(packer(data, metadata)) == lstarlift(bundle)(zip(data, metadata)) + + +def test_rest_packer(data, metadata): + packer = RestPacker() + print(metadata) + assert list(packer(data, metadata)) == lstarlift(pack)(zip(data, metadata)) + + +@pytest.fixture +def data(input_data_items): + return input_data_items