From 157858cfeef480010d22d060ad2f08d08ad8f48b Mon Sep 17 00:00:00 2001
From: Thomas Patzke <thomas@patzke.org>
Date: Sat, 13 Jul 2024 02:13:31 +0200
Subject: [PATCH] Added test for pipelines resolution from directory

---
 tests/files/pipelines/pipeline-1.yml |  6 ++++++
 tests/files/pipelines/pipeline-2.yml |  6 ++++++
 tests/test_processing_resolver.py    | 20 +++++++++++++++++++-
 3 files changed, 31 insertions(+), 1 deletion(-)
 create mode 100644 tests/files/pipelines/pipeline-1.yml
 create mode 100644 tests/files/pipelines/pipeline-2.yml

diff --git a/tests/files/pipelines/pipeline-1.yml b/tests/files/pipelines/pipeline-1.yml
new file mode 100644
index 00000000..eb50f00b
--- /dev/null
+++ b/tests/files/pipelines/pipeline-1.yml
@@ -0,0 +1,6 @@
+name: Test 1
+priority: 10
+transformations:
+  - id: test-1
+    type: field_name_suffix
+    suffix: .test
\ No newline at end of file
diff --git a/tests/files/pipelines/pipeline-2.yml b/tests/files/pipelines/pipeline-2.yml
new file mode 100644
index 00000000..a8d4038b
--- /dev/null
+++ b/tests/files/pipelines/pipeline-2.yml
@@ -0,0 +1,6 @@
+name: Test 2
+priority: 10
+transformations:
+  - id: test-2
+    type: field_name_prefix
+    prefix: test.
\ No newline at end of file
diff --git a/tests/test_processing_resolver.py b/tests/test_processing_resolver.py
index 5270b6ea..1ef942d8 100644
--- a/tests/test_processing_resolver.py
+++ b/tests/test_processing_resolver.py
@@ -5,7 +5,10 @@
 )
 from sigma.processing.resolver import ProcessingPipelineResolver
 from sigma.processing.pipeline import ProcessingPipeline, ProcessingItem
-from sigma.processing.transformations import AddFieldnameSuffixTransformation
+from sigma.processing.transformations import (
+    AddFieldnamePrefixTransformation,
+    AddFieldnameSuffixTransformation,
+)
 from collections.abc import Iterable
 
 
@@ -58,6 +61,21 @@ def test_resolve_file(processing_pipeline_resolver: ProcessingPipelineResolver):
     )
 
 
+def test_resolve_directory(processing_pipeline_resolver):
+    assert processing_pipeline_resolver.resolve(["tests/files/pipelines"]) == ProcessingPipeline(
+        items=[
+            ProcessingItem(
+                AddFieldnameSuffixTransformation(".test"),
+                identifier="test-1",
+            ),
+            ProcessingItem(
+                AddFieldnamePrefixTransformation("test."),
+                identifier="test-2",
+            ),
+        ],
+    )
+
+
 def test_resolve_callable():
     pipeline = ProcessingPipeline(
         [ProcessingItem(AddFieldnameSuffixTransformation(".item-1"))],