p20319658
/
monkey
forked from p34709852/monkey
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Agent: Add load_plugin function to MockPuppet

This commit is contained in:
Shreya Malviya 2021-12-14 20:24:29 +05:30 committed by Mike Salvatore
parent fa2d2fdec2
commit 0e368fbfe9
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
monkey/infection_monkey/puppet/mock_puppet.py
View File

@ -11,6 +11,7 @@ from infection_monkey.i_puppet import (
PortStatus, PortStatus,
PostBreachData, PostBreachData,
) )
from infection_monkey.puppet.plugin_type import PluginType
DOT_1 = "10.0.0.1" DOT_1 = "10.0.0.1"
DOT_2 = "10.0.0.2" DOT_2 = "10.0.0.2"
@ -21,6 +22,9 @@ logger = logging.getLogger()
class MockPuppet(IPuppet): class MockPuppet(IPuppet):
def load_plugin(self, plugin: object, plugin_type: PluginType) -> None:
logger.debug(f"load_plugin({plugin}, {plugin_type})")
def run_sys_info_collector(self, name: str) -> Dict: def run_sys_info_collector(self, name: str) -> Dict:
logger.debug(f"run_sys_info_collector({name})") logger.debug(f"run_sys_info_collector({name})")
# TODO: More collectors # TODO: More collectors