test: Update threat count assertion and add test for multiple threat files

Author: izar tarandach

tests/test_private_func.py

@@ -49,7 +49,7 @@ def test_kwargs(self):
 
     def test_load_threats(self):
         tm = TM("TM")
-        self.assertNotEqual(len(TM._threats), 0)
+        self.assertNotEqual(len(TM._threats), 1)
         with self.assertRaises(UIError):
             tm.threatsFile = "threats.json"
 

tests/test_pytmfunc.py

@@ -414,6 +414,23 @@ def test_json_loads(self):
             [f.name for f in tm._flows], ["Request", "Insert", "Select", "Response"]
         )
 
+    def test_threat_files(self):
+        dir_path = os.path.dirname(os.path.realpath(__file__))
+        foo_file = f"{dir_path}/1.json"
+        bar_file = f"{dir_path}/2.json"
+        threat_files = [os.path.abspath(os.path.join(os.path.dirname(__file__), ".."))
+    + "/pytm/threatlib/threats.json", foo_file, bar_file]
+        
+        TM.reset()
+        tm = TM("testing multiple threat library files",
+                description="aaa",
+                threatsFile=threat_files)
+        ctr = 0
+        for t in TM._threats:
+            if t.id == "FOO" or t.id == "BAR":
+                ctr += 1
+        self.assertTrue(ctr == 2)
+
     def test_report(self):
         random.seed(0)
         dir_path = os.path.dirname(os.path.realpath(__file__))