p59342861
/
monkey
forked from p15670423/monkey
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

UT: Add a new fixture for config object

This commit is contained in:
vakarisz 2022-06-21 15:50:37 +03:00 committed by Mike Salvatore
parent d8ac441c59
commit e83995d962
2 changed files with 130 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

127
monkey/tests/data_for_tests/monkey_configs/default_config.py Normal file
View File

@ -0,0 +1,127 @@
from common import OperatingSystems
from common.configuration import AgentConfigurationSchema
flat_config = {
"keep_tunnel_open_time": 30,
"post_breach_actions": [
{"name": "CommunicateAsBackdoorUser", "options": {}},
{"name": "ModifyShellStartupFiles", "options": {}},
{"name": "HiddenFiles", "options": {}},
{"name": "TrapCommand", "options": {}},
{"name": "ChangeSetuidSetgid", "options": {}},
{"name": "ScheduleJobs", "options": {}},
{"name": "Timestomping", "options": {}},
{"name": "AccountDiscovery", "options": {}},
{"name": "ProcessListCollection", "options": {}},
],
"credential_collectors": [
{"name": "MimikatzCollector", "options": {}},
{"name": "SSHCollector", "options": {}},
],
"payloads": [
{
"name": "ransomware",
"options": {
"encryption": {
"enabled": True,
"directories": {"linux_target_dir": "", "windows_target_dir": ""},
},
"other_behaviors": {"readme": True},
},
}
],
"custom_pbas": {
"linux_command": "",
"linux_filename": "",
"windows_command": "",
"windows_filename": "",
},
"propagation": {
"network_scan": {
"tcp": {
"timeout": 3000,
"ports": [
22,
80,
135,
443,
445,
2222,
3306,
3389,
5985,
5986,
7001,
8008,
8080,
8088,
8983,
9200,
9600,
],
},
"icmp": {"timeout": 1000},
"fingerprinters": [
{"name": "elastic", "options": {}},
{
"name": "http",
"options": {"http_ports": [80, 443, 7001, 8008, 8080, 8983, 9200, 9600]},
},
{"name": "mssql", "options": {}},
{"name": "smb", "options": {}},
{"name": "ssh", "options": {}},
],
"targets": {
"blocked_ips": [],
"inaccessible_subnets": [],
"local_network_scan": True,
"subnets": [],
},
},
"maximum_depth": 2,
"exploitation": {
"options": {"http_ports": [80, 443, 7001, 8008, 8080, 8983, 9200, 9600]},
"brute_force": [
{
"name": "MSSQLExploiter",
"options": {},
"supported_os": [OperatingSystems.WINDOWS.name],
},
{
"name": "PowerShellExploiter",
"options": {},
"supported_os": [OperatingSystems.WINDOWS.name],
},
{
"name": "SSHExploiter",
"options": {},
"supported_os": [OperatingSystems.LINUX.name],
},
{
"name": "SmbExploiter",
"options": {"smb_download_timeout": 30},
"supported_os": [OperatingSystems.WINDOWS.name],
},
{
"name": "WmiExploiter",
"options": {"smb_download_timeout": 30},
"supported_os": [OperatingSystems.WINDOWS.name],
},
],
"vulnerability": [
{
"name": "HadoopExploiter",
"options": {},
"supported_os": [OperatingSystems.LINUX.name, OperatingSystems.WINDOWS.name],
},
{
"name": "Log4ShellExploiter",
"options": {},
"supported_os": [OperatingSystems.LINUX.name, OperatingSystems.WINDOWS.name],
},
],
},
},
}
DEFAULT_CONFIG = AgentConfigurationSchema().load(flat_config)

3
monkey/tests/unit_tests/conftest.py
View File

@ -5,6 +5,9 @@ from typing import Callable, Dict
import pytest import pytest
from _pytest.monkeypatch import MonkeyPatch from _pytest.monkeypatch import MonkeyPatch
from tests.data_for_tests.monkey_configs.default_config import DEFAULT_CONFIG
from common.configuration import AgentConfiguration
MONKEY_BASE_PATH = str(Path(__file__).parent.parent.parent) MONKEY_BASE_PATH = str(Path(__file__).parent.parent.parent)
sys.path.insert(0, MONKEY_BASE_PATH) sys.path.insert(0, MONKEY_BASE_PATH)