From 6cc67f3c1df6f2b610afe27a18780f4e14e10217 Mon Sep 17 00:00:00 2001 From: tek Date: Fri, 4 Mar 2022 12:34:54 +0100 Subject: [PATCH] Fixes testing issue --- tests/ios/test_sms.py | 6 +++--- tests/test_check_backup.py | 4 ++-- 2 files changed, 5 insertions(+), 5 deletions(-) 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