Shofel2_T124_python/venv/lib/python3.10/site-packages/yamale/yamale_testcase.py

50 lines
1.3 KiB
Python

import glob
import os
import itertools
from unittest import TestCase
import yamale
class YamaleTestCase(TestCase):
""" TestCase for easily validating YAML in your own tests.
`schema`: String of path to the schema file to use. One schema file per test case.
`yaml`: String or list of yaml files to validate. Accepts globs.
`base_dir`: String path to prepend to all other paths. This is optional.
"""
schema = None
yaml = None
base_dir = None
def validate(self, validators=None):
schema = self.schema
yaml = self.yaml
base_dir = self.base_dir
if schema is None:
return
if type(yaml) != list:
yaml = [yaml]
if base_dir is not None:
schema = os.path.join(base_dir, schema)
yaml = {os.path.join(base_dir, y) for y in yaml}
# Run yaml through glob and flatten list
yaml = set(itertools.chain(*map(glob.glob, yaml)))
# Remove schema from set of data files
yaml = yaml - {schema}
yamale_schema = yamale.make_schema(schema, validators=validators)
yamale_data = itertools.chain(*map(yamale.make_data, yaml))
for result in yamale.validate(yamale_schema, yamale_data):
if not result.isValid():
raise ValueError(result)
return True