wip: decorator + arguments tests

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])