summaryrefslogtreecommitdiffstats
path: root/tests
diff options
context:
space:
mode:
Diffstat (limited to 'tests')
-rw-r--r--tests/__init__.py0
-rw-r--r--tests/conftest.py28
-rw-r--r--tests/test_core.py38
3 files changed, 60 insertions, 6 deletions
diff --git a/tests/__init__.py b/tests/__init__.py
new file mode 100644
index 0000000..e69de29
--- /dev/null
+++ b/tests/__init__.py
diff --git a/tests/conftest.py b/tests/conftest.py
index e598a79..7d5e2c7 100644
--- a/tests/conftest.py
+++ b/tests/conftest.py
@@ -1,7 +1,10 @@
+import click
+import pytest
+
from enum import Enum
from pydantic import BaseModel, Field
-
-import pytest
+from pydanclick import core
+from typing import Generator
class GenericArguments(BaseModel):
@@ -15,3 +18,24 @@ class GenericArguments(BaseModel):
version: int = Field(gt=0)
force: bool = Field(default=False, description="Force ABC.")
fruit: FruitEnum
+
+
+@pytest.fixture
+def valid_arguments() -> Generator[GenericArguments, None, None]:
+ yield GenericArguments(
+ name="fruit_manager",
+ version=12,
+ force=True,
+ fruit="pineapple",
+ )
+
+
+@pytest.fixture
+def valid_command_entrypoint() -> Generator[click.Command, None, None]:
+
+ @click.command()
+ @core.generate_cli_options()
+ def entrypoint(parameters: GenericArguments):
+ pass
+
+ yield entrypoint
diff --git a/tests/test_core.py b/tests/test_core.py
index 5993abc..cc4755f 100644
--- a/tests/test_core.py
+++ b/tests/test_core.py
@@ -1,8 +1,38 @@
+import click
+import pickle
import pytest
-from pydanclick import core, schemas
-from tests.conftest import GenericArguments
+def test_get_option_arguments_ok(valid_arguments):
+ # core.get_option_arguments(schema, parameter, "mega-option")
+ pass
-def test_get_option_arguments_ok():
- core.get_option_arguments()
+
+def test_generate_cli_options_start(valid_command_entrypoint):
+
+ with pytest.raises(SystemExit):
+ valid_command_entrypoint()
+
+
+@pytest.mark.parametrize(
+ "argument_info",
+ [
+ ["name", ["--name"], click.STRING],
+ ["version", ["--version"], click.IntRange(min=0)],
+ ["force", ["--force"], click.BOOL],
+ ["fruit", ["--fruit"], click.STRING],
+ ]
+)
+def test_generate_cli_options_arguments(
+ valid_command_entrypoint,
+ argument_info,
+):
+ option = next(
+ filter(
+ lambda el: el.name == argument_info[0],
+ valid_command_entrypoint.params
+ )
+ )
+
+ assert option.opts == argument_info[1]
+ assert pickle.dumps(option.type) == pickle.dumps(argument_info[2])
remember that computers suck.