rosetta-ce 1.7.2__tar.gz → 1.7.3__tar.gz

{rosetta_ce-1.7.2/rosetta_ce.egg-info → rosetta_ce-1.7.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: rosetta-ce
-Version: 1.7.2
+Version: 1.7.3
 Summary: Rosetta is a Python package that can be used to fake security logs and alerts for testing different detection and response use cases.
 Home-page: https://github.com/ayman-m/rosetta
 Author: Ayman Mahmoud

{rosetta_ce-1.7.2 → rosetta_ce-1.7.3}/rosetta/rfaker.py

@@ -837,6 +837,7 @@ class Events:
         # Generate analyst list if not provided in observables
         incident_types = observables.incident_types if observables and observables.incident_types else INCIDENTS_TYPES
         analysts = observables.analysts if observables and observables.analysts else [faker.unique.first_name() for _ in range(10)]
+        severities = observables.severity if observables and observables.severity else [faker.random_int(min=1, max=5) for _ in range(10)]
 
         incident_type_cycle = itertools.cycle(incident_types)
         for i in range(count):
@@ -844,8 +845,8 @@ class Events:
             duration = random.randint(1, 5)
             incident_type = next(incident_type_cycle)
             analyst = random.choice(analysts)
-            severity = Events._set_field('severity', observables) or faker.random_int(min=1, max=5)
-            description = Events._set_field('terms', observables) or faker.sentence(nb_words=10)
+            severity =  random.choice(severities)
+            description = Events._set_field('terms')
 
             # Add base fields
             incident = {}

{rosetta_ce-1.7.2 → rosetta_ce-1.7.3/rosetta_ce.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: rosetta-ce
-Version: 1.7.2
+Version: 1.7.3
 Summary: Rosetta is a Python package that can be used to fake security logs and alerts for testing different detection and response use cases.
 Home-page: https://github.com/ayman-m/rosetta
 Author: Ayman Mahmoud

{rosetta_ce-1.7.2 → rosetta_ce-1.7.3}/setup.py

@@ -5,7 +5,7 @@ with open("README.md", "r") as fh:
 
 setuptools.setup(
     name="rosetta-ce",
-    version="1.7.2",
+    version="1.7.3",
     author="Ayman Mahmoud",
     author_email="content@ayman.online",
     description="Rosetta is a Python package that can be used to fake security logs and alerts for testing different "

{rosetta_ce-1.7.2 → rosetta_ce-1.7.3}/tests/test_rfaker.py

@@ -150,5 +150,15 @@ class TestRFaker(unittest.TestCase):
             self.assertIn(observables_list.src_host[0], message)
 
 
+    def test_incident(self):
+        start_time = time.time()
+        fake_messages = Events.incidents(count=10)
+        end_time = time.time()
+        latency = end_time - start_time
+        print(f"Incident generation latency for 10 logs: {latency:.4f} seconds")
+
+        self.assertTrue(isinstance(fake_messages, list))
+        self.assertEqual(len(fake_messages), 10)
+
 if __name__ == '__main__':
     unittest.main()