from typing import Annotated

import pydantic
from polyfactory.factories.pydantic_factory import ModelFactory
from pydantic import AfterValidator, Field

the_list = ["foo", "bar", "baz"]  # Some dynamically populated list


def l_validator(v):
    if v not in the_list:
        raise ValueError(f"{v} not in {the_list}")
    return v


TheListType = Annotated[str, AfterValidator(l_validator), Field(examples=the_list)]


class WithoutNone(pydantic.BaseModel):
    f: TheListType


class WithNone(pydantic.BaseModel):
    f: TheListType | None


class WithoutNoneFactory(ModelFactory[WithoutNone]):
    __check_model__ = True
    __use_examples__ = True


class WithNoneFactory(ModelFactory[WithNone]):
    __check_model__ = True
    __use_examples__ = True

# These three works fine
print("Without, batch:", WithoutNoneFactory.batch(2))
print("With, batch:", WithNoneFactory.batch(2))
print("Without, coverage:", list(WithoutNoneFactory.coverage()))
# Breaks here
print("With, coverage:", list(WithNoneFactory.coverage()))