From 189b1d7fc63cca842ec5617223383dcf7e3569d2 Mon Sep 17 00:00:00 2001
From: Nex <nex@nex.sx>
Date: Wed, 2 Feb 2022 18:14:10 +0100
Subject: [PATCH] Fixed tests

---
 tests/common/test_indicators.py       | 10 +++++-----
 tests/ios/test_datausage.py           |  2 +-
 tests/ios/test_manifest.py            |  2 +-
 tests/ios/test_safari_browserstate.py |  2 +-
 4 files changed, 8 insertions(+), 8 deletions(-)

diff --git a/tests/common/test_indicators.py b/tests/common/test_indicators.py
index 7d7053a..dbfdd6b 100644
--- a/tests/common/test_indicators.py
+++ b/tests/common/test_indicators.py
@@ -13,11 +13,11 @@ class TestIndicators:
     def test_parse_stix2(self, indicator_file):
         ind = Indicators(log=logging)
         ind.load_indicators_files([indicator_file], load_default=False)
-        assert ind.ioc_files[0]["count"] == 4
-        assert len(ind.ioc_files[0]["domains"]) == 1
-        assert len(ind.ioc_files[0]["emails"]) == 1
-        assert len(ind.ioc_files[0]["file_names"]) == 1
-        assert len(ind.ioc_files[0]["processes"]) == 1
+        assert ind.ioc_collections[0]["count"] == 4
+        assert len(ind.ioc_collections[0]["domains"]) == 1
+        assert len(ind.ioc_collections[0]["emails"]) == 1
+        assert len(ind.ioc_collections[0]["file_names"]) == 1
+        assert len(ind.ioc_collections[0]["processes"]) == 1
 
     def test_check_domain(self, indicator_file):
         ind = Indicators(log=logging)
diff --git a/tests/ios/test_datausage.py b/tests/ios/test_datausage.py
index a89df1f..6fdd9e1 100644
--- a/tests/ios/test_datausage.py
+++ b/tests/ios/test_datausage.py
@@ -25,7 +25,7 @@ class TestDatausageModule:
         ind = Indicators(log=logging)
         ind.parse_stix2(indicator_file)
         # Adds a file that exists in the manifest.
-        ind.ioc_files[0]["processes"].append("CumulativeUsageTracker")
+        ind.ioc_collections[0]["processes"].append("CumulativeUsageTracker")
         m.indicators = ind
         run_module(m)
         assert len(m.detected) == 2
diff --git a/tests/ios/test_manifest.py b/tests/ios/test_manifest.py
index 0c9c8d4..37857fb 100644
--- a/tests/ios/test_manifest.py
+++ b/tests/ios/test_manifest.py
@@ -24,7 +24,7 @@ class TestManifestModule:
         m = Manifest(base_folder=get_backup_folder(), log=logging, results=[])
         ind = Indicators(log=logging)
         ind.parse_stix2(indicator_file)
-        ind.ioc_files[0]["file_names"].append("com.apple.CoreBrightness.plist")
+        ind.ioc_collections[0]["file_names"].append("com.apple.CoreBrightness.plist")
         m.indicators = ind
         run_module(m)
         assert len(m.detected) == 1
diff --git a/tests/ios/test_safari_browserstate.py b/tests/ios/test_safari_browserstate.py
index d5eb34e..b8179cd 100644
--- a/tests/ios/test_safari_browserstate.py
+++ b/tests/ios/test_safari_browserstate.py
@@ -28,7 +28,7 @@ class TestSafariBrowserStateModule:
         ind = Indicators(log=logging)
         ind.parse_stix2(indicator_file)
         # Adds a file that exists in the manifest.
-        ind.ioc_files[0]["domains"].append("en.wikipedia.org")
+        ind.ioc_collections[0]["domains"].append("en.wikipedia.org")
         m.indicators = ind
         run_module(m)
         assert len(m.detected) == 1