mvt/tests/ios/test_datausage.py

32 lines
1.1 KiB
Python
Raw Normal View History

2022-01-18 15:00:03 +00:00
# Mobile Verification Toolkit (MVT)
2022-01-30 19:15:01 +00:00
# Copyright (c) 2021-2022 The MVT Project Authors.
2022-01-18 15:00:03 +00:00
# Use of this software is governed by the MVT License 1.1 that can be found at
# https://license.mvt.re/1.1/
2021-12-16 11:50:12 +00:00
import logging
from mvt.common.indicators import Indicators
2021-12-16 11:50:12 +00:00
from mvt.common.module import run_module
2022-01-18 15:00:03 +00:00
from mvt.ios.modules.mixed.net_datausage import Datausage
2021-12-16 11:50:12 +00:00
2022-01-07 16:51:21 +00:00
from ..utils import get_backup_folder
2021-12-16 11:50:12 +00:00
2022-01-18 15:00:03 +00:00
2021-12-16 11:50:12 +00:00
class TestDatausageModule:
def test_datausage(self):
m = Datausage(base_folder=get_backup_folder(), log=logging, results=[])
2021-12-16 11:50:12 +00:00
run_module(m)
assert len(m.results) == 42
assert len(m.timeline) == 60
assert len(m.detected) == 0
2022-01-07 16:51:21 +00:00
def test_detection(self, indicator_file):
m = Datausage(base_folder=get_backup_folder(), log=logging, results=[])
2021-12-16 11:50:12 +00:00
ind = Indicators(log=logging)
2022-01-07 16:51:21 +00:00
ind.parse_stix2(indicator_file)
2022-01-18 15:33:13 +00:00
# Adds a file that exists in the manifest.
ind.ioc_files[0]["processes"].append("CumulativeUsageTracker")
2021-12-16 11:50:12 +00:00
m.indicators = ind
run_module(m)
assert len(m.detected) == 2