p15670423
/
monkey
1
0
Fork 5
Code Issues Pull Requests 3 Projects Releases Wiki Activity

Island: Rename some variables in test_agent_signals.py

This commit is contained in:
Shreya Malviya 2022-09-23 17:13:30 +05:30 committed by Mike Salvatore
parent 2d42355e2c
commit 14c615e238
1 changed files with 14 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -11,14 +11,14 @@ from monkey_island.cc.repository import RetrievalError, StorageError
from monkey_island.cc.resources import AgentSignals from monkey_island.cc.resources import AgentSignals
from monkey_island.cc.services import AgentSignalsService from monkey_island.cc.services import AgentSignalsService
TIMESTAMP = 123456789 TIMESTAMP_1 = 123456789
TIMESTAMP_1 = 123546789 TIMESTAMP_2 = 123546789
SIGNALS = Signals(terminate=TIMESTAMP)
SIGNALS_1 = Signals(terminate=TIMESTAMP_1) SIGNALS_1 = Signals(terminate=TIMESTAMP_1)
SIGNALS_2 = Signals(terminate=TIMESTAMP_2)
AGENT_ID = UUID("c0dd10b3-e21a-4da9-9d96-a99c19ebd7c5") AGENT_ID_1 = UUID("c0dd10b3-e21a-4da9-9d96-a99c19ebd7c5")
AGENT_ID_1 = UUID("9b4279f6-6ec5-4953-821e-893ddc71a988") AGENT_ID_2 = UUID("9b4279f6-6ec5-4953-821e-893ddc71a988")
@pytest.fixture @pytest.fixture
@ -51,7 +51,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={"kill_time": TIMESTAMP}, json={"kill_time": TIMESTAMP_1},
follow_redirects=True, follow_redirects=True,
) )
assert resp.status_code == HTTPStatus.NO_CONTENT assert resp.status_code == HTTPStatus.NO_CONTENT
@ -62,8 +62,8 @@ def test_agent_signals_terminate_all_post(flask_client):
[ [
"bad timestamp", "bad timestamp",
{}, {},
{"wrong_key": TIMESTAMP}, {"wrong_key": TIMESTAMP_1},
TIMESTAMP, TIMESTAMP_1,
], ],
) )
def test_agent_signals_terminate_all_post__invalid_timestamp(flask_client, bad_data): def test_agent_signals_terminate_all_post__invalid_timestamp(flask_client, bad_data):
@ -77,7 +77,10 @@ def test_agent_signals_terminate_all_post__invalid_timestamp(flask_client, bad_d
@pytest.mark.parametrize( @pytest.mark.parametrize(
"url, signals", "url, signals",
[(f"/api/agent-signals/{AGENT_ID}", SIGNALS), (f"/api/agent-signals/{AGENT_ID_1}", SIGNALS_1)], [
(f"/api/agent-signals/{AGENT_ID_1}", SIGNALS_1),
(f"/api/agent-signals/{AGENT_ID_2}", SIGNALS_2),
],
) )
def test_agent_signals_get(flask_client, mock_agent_signals_service, url, signals): def test_agent_signals_get(flask_client, mock_agent_signals_service, url, signals):
mock_agent_signals_service.get_signals.return_value = signals mock_agent_signals_service.get_signals.return_value = signals
@ -89,8 +92,8 @@ def test_agent_signals_get(flask_client, mock_agent_signals_service, url, signal
@pytest.mark.parametrize( @pytest.mark.parametrize(
"url, error", "url, error",
[ [
(f"/api/agent-signals/{AGENT_ID}", RetrievalError), (f"/api/agent-signals/{AGENT_ID_1}", RetrievalError),
(f"/api/agent-signals/{AGENT_ID_1}", StorageError), (f"/api/agent-signals/{AGENT_ID_2}", StorageError),
], ],
) )
def test_agent_signals_get__internal_server_error( def test_agent_signals_get__internal_server_error(