OpenMetadata/ingestion/tests/unit/test_workflow_parse_example_config.py

30 lines
917 B
Python
Raw Normal View History

from os import walk
from unittest import TestCase
import yaml
from metadata.ingestion.api.parser import parse_workflow_config_gracefully
try:
from importlib.resources import files
except ImportError:
from importlib_resources import files
class TestWorkflowParse(TestCase):
"""
Test parsing scenarios of JSON Schemas
"""
def test_parse_workflow_config(self):
package_path = f'{files("metadata")}/examples/workflows'
workflow_files = [files for _, _, files in walk(package_path)]
for yaml_file in workflow_files[0]:
with self.subTest(file_name=yaml_file):
with open(f"{package_path}/{yaml_file}", "r") as file:
file_content = file.read()
self.assertTrue(
parse_workflow_config_gracefully(yaml.safe_load(file_content))
)
file.close()