conftest.py

from typing import Any, Dict

import pytest
from pydantic import BaseModel

from aws_lambda_powertools.utilities.parser import BaseEnvelope


@pytest.fixture
def pydanticv2_only():
    from pydantic import __version__

    version = __version__.split(".")
    if version[0] != "2":
        pytest.skip("pydanticv2 test only")


@pytest.fixture
def dummy_event():
    return {"payload": {"message": "hello world"}}


@pytest.fixture
def dummy_schema():
    """Wanted payload structure"""

    class MyDummyModel(BaseModel):
        message: str

    return MyDummyModel


@pytest.fixture
def dummy_envelope_schema():
    """Event wrapper structure"""

    class MyDummyEnvelopeSchema(BaseModel):
        payload: Dict

    return MyDummyEnvelopeSchema


@pytest.fixture
def dummy_envelope(dummy_envelope_schema):
    class MyDummyEnvelope(BaseEnvelope):
        """Unwrap dummy event within payload key"""

        def parse(self, data: Dict[str, Any], model: BaseModel):
            parsed_enveloped = dummy_envelope_schema(**data)
            return self._parse(data=parsed_enveloped.payload, model=model)

    return MyDummyEnvelope