OpenMetadata/ingestion/tests/unit/test_workflow_parse_example_config.py

from os import walk
from pathlib import Path
from unittest import TestCase

import yaml

from metadata.ingestion.api.parser import parse_workflow_config_gracefully


class TestWorkflowParse(TestCase):
    """
    Test parsing scenarios of JSON Schemas
    """

    def test_parse_workflow_config(self):
        package_path = (
            f"{Path(__file__).parent.parent.parent}/src/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()