adapt tests to new constructor

Author: p-

tests/test_yaml_parser.py

@@ -18,6 +18,7 @@ def test_yaml_parser_basic_functionality(self):
         """Test basic YAML parsing functionality."""
         # create a temporary directory with test yaml files
         with tempfile.TemporaryDirectory() as temp_dir:
+            temp_path = Path(temp_dir)
             test_yaml_content = {
                 'seclab-taskflow-agent': {
                     'type': 'taskflow',
@@ -33,18 +34,19 @@ def test_yaml_parser_basic_functionality(self):
                 ]
             }
 
-            test_file = Path(temp_dir) / 'test_taskflow.yaml'
+            test_file = temp_path / 'test_taskflow.yaml'
             with open(test_file, 'w') as f:
                 yaml.dump(test_yaml_content, f)
 
-            # Test parsing
-            parser = YamlParser(temp_dir)
-            result = parser.get_yaml_dict(recurse=False)
+            parser = YamlParser(temp_path)
+            # get all yaml files in the directory
+            yaml_files = temp_path.glob('*.yaml')
+            result = parser.get_yaml_dict(yaml_files)
 
-            assert 'test_taskflow' in result
-            assert result['test_taskflow']['seclab-taskflow-agent']['type'] == 'taskflow'
-            assert len(result['test_taskflow']['taskflow']) == 1
-            assert result['test_taskflow']['taskflow'][0]['task']['agents'] == ['assistant']
+            assert 'test_taskflow.yaml' in result
+            assert result['test_taskflow.yaml']['seclab-taskflow-agent']['type'] == 'taskflow'
+            assert len(result['test_taskflow.yaml']['taskflow']) == 1
+            assert result['test_taskflow.yaml']['taskflow'][0]['task']['agents'] == ['assistant']
 
 
 class TestRealTaskflowFiles:
@@ -53,29 +55,36 @@ class TestRealTaskflowFiles:
     def test_parse_example_taskflows(self):
         """Test parsing the actual example taskflow files."""
         # this test uses the actual taskflows in the project
-        parser = YamlParser('taskflows/examples')
-        result = parser.get_yaml_dict()
+        examples_path = Path('taskflows/examples').absolute()
+        parser = YamlParser(examples_path)
+        
+        # Get all YAML files in the examples directory
+        yaml_files = examples_path.glob('*.yaml')
+        result = parser.get_yaml_dict(yaml_files)
 
         # should contain example files
         assert len(result) > 0
 
         # check that example.yaml is parsed correctly
-        example_task_flow = result['example']
+        example_task_flow = result['example.yaml']
         assert 'taskflow' in example_task_flow
         assert isinstance(example_task_flow['taskflow'], list)
-        assert len(example_task_flow['taskflow']) ==  4 # 4 tasks in taskflow
+        assert len(example_task_flow['taskflow']) == 4  # 4 tasks in taskflow
         assert example_task_flow['taskflow'][0]['task']['max_steps'] == 20
 
     def test_parse_all_taskflows(self):
         """Test parsing all example taskflow files in the project."""
-        parser = YamlParser('taskflows')
-        result = parser.get_yaml_dict(recurse=True, dir_namespace=True)
+        taskflows_path = Path('taskflows').absolute()
+        parser = YamlParser(taskflows_path)
+        
+        yaml_files = taskflows_path.rglob('*.yaml')
+        result = parser.get_yaml_dict(yaml_files)
 
         # should contain all taskflow files (including subdirs)
         assert len(result) == 13
 
-        # check access for files with namespacing
-        example_files = [key for key in result.keys() if key.startswith('examples/')]
+        # check access for files with directory structure in names
+        example_files = [key for key in result.keys() if 'examples/' in key]
         assert len(example_files) > 0