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

UT: Fix POST tests in test_agent_signals.py

This commit is contained in:
Shreya Malviya 2022-09-23 14:04:00 +05:30 committed by Mike Salvatore
parent 1afe625395
commit 5bf63c1221
1 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
monkey/tests/unit_tests/monkey_island/cc/resources/test_agent_signals.py
View File

@ -7,6 +7,7 @@ from tests.common import StubDIContainer
from monkey_island.cc.event_queue import IIslandEventQueue from monkey_island.cc.event_queue import IIslandEventQueue
from monkey_island.cc.resources import AgentSignals from monkey_island.cc.resources import AgentSignals
from monkey_island.cc.services import AgentSignalsService
TIMESTAMP = 123456789 TIMESTAMP = 123456789
@ -31,11 +32,13 @@ def flask_client_builder(build_flask_client):
def inner(side_effect=None): def inner(side_effect=None):
container = StubDIContainer() container = StubDIContainer()
# TODO: Add AgentSignalsService and add values on publish
mock_island_event_queue = MagicMock(spec=IIslandEventQueue) mock_island_event_queue = MagicMock(spec=IIslandEventQueue)
mock_island_event_queue.publish.side_effect = side_effect mock_island_event_queue.publish.side_effect = side_effect
container.register_instance(IIslandEventQueue, mock_island_event_queue) container.register_instance(IIslandEventQueue, mock_island_event_queue)
mock_agent_signals_service = MagicMock(spec=AgentSignalsService)
container.register_instance(AgentSignalsService, mock_agent_signals_service)
with build_flask_client(container) as flask_client: with build_flask_client(container) as flask_client:
return flask_client return flask_client
@ -50,7 +53,7 @@ def flask_client(flask_client_builder):
def test_agent_signals_terminate_all_post(flask_client): def test_agent_signals_terminate_all_post(flask_client):
resp = flask_client.post( resp = flask_client.post(
AgentSignals.urls[0], AgentSignals.urls[0],
json={"terminate": TIMESTAMP}, json={"kill_time": TIMESTAMP},
follow_redirects=True, follow_redirects=True,
) )
assert resp.status_code == HTTPStatus.NO_CONTENT assert resp.status_code == HTTPStatus.NO_CONTENT
@ -62,7 +65,6 @@ def test_agent_signals_terminate_all_post(flask_client):
"bad timestamp", "bad timestamp",
{}, {},
{"wrong_key": TIMESTAMP}, {"wrong_key": TIMESTAMP},
{"extra_key": "blah", "terminate": TIMESTAMP},
TIMESTAMP, TIMESTAMP,
], ],
) )