pyinfra/tests/processing_test.py

import gzip
import json
from operator import itemgetter

import pytest

from pyinfra.payload_processing.processor import make_payload_processor


@pytest.fixture(scope="session")
def file_processor_mock():
    def inner(json_file: dict):
        return [json_file]

    return inner


@pytest.fixture
def target_file():
    contents = {"numberOfPages": 10, "content1": "value1", "content2": "value2"}
    return gzip.compress(json.dumps(contents).encode("utf-8"))


@pytest.fixture
def file_names(request_payload):
    dossier_id, file_id, target_suffix, response_suffix = itemgetter(
        "dossierId",
        "fileId",
        "targetFileExtension",
        "responseFileExtension",
    )(request_payload)
    return f"{dossier_id}/{file_id}.{target_suffix}", f"{dossier_id}/{file_id}.{response_suffix}"


@pytest.fixture(scope="session")
def payload_processor(file_processor_mock, processing_config):
    yield make_payload_processor(file_processor_mock, processing_config)


@pytest.mark.parametrize("client_name", ["s3"], scope="session")
@pytest.mark.parametrize("monitoring_enabled", [True, False], scope="session")
class TestPayloadProcessor:
    def test_payload_processor_yields_correct_response_and_uploads_result(
        self,
        payload_processor,
        storage,
        bucket_name,
        request_payload,
        response_payload,
        target_file,
        file_names,
    ):
        storage.clear_bucket(bucket_name)
        storage.put_object(bucket_name, file_names[0], target_file)
        response = payload_processor(request_payload)

        assert response == response_payload

        data_received = storage.get_object(bucket_name, file_names[1])

        assert json.loads((gzip.decompress(data_received)).decode("utf-8")) == {
            **request_payload,
            "data": [json.loads(gzip.decompress(target_file).decode("utf-8"))],
        }

    def test_catching_of_processing_failure(self, payload_processor, storage, bucket_name, request_payload):
        storage.clear_bucket(bucket_name)
        with pytest.raises(Exception):
            payload_processor(request_payload)