p20319658
/
monkey
forked from p34709852/monkey
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Island: Get Island mode from simulation repository in IslandMode's GET

This commit is contained in:
Shreya Malviya 2022-09-21 16:31:24 +05:30
parent 1f96ab5ea9
commit d9d388d41c
1 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
monkey/monkey_island/cc/resources/island_mode.py
View File

@ -6,6 +6,7 @@ from flask import request
from monkey_island.cc.event_queue import IIslandEventQueue, IslandEventTopic from monkey_island.cc.event_queue import IIslandEventQueue, IslandEventTopic
from monkey_island.cc.models import IslandMode as IslandModeEnum from monkey_island.cc.models import IslandMode as IslandModeEnum
from monkey_island.cc.repository import ISimulationRepository
from monkey_island.cc.resources.AbstractResource import AbstractResource from monkey_island.cc.resources.AbstractResource import AbstractResource
from monkey_island.cc.resources.request_authentication import jwt_required from monkey_island.cc.resources.request_authentication import jwt_required
from monkey_island.cc.services import IslandModeService from monkey_island.cc.services import IslandModeService
@ -17,10 +18,14 @@ class IslandMode(AbstractResource):
urls = ["/api/island/mode"] urls = ["/api/island/mode"]
def __init__( def __init__(
self, island_mode_service: IslandModeService, island_event_queue: IIslandEventQueue self,
island_mode_service: IslandModeService,
island_event_queue: IIslandEventQueue,
simulation_repository: ISimulationRepository,
): ):
self._island_mode_service = island_mode_service self._island_mode_service = island_mode_service
self._island_event_queue = island_event_queue self._island_event_queue = island_event_queue
self._simulation_repository = simulation_repository
@jwt_required @jwt_required
def put(self): def put(self):
@ -38,5 +43,5 @@ class IslandMode(AbstractResource):
@jwt_required @jwt_required
def get(self): def get(self):
island_mode = self._island_mode_service.get_mode() island_mode = self._simulation_repository.get_mode()
return island_mode.value, HTTPStatus.OK return island_mode.value, HTTPStatus.OK