pytest-jsonschema 1.0.0__py3-none-any.whl

pytest_jsonschema/__init__.py

@@ -0,0 +1 @@
+__version__ = "1.0.0"

pytest_jsonschema/fixtures.py

@@ -0,0 +1,44 @@
+from . import loaders
+from . import schemas
+from . import types
+from . import validator
+
+import pytest
+
+
+@pytest.fixture
+def schema_validate_file() -> types.FileValidator:
+    """Validate a file againstt a known JSON Schema."""
+
+    def func(path: types.StrPath, schema_name: str, file_type: str = "") -> bool:
+        """Validate a file against a known JSON Schema."""
+        data = loaders.data_from_file(path, file_type=file_type)
+        schema = schemas.load(schema_name)
+        return validator.validate(data, schema)
+
+    return func
+
+
+@pytest.fixture
+def schema_validate_string() -> types.StrValidator:
+    """Validate a string againstt a known JSON Schema."""
+
+    def func(data: str, schema_name: str, file_type: str = "") -> bool:
+        """Validate a string against a known JSON Schema."""
+        parsed = loaders.data_from_string(data, file_type=file_type)
+        schema = schemas.load(schema_name)
+        return validator.validate(parsed, schema)
+
+    return func
+
+
+@pytest.fixture
+def schema_validate() -> types.Validator:
+    """Validate a string againstt a known JSON Schema."""
+
+    def func(data: types.DataStructure, schema_name: str) -> bool:
+        """Validate a file against a known JSON Schema."""
+        schema = schemas.load(schema_name)
+        return validator.validate(data, schema)
+
+    return func

pytest_jsonschema/loaders.py

@@ -0,0 +1,50 @@
+from . import types
+from collections.abc import Callable
+from pathlib import Path
+from ruamel.yaml import YAML
+
+import json
+import sys
+
+
+if sys.version_info >= (3, 12):
+    import tomllib
+else:
+    import tomli as tomllib
+
+yaml = YAML()
+
+
+_FILETYPES: dict[str, str] = {
+    ".json": "json",
+    ".toml": "toml",
+    ".yaml": "yaml",
+    ".yml": "yaml",
+}
+
+_LOADERS: dict[str, Callable] = {
+    "json": json.loads,
+    "toml": tomllib.loads,
+    "yaml": yaml.load,
+}
+
+
+def _guess_format(path: Path) -> str:
+    """Given a path, try to guess the correct filetype."""
+    extension = path.suffix
+    return _FILETYPES.get(extension, "")
+
+
+def data_from_file(path: types.StrPath, file_type: str = "") -> dict | list:
+    """Load data from file."""
+    path = Path(path)
+    if file_type not in _LOADERS:
+        file_type = _guess_format(path)
+    data = path.read_text()
+    return data_from_string(data, file_type)
+
+
+def data_from_string(data: str, file_type: str) -> dict:
+    """Load data from string."""
+    loader = _LOADERS.get(file_type)
+    return loader(data) if loader else {}

pytest_jsonschema/schemas/__init__.py

@@ -0,0 +1,9 @@
+from pathlib import Path
+
+import json
+
+
+def load(schema_name: str) -> dict:
+    """Load data from string."""
+    base_path = Path(__file__).parent / f"{schema_name}.json"
+    return json.loads(base_path.read_text())

pytest_jsonschema/schemas/ansible-vars.json

@@ -0,0 +1,30 @@
+{
+  "$id": "https://raw.githubusercontent.com/ansible/ansible-lint/main/src/ansiblelint/schemas/vars.json",
+  "$schema": "http://json-schema.org/draft-07/schema",
+  "anyOf": [
+    {
+      "additionalProperties": false,
+      "patternProperties": {
+        "^(?!(False|None|True|and|any_errors_fatal|as|assert|async|await|become|become_exe|become_flags|become_method|become_user|break|check_mode|class|collections|connection|continue|debugger|def|del|diff|elif|else|environment|except|fact_path|finally|for|force_handlers|from|gather_facts|gather_subset|gather_timeout|global|handlers|hosts|if|ignore_errors|ignore_unreachable|import|in|is|lambda|max_fail_percentage|module_defaults|name|no_log|nonlocal|not|or|order|pass|port|post_tasks|pre_tasks|raise|remote_user|return|roles|run_once|serial|strategy|tags|tasks|throttle|timeout|try|vars|vars_files|vars_prompt|while|with|yield)$)[a-zA-Z_][\\w]*$": {}
+      },
+      "type": "object"
+    },
+    {
+      "pattern": "^\\$ANSIBLE_VAULT;",
+      "type": "string"
+    },
+    {
+      "type": "null"
+    }
+  ],
+  "description": "https://docs.ansible.com/ansible/latest/playbook_guide/playbooks_variables.html",
+  "examples": [
+    "playbooks/vars/*.yml",
+    "vars/*.yml",
+    "defaults/*.yml",
+    "host_vars/*.yml",
+    "group_vars/*.yml"
+  ],
+  "markdownDescription": "See [Using Variables](https://docs.ansible.com/ansible/latest/playbook_guide/playbooks_variables.html)",
+  "title": "Ansible Vars Schema"
+}