From 85717f60405db1b1a1fe9a475ef624efc9bdce92 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Hubert=20S=C4=99kowski?= <hsekowski@splunk.com>
Date: Fri, 18 Feb 2022 08:33:08 +0100
Subject: [PATCH] fix: unit test addopted to the change

---
 .../requirement_tests/test_generator.py       |  2 +-
 .../test_test_generator.py                    | 24 +++++++++++++++----
 2 files changed, 21 insertions(+), 5 deletions(-)

diff --git a/pytest_splunk_addon/standard_lib/requirement_tests/test_generator.py b/pytest_splunk_addon/standard_lib/requirement_tests/test_generator.py
index 99db2aed5..98ca8d58f 100644
--- a/pytest_splunk_addon/standard_lib/requirement_tests/test_generator.py
+++ b/pytest_splunk_addon/standard_lib/requirement_tests/test_generator.py
@@ -118,7 +118,7 @@ def generate_cim_req_params(self):
             for file1 in os.listdir(req_file_path):
                 filename = os.path.join(req_file_path, file1)
                 LOGGER.info(filename)
-                if filename.endswith(".log") or filename.endswith(".log"):
+                if filename.endswith(".log") or filename.endswith(".xml"):
                     try:
                         self.check_xml_format(filename)
                     except Exception:
diff --git a/tests/unit/tests_standard_lib/test_requirement_tests/test_test_generator.py b/tests/unit/tests_standard_lib/test_requirement_tests/test_test_generator.py
index d299b2c1a..11930b3c4 100644
--- a/tests/unit/tests_standard_lib/test_requirement_tests/test_test_generator.py
+++ b/tests/unit/tests_standard_lib/test_requirement_tests/test_test_generator.py
@@ -133,11 +133,27 @@ def test_extract_params():
             ["requirement.xml"],
             [True],
             ["syslog"],
-            {"event": ["event_1", "event_2"]},
-            [["model_1:dataset_1", "model_2:dataset_2"], ["model_3:dataset_3"]],
-            ["event_name_2"],
+            {"event": ["<34>Oct 11 22:14:15 machine1 pr1:event_1"]},
+            [["model_1:dataset_1", "model_2:dataset_2"]],
+            ["event_name_1"],
             [{"field1": "value1", "field2": "value2"}, {"field3": "value3"}],
-            [],
+            [
+                (
+                    {
+                        "model_list": [
+                            ("model_1", "dataset_1", ""),
+                            ("model_2", "dataset_2", ""),
+                        ],
+                        "escaped_event": "event_1",
+                        "exceptions_dict": {"field3": "value3"},
+                        "Key_value_dict": {"field1": "value1", "field2": "value2"},
+                        "modinput_params": None,
+                        "transport_type": "syslog",
+                    },
+                    "model_1:dataset_1 "
+                    "model_2:dataset_2::fake_path/requirement.log::event_no::1::event_name::event_name_1",
+                ),
+            ],
         ),
         (
             ["req.log"],