mirror of
https://github.com/mvt-project/mvt.git
synced 2024-06-25 22:08:55 +00:00
b35cd4bc73
This way when a detection is logged, the user can know which STIX2 file was matched by the module
33 lines
1.2 KiB
Python
33 lines
1.2 KiB
Python
# Mobile Verification Toolkit (MVT)
|
|
# Copyright (c) 2021 The MVT Project Authors.
|
|
# Use of this software is governed by the MVT License 1.1 that can be found at
|
|
# https://license.mvt.re/1.1/
|
|
|
|
import logging
|
|
import os
|
|
|
|
from mvt.common.indicators import Indicators
|
|
|
|
|
|
class TestIndicators:
|
|
def test_parse_stix2(self, indicator_file):
|
|
ind = Indicators(log=logging)
|
|
ind.load_indicators_files([indicator_file], load_default=False)
|
|
assert ind.ioc_files[0]["count"] == 4
|
|
assert len(ind.ioc_files[0]["domains"]) == 1
|
|
assert len(ind.ioc_files[0]["emails"]) == 1
|
|
assert len(ind.ioc_files[0]["file_names"]) == 1
|
|
assert len(ind.ioc_files[0]["processes"]) == 1
|
|
|
|
def test_check_domain(self, indicator_file):
|
|
ind = Indicators(log=logging)
|
|
ind.load_indicators_files([indicator_file], load_default=False)
|
|
assert ind.check_domain("https://www.example.org/foobar")
|
|
assert ind.check_domain("http://example.org:8080/toto")
|
|
|
|
def test_env_stix(self, indicator_file):
|
|
os.environ["MVT_STIX2"] = indicator_file
|
|
ind = Indicators(log=logging)
|
|
ind.load_indicators_files([], load_default=False)
|
|
assert ind.total_ioc_count == 4
|