import multiprocessing as mp
from typing import Type


class ResourceProcess(mp.Process):
    def __init__(
        self,
        *args,
        **kwargs,
    ):
        super().__init__(
            *args,
            **kwargs,
        )
        resource_id = 0
        self._resource_id = resource_id

    @property
    def resource_id(self):
        return self._resource_id


def create_resource_context() -> Type[mp.context.DefaultContext]:
    context = mp.get_context()
    context.Process = ResourceProcess
    return context


def test_resource_pool(ctx):
    with ctx.Pool(2) as pool:
        payload = (tuple(range(3)) for _ in range(5))
        response = pool.starmap(
            _dummy_func,
            payload,
        )

    assert response == (
        expected := list(
            (
                (
                    (0, 1, 2),
                    {},
                ),
            )
            * 5
        )
    ), f"{response=}!={expected}"


def _dummy_func(*args, **kwargs):
    return args, kwargs


if __name__ == "__main__":

    test_resource_pool(mp.get_context())
    print("Standard pool success")
    ctx = create_resource_context()
    test_resource_pool(ctx)
    print("Inherited pool success")