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

UT: Fix database_initializer test

This commit is contained in:
Ilija Lazoroski 2021-09-30 10:09:47 +02:00
parent 1748955213
commit 9ea5a56abd
1 changed files with 12 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
monkey/tests/unit_tests/monkey_island/cc/setup/mongo/test_database_initializer.py
View File

@ -1,11 +1,9 @@
import json.decoder
from pathlib import Path from pathlib import Path
from unittest.mock import MagicMock from unittest.mock import MagicMock
import mongomock import mongomock
import pytest import pytest
from monkey_island.cc.server_utils.consts import MONKEY_ISLAND_ABS_PATH
from monkey_island.cc.setup.mongo.database_initializer import reset_database from monkey_island.cc.setup.mongo.database_initializer import reset_database
@ -47,15 +45,16 @@ def test_store_mitigations_on_mongo__invalid_mitigation(
reset_database() reset_database()
def test_get_all_mitigations(): def test_get_all_mitigations(monkeypatch, fake_mongo, fake_config):
attack_mitigation_path = ( fake_mongo.db.validate_collection = MagicMock(return_value=True)
Path(MONKEY_ISLAND_ABS_PATH) / "cc" / "setup" / "mongo" / "attack_mitigations.json"
) reset_database()
mitigations = list(fake_mongo.db.attack_mitigations.find({}))
with open(attack_mitigation_path) as mitigations:
mitigations = json.load(mitigations)["data"]
assert len(mitigations) >= 266 assert len(mitigations) >= 266
mitigation = next(iter(mitigations))["mitigations"][0]
mitigation = mitigations[0]["mitigations"][0]
assert mitigation["name"] is not None assert mitigation["name"] is not None
assert mitigation["description"] is not None assert mitigation["description"] is not None
assert mitigation["url"] is not None assert mitigation["url"] is not None