adding tests for logging

tests/test_attack_log.py

@@ -122,6 +122,100 @@ class TestMachineConfig(unittest.TestCase):
         self.assertEqual(data[0]["kali_name"], attack_name)
         self.assertEqual(data[0]["hunting_tag"], "MITRE_" + ttp)
 
+    def test_narration_start(self):
+        """ Starting a narration """
+        al = AttackLog()
+        text = "texttextext"
+
+        al.start_narration(text
+                           )
+        data = al.get_dict()
+        self.assertEqual(data[0]["event"], "start")
+        self.assertEqual(data[0]["type"], "narration")
+        self.assertEqual(data[0]["sub_type"], "user defined narration")
+        self.assertEqual(data[0]["text"], text)
+
+    def test_build_start(self):
+        """ Starting a build """
+        al = AttackLog()
+        dl_uri = "asource"
+        dl_uris = "a target"
+        payload = "1234"
+        platform = "a name"
+        architecture = "arch"
+        lhost = "lhost"
+        lport = 8080
+        filename = "afilename"
+        encoding = "encoded"
+        encoded_filename = "ef"
+        sRDI_conversion = True
+        for_step = 4
+        comment = "this is a comment"
+
+        al.start_build(dl_uri=dl_uri,
+                       dl_uris=dl_uris,
+                       payload=payload,
+                       platform=platform,
+                       architecture=architecture,
+                       lhost=lhost,
+                       lport=lport,
+                       filename=filename,
+                       encoding=encoding,
+                       encoded_filename=encoded_filename,
+                       sRDI_conversion=sRDI_conversion,
+                       for_step=for_step,
+                       comment=comment
+                       )
+        data = al.get_dict()
+        self.assertEqual(data[0]["event"], "start")
+        self.assertEqual(data[0]["type"], "build")
+        self.assertEqual(data[0]["dl_uri"], dl_uri)
+        self.assertEqual(data[0]["dl_uris"], dl_uris)
+        self.assertEqual(data[0]["payload"], payload)
+        self.assertEqual(data[0]["platform"], platform)
+        self.assertEqual(data[0]["architecture"], architecture)
+        self.assertEqual(data[0]["lhost"], lhost)
+        self.assertEqual(data[0]["lport"], lport)
+        self.assertEqual(data[0]["filename"], filename)
+        self.assertEqual(data[0]["encoding"], encoding)
+        self.assertEqual(data[0]["encoded_filename"], encoded_filename)
+        self.assertEqual(data[0]["sRDI_conversion"], sRDI_conversion)
+        self.assertEqual(data[0]["for_step"], for_step)
+        self.assertEqual(data[0]["comment"], comment)
+
+    def test_build_start_default(self):
+        """ Starting a build default values"""
+        al = AttackLog()
+
+        al.start_build()
+        data = al.get_dict()
+        self.assertEqual(data[0]["event"], "start")
+        self.assertEqual(data[0]["type"], "build")
+        self.assertEqual(data[0]["dl_uri"], None)
+        self.assertEqual(data[0]["dl_uris"], None)
+        self.assertEqual(data[0]["payload"], None)
+        self.assertEqual(data[0]["platform"], None)
+        self.assertEqual(data[0]["architecture"], None)
+        self.assertEqual(data[0]["lhost"], None)
+        self.assertEqual(data[0]["lport"], None)
+        self.assertEqual(data[0]["filename"], None)
+        self.assertEqual(data[0]["encoding"], None)
+        self.assertEqual(data[0]["encoded_filename"], None)
+        self.assertEqual(data[0]["sRDI_conversion"], False)
+        self.assertEqual(data[0]["for_step"], None)
+        self.assertEqual(data[0]["comment"], None)
+
+    def test_build_stop(self):
+        """ Stopping a build """
+        al = AttackLog()
+        logid = "lid"
+
+        al.stop_build(logid=logid)
+        data = al.get_dict()
+        self.assertEqual(data[0]["event"], "stop")
+        self.assertEqual(data[0]["type"], "build")
+        self.assertEqual(data[0]["logid"], logid)
+
     def test_metasploit_attack_start(self):
         """ Starting a metasploit attack """
         al = AttackLog()
@@ -285,3 +379,64 @@ class TestMachineConfig(unittest.TestCase):
     def test_mitre_fix_ttp_is_MITRE_SOMETHING(self):
         """ Testing the mitre ttp fix for ttp being MITRE_ """
         self.assertEqual(app.attack_log.__mitre_fix_ttp__("MITRE_FOO"), "MITRE_FOO")
+
+    # tests for a bunch of default data covering caldera attacks. That way we will have some fallback if no data is submitted:
+    def test_get_caldera_default_name_missing(self):
+        """ Testing getting the caldera default name """
+        al = AttackLog()
+        self.assertEqual(al.get_caldera_default_name("missing"), None)
+
+    def test_get_caldera_default_name(self):
+        """ Testing getting the caldera default name """
+        al = AttackLog()
+        self.assertEqual(al.get_caldera_default_name("bd527b63-9f9e-46e0-9816-b8434d2b8989"), "whoami")
+
+    def test_get_caldera_default_description_missing(self):
+        """ Testing getting the caldera default description """
+        al = AttackLog()
+        self.assertEqual(al.get_caldera_default_description("missing"), None)
+
+    def test_get_caldera_default_description(self):
+        """ Testing getting the caldera default description """
+        al = AttackLog()
+        self.assertEqual(al.get_caldera_default_description("bd527b63-9f9e-46e0-9816-b8434d2b8989"), "Obtain user from current session")
+
+    def test_get_caldera_default_tactics_missing(self):
+        """ Testing getting the caldera default tactics """
+        al = AttackLog()
+        self.assertEqual(al.get_caldera_default_tactics("missing"), None)
+
+    def test_get_caldera_default_tactics(self):
+        """ Testing getting the caldera default tactics """
+        al = AttackLog()
+        self.assertEqual(al.get_caldera_default_tactics("bd527b63-9f9e-46e0-9816-b8434d2b8989"), "System Owner/User Discovery")
+
+    def test_get_caldera_default_tactics_id_missing(self):
+        """ Testing getting the caldera default tactics_id """
+        al = AttackLog()
+        self.assertEqual(al.get_caldera_default_tactics_id("missing"), None)
+
+    def test_get_caldera_default_tactics_id(self):
+        """ Testing getting the caldera default tactics_id """
+        al = AttackLog()
+        self.assertEqual(al.get_caldera_default_tactics_id("bd527b63-9f9e-46e0-9816-b8434d2b8989"), "T1033")
+
+    def test_get_caldera_default_situation_description_missing(self):
+        """ Testing getting the caldera default situation_description """
+        al = AttackLog()
+        self.assertEqual(al.get_caldera_default_situation_description("missing"), None)
+
+    def test_get_caldera_default_situation_description(self):
+        """ Testing getting the caldera default situation_description """
+        al = AttackLog()
+        self.assertEqual(al.get_caldera_default_situation_description("bd527b63-9f9e-46e0-9816-b8434d2b8989"), None)
+
+    def test_get_caldera_default_countermeasure_missing(self):
+        """ Testing getting the caldera default countermeasure """
+        al = AttackLog()
+        self.assertEqual(al.get_caldera_default_countermeasure("missing"), None)
+
+    def test_get_caldera_default_countermeasure(self):
+        """ Testing getting the caldera default countermeasure """
+        al = AttackLog()
+        self.assertEqual(al.get_caldera_default_countermeasure("bd527b63-9f9e-46e0-9816-b8434d2b8989"), None)