Tests: Use SSHExploiter instead of WmiExploiter in expliot telem tests

WmiExploiter relies on impacket. Importing impacket is slow, which has a
negative impact on the speed of pytest collection. SSHExploiter is much
quicker to import.
This commit is contained in:
Mike Salvatore 2021-07-19 14:06:31 -04:00
parent 845c9d9ac3
commit 2496ed0889
1 changed files with 4 additions and 4 deletions

View File

@ -2,7 +2,7 @@ import json
import pytest import pytest
from infection_monkey.exploit.wmiexec import WmiExploiter from infection_monkey.exploit.sshexec import SSHExploiter
from infection_monkey.model.host import VictimHost from infection_monkey.model.host import VictimHost
from infection_monkey.telemetry.exploit_telem import ExploitTelem from infection_monkey.telemetry.exploit_telem import ExploitTelem
@ -19,10 +19,10 @@ HOST_AS_DICT = {
"default_tunnel": None, "default_tunnel": None,
"default_server": None, "default_server": None,
} }
EXPLOITER = WmiExploiter(HOST) EXPLOITER = SSHExploiter(HOST)
EXPLOITER_NAME = "WmiExploiter" EXPLOITER_NAME = "SSHExploiter"
EXPLOITER_INFO = { EXPLOITER_INFO = {
"display_name": WmiExploiter._EXPLOITED_SERVICE, "display_name": SSHExploiter._EXPLOITED_SERVICE,
"started": "", "started": "",
"finished": "", "finished": "",
"vulnerable_urls": [], "vulnerable_urls": [],