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

Tests: Remove post tests that interact with the model

This commit is contained in:
Ilija Lazoroski 2021-07-14 11:09:57 +02:00
parent 5fe7a9d204
commit 56b5e8bb87
1 changed files with 3 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
monkey/tests/unit_tests/monkey_island/cc/resources/test_island_mode.py
View File

@ -42,28 +42,15 @@ def test_island_mode_post__internal_server_error(monkeypatch, flask_client):
assert resp.status_code == 500
def test_island_mode_post__set_model(flask_client, uses_database):
flask_client.post(
"/api/island-mode", data=json.dumps({"mode": "ransomware"}), follow_redirects=True
)
assert IslandMode.objects[0].mode == "ransomware"
def test_island_mode_post__set_invalid_model(flask_client, uses_database):
flask_client.post(
"/api/island-mode", data=json.dumps({"mode": "bogus mode"}), follow_redirects=True
)
assert len(IslandMode.objects) == 0
@pytest.mark.parametrize("mode", ["ransomware", "advanced"])
def test_island_mode_get(flask_client, uses_database, mode):
def test_island_mode_endpoint(flask_client, uses_database, mode):
flask_client.post("/api/island-mode", data=json.dumps({"mode": mode}), follow_redirects=True)
resp = flask_client.get("/api/island-mode", follow_redirects=True)
assert resp.status_code == 200
assert json.loads(resp.data)["mode"] == mode
def test_island_mode_get__invalid_mode(flask_client, uses_database):
def test_island_mode_endpoint__invalid_mode(flask_client, uses_database):
flask_client.post(
"/api/island-mode", data=json.dumps({"mode": "bogus_mode"}), follow_redirects=True
)