27 lines
950 B
Python
27 lines
950 B
Python
import pytest
|
|
from funcy import compose, lzip
|
|
|
|
from pyinfra.server.packer.packers.identity import bundle
|
|
from pyinfra.server.utils import pack, unpack
|
|
from pyinfra.utils.func import lstarlift
|
|
from test.utils.server import bytes_to_string
|
|
|
|
|
|
@pytest.mark.parametrize("n_items", [0, 2])
|
|
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)
|
|
]
|
|
|
|
|
|
@pytest.mark.parametrize("n_items", [0, 2])
|
|
def test_pack_unpack(input_data_items, metadata):
|
|
assert lstarlift(compose(unpack, pack))(zip(input_data_items, metadata)) == lzip(input_data_items, metadata)
|
|
|
|
|
|
@pytest.mark.parametrize("n_items", [0, 2])
|
|
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)
|
|
]
|