Fixes testing issue

This commit is contained in:
tek 2022-03-04 12:34:54 +01:00
parent 0d5377597f
commit 6cc67f3c1d
2 changed files with 5 additions and 5 deletions

View File

@ -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.

View File

@ -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