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

UT: Use OperatingSystems enum instead of strings

This commit is contained in:
vakarisz 2022-06-20 12:30:25 +03:00
parent 0474e2a5f7
commit 3757e33180
1 changed files with 25 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

33
monkey/tests/unit_tests/monkey_island/cc/services/test_config.py
View File

@ -1,5 +1,6 @@
import pytest import pytest
from common import OperatingSystems
from monkey_island.cc.services.config import ConfigService from monkey_island.cc.services.config import ConfigService
# If tests fail because config path is changed, sync with # If tests fail because config path is changed, sync with
@ -171,24 +172,40 @@ def test_format_config_for_agent__exploiters():
"http_ports": [80, 443, 7001, 8008, 8080, 9200], "http_ports": [80, 443, 7001, 8008, 8080, 9200],
}, },
"brute_force": [ "brute_force": [
{"name": "MSSQLExploiter", "supported_os": ["WINDOWS"], "options": {}}, {"name": "MSSQLExploiter", "supported_os": [OperatingSystems.WINDOWS], "options": {}},
{"name": "PowerShellExploiter", "supported_os": ["WINDOWS"], "options": {}}, {
{"name": "SSHExploiter", "supported_os": ["LINUX"], "options": {}}, "name": "PowerShellExploiter",
"supported_os": [OperatingSystems.WINDOWS],
"options": {},
},
{"name": "SSHExploiter", "supported_os": [OperatingSystems.LINUX], "options": {}},
{ {
"name": "SmbExploiter", "name": "SmbExploiter",
"supported_os": ["WINDOWS"], "supported_os": [OperatingSystems.WINDOWS],
"options": {"smb_download_timeout": 30}, "options": {"smb_download_timeout": 30},
}, },
{ {
"name": "WmiExploiter", "name": "WmiExploiter",
"supported_os": ["WINDOWS"], "supported_os": [OperatingSystems.WINDOWS],
"options": {"smb_download_timeout": 30}, "options": {"smb_download_timeout": 30},
}, },
], ],
"vulnerability": [ "vulnerability": [
{"name": "HadoopExploiter", "supported_os": ["LINUX", "WINDOWS"], "options": {}}, {
{"name": "Log4ShellExploiter", "supported_os": ["LINUX", "WINDOWS"], "options": {}}, "name": "HadoopExploiter",
{"name": "ZerologonExploiter", "supported_os": ["WINDOWS"], "options": {}}, "supported_os": [OperatingSystems.LINUX, OperatingSystems.WINDOWS],
"options": {},
},
{
"name": "Log4ShellExploiter",
"supported_os": [OperatingSystems.LINUX, OperatingSystems.WINDOWS],
"options": {},
},
{
"name": "ZerologonExploiter",
"supported_os": [OperatingSystems.WINDOWS],
"options": {},
},
], ],
} }
flat_monkey_config = ConfigService.format_flat_config_for_agent() flat_monkey_config = ConfigService.format_flat_config_for_agent()