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

Island: Raise errors if repositories do not exist

This commit is contained in:
Kekoa Kaaikala 2022-09-29 20:03:11 +00:00
parent dd0c504743
commit f7a30e4608
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/monkey_island/cc/services/reporting/report.py
View File

@ -139,8 +139,10 @@ class ReportService:
@classmethod @classmethod
def get_accessible_machines(cls, machine: Machine): def get_accessible_machines(cls, machine: Machine):
if cls._node_repository is None or cls._machine_repository is None: if cls._node_repository is None:
return [] raise RuntimeError("Node repository does not exist")
elif cls._machine_repository is None:
raise RuntimeError("Machine repository does not exist")
nodes = cls._node_repository.get_nodes() nodes = cls._node_repository.get_nodes()
machine_iter = (node for node in nodes if node.machine_id == machine.id) machine_iter = (node for node in nodes if node.machine_id == machine.id)
@ -155,7 +157,7 @@ class ReportService:
@classmethod @classmethod
def get_all_machines(cls) -> Iterable[Machine]: def get_all_machines(cls) -> Iterable[Machine]:
if cls._machine_repository is None: if cls._machine_repository is None:
return iter(()) raise RuntimeError("Machine repository does not exist")
machines = cls._machine_repository.get_machines() machines = cls._machine_repository.get_machines()
t1, t2 = tee(machines) t1, t2 = tee(machines)