import logging
from functools import partial
from itertools import chain, repeat
from operator import methodcaller
from typing import Iterable

import pytest
import requests
from funcy import rcompose, compose, rpartial, identity, lmap, ilen, first

from pyinfra.rest import pack, unpack, bundle
from pyinfra.utils.func import lift, starlift, parallel_map, star, lstarlift

logger = logging.getLogger("PIL.PngImagePlugin")
logger.setLevel(logging.WARNING)


def dispatch_methods(input_data):
    return *repeat(requests.patch, ilen(input_data) - 1), requests.post


def post_partial(url, input_data: Iterable[bytes], metadata):
    def send(method, data):
        return method(url, json=data)

    pack_data_and_metadata_for_rest_transfer = lift(rpartial(pack, metadata))
    dispatch_http_method_left_and_forward_data_right = parallel_map(dispatch_methods, lift(identity))
    send_data_with_method_to_analyzer = starlift(send)
    extract_payload_from_responses = lift(methodcaller("json"))
    flatten_buffered_payloads = chain.from_iterable
    interpret_payloads = lift(compose(star(bundle), unpack))

    input_data_to_result_data = rcompose(
        pack_data_and_metadata_for_rest_transfer,
        dispatch_http_method_left_and_forward_data_right,
        send_data_with_method_to_analyzer,
        extract_payload_from_responses,
        flatten_buffered_payloads,
        interpret_payloads,
    )

    return input_data_to_result_data(input_data)


@pytest.mark.parametrize("item_type", ["string", "image"])
def test_sending_partial_request(url, data_items, metadata, operation, server_process):
    expected = lmap(compose(first, lstarlift(bundle), partial(operation, metadata=metadata)), data_items)
    output = list(post_partial(f"{url}/process", data_items, metadata))

    assert output == expected