diff --git a/tests/ios/test_sms.py b/tests/ios/test_sms.py index 16adf3b..860f1dc 100644 --- a/tests/ios/test_sms.py +++ b/tests/ios/test_sms.py @@ -9,19 +9,19 @@ from mvt.common.indicators import Indicators from mvt.common.module import run_module from mvt.ios.modules.mixed.sms import SMS -from ..utils import get_backup_folder +from ..utils import get_ios_backup_folder class TestSMSModule: def test_sms(self): - m = SMS(base_folder=get_backup_folder(), log=logging, results=[]) + m = SMS(base_folder=get_ios_backup_folder(), log=logging, results=[]) run_module(m) assert len(m.results) == 1 assert len(m.timeline) == 1 assert len(m.detected) == 0 def test_detection(self, indicator_file): - m = SMS(base_folder=get_backup_folder(), log=logging, results=[]) + m = SMS(base_folder=get_ios_backup_folder(), log=logging, results=[]) ind = Indicators(log=logging) ind.parse_stix2(indicator_file) # Adds a file that exists in the manifest. diff --git a/tests/test_check_backup.py b/tests/test_check_backup.py index e71ef4b..c099ddd 100644 --- a/tests/test_check_backup.py +++ b/tests/test_check_backup.py @@ -7,12 +7,12 @@ from click.testing import CliRunner from mvt.ios.cli import check_backup -from .utils import get_backup_folder +from .utils import get_ios_backup_folder class TestCheckBackupCommand: def test_check(self): runner = CliRunner() - path = get_backup_folder() + path = get_ios_backup_folder() result = runner.invoke(check_backup, [path]) assert result.exit_code == 0