From 2637701b361fe86525e03f5e9e08237a4c86ae02 Mon Sep 17 00:00:00 2001
From: Gustavo Leite <gustavoleite.ti@gmail.com>
Date: Thu, 5 Sep 2024 12:54:57 -0300
Subject: [PATCH] test(schema): add tests for schema validation

---
 pyproject.toml       |  8 +++-
 tests/test_schema.py | 97 ++++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 104 insertions(+), 1 deletion(-)
 create mode 100644 tests/test_schema.py

diff --git a/pyproject.toml b/pyproject.toml
index 505aa29..4bb6907 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -27,7 +27,8 @@ dev = [
     "black==24.4.2",
     "black[jupyter]==24.4.2",
     "isort==5.13.2",
-    "taskipy==1.13.0"
+    "taskipy==1.13.0",
+    "pytest",
 ]
 
 [project.scripts]
@@ -50,3 +51,8 @@ line_length = 79
 [tool.taskipy.tasks]
 lint = "black --check --diff . && isort --check --diff ."
 format = "black . && isort ."
+
+[tool.pytest.ini_options]
+pythonpath = [
+    ".",
+]
diff --git a/tests/test_schema.py b/tests/test_schema.py
new file mode 100644
index 0000000..4959063
--- /dev/null
+++ b/tests/test_schema.py
@@ -0,0 +1,97 @@
+import pytest
+from pydantic import ValidationError
+
+from spinner.schema import SpinnerMetadata
+
+# TEST: metadata -----------------------------------------------------------------------
+
+
+@pytest.fixture
+def metadata():
+    return {
+        "description": "Lorem ipsum dolor sit amet.",
+        "version": "1.0",
+        "runs": 2,
+        "timeout": 5,
+        "retry": True,
+        "retry_limit": 1,
+    }
+
+
+def test_metadata_valid(metadata):
+    """This test should always pass."""
+    SpinnerMetadata(**metadata)
+
+
+# TEST: metadata.version ---------------------------------------------------------------
+
+
+@pytest.fixture(params=["", "1", "1."])
+def metadata_invalid_version(metadata, request):
+    metadata["version"] = request.param
+    return metadata
+
+
+def test_metadata_version_invalid(metadata_invalid_version):
+    with pytest.raises(ValidationError) as _error:
+        SpinnerMetadata(**metadata_invalid_version)
+
+
+# TEST: metadata.runs ------------------------------------------------------------------
+
+
+@pytest.fixture(params=[0, -1])
+def metadata_invalid_runs(metadata, request):
+    metadata["runs"] = request.param
+    return metadata
+
+
+def test_metadata_runs_invalid(metadata_invalid_runs):
+    with pytest.raises(ValidationError) as _error:
+        SpinnerMetadata(**metadata_invalid_runs)
+
+
+# TEST: metadata.timeout ---------------------------------------------------------------
+
+
+@pytest.fixture(params=[0.0, -1.0])
+def metadata_invalid_timeout(metadata, request):
+    metadata["timeout"] = request.param
+    return metadata
+
+
+def test_metadata_timeout_none(metadata):
+    metadata["timeout"] = None
+    SpinnerMetadata(**metadata)
+
+
+def test_metadata_timeout_invalid(metadata_invalid_timeout):
+    with pytest.raises(ValidationError) as _error:
+        SpinnerMetadata(**metadata_invalid_timeout)
+
+
+# TEST: metadata.retry -----------------------------------------------------------------
+
+
+@pytest.fixture
+def metadata_no_retry(metadata):
+    metadata.pop("retry", None)
+    return metadata
+
+
+def test_metadata_retry_missing(metadata_no_retry):
+    SpinnerMetadata(**metadata_no_retry)
+
+
+# TEST: metadata.retry_limit -----------------------------------------------------------
+
+
+@pytest.fixture(params=[-1])
+def metadata_invalid_retry_limit(metadata, request):
+    metadata["retry_limit"] = request.param
+    return metadata
+
+
+def test_metadata_retry_limit_invalid(metadata_invalid_retry_limit):
+    with pytest.raises(ValidationError) as _error:
+        SpinnerMetadata(**metadata_invalid_retry_limit)