import os
import unittest

from mcpuniverse.workflows.builder import WorkflowConfig


class TestWorkflowConfig(unittest.TestCase):

    def test(self):
        folder = os.path.dirname(os.path.realpath(__file__))
        config_file = os.path.join(folder, "../data/config/yaml_config.yaml")
        objects = WorkflowConfig.load(config_file)
        self.assertEqual(objects[0].kind, "llm")
        self.assertEqual(objects[1].kind, "agent")


if __name__ == "__main__":
    unittest.main()