path: root/tests/conftest.py

import click
import pytest

from enum import Enum
from pydantic import BaseModel, Field
from pydanclick import core, decorators
from typing import Generator


class GenericArguments(BaseModel):

    class FruitEnum(str, Enum):
        apple = "apple"
        banana = "banana"
        pineapple = "pineapple"

    name: str = Field(min_length=10)
    version: int = Field(gt=0)
    force: bool = Field(default=False, description="Force ABC.")
    fruit: FruitEnum

    integer_gt_lt: int = Field(gt=16, lt=64)
    integer_ge_le: int = Field(ge=16, le=64)
    integer_gt_le: int = Field(gt=16, le=64)
    integer_ge_lt: int = Field(ge=16, lt=64)


@pytest.fixture
def valid_arguments() -> Generator[GenericArguments, None, None]:
    yield GenericArguments(
        name="fruit_manager",
        version=12,
        force=True,
        fruit="pineapple",
        integer_gt_lt=20,
        integer_ge_le=21,
        integer_gt_le=22,
        integer_ge_lt=23,
    )


@pytest.fixture
def valid_command_entrypoint() -> Generator[click.Command, None, None]:

    @decorators.command()
    def entrypoint(parameters: GenericArguments):
        pass

    yield entrypoint