mirror of https://github.com/mvt-project/mvt.git
Fixes testing issue
This commit is contained in:
parent
0d5377597f
commit
6cc67f3c1d
|
@ -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.
|
||||
|
|
|
@ -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
|
||||
|
|
Loading…
Reference in New Issue