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

Agent: Simplify RELAY_CONTROL_MESSAGE import

This commit is contained in:
Kekoa Kaaikala 2022-09-06 14:21:07 +00:00
parent b5345a38d9
commit 3ad96a7e6b
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
monkey/infection_monkey/network/relay/__init__.py
View File

@ -1,4 +1,4 @@
from .relay_connection_handler import RelayConnectionHandler
from .relay_connection_handler import RelayConnectionHandler, RELAY_CONTROL_MESSAGE
from .relay_user_handler import RelayUser, RelayUserHandler
from .sockets_pipe import SocketsPipe
from .tcp_connection_handler import TCPConnectionHandler

2
monkey/tests/unit_tests/infection_monkey/network/relay/test_relay_connection_handler.py
View File

@ -5,11 +5,11 @@ from unittest.mock import MagicMock
import pytest
from monkey.infection_monkey.network.relay import (
RELAY_CONTROL_MESSAGE,
RelayConnectionHandler,
RelayUserHandler,
TCPPipeSpawner,
)
from monkey.infection_monkey.network.relay.relay_connection_handler import RELAY_CONTROL_MESSAGE
USER_ADDRESS = "0.0.0.1"