Island: Register AgentOTP resource with other authentication resources

guardicore · Mar 10, 2023 · a975ae4 · a975ae4
1 parent 6e4a3a4
commit a975ae4
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/monkey/monkey_island/cc/app.py b/monkey/monkey_island/cc/app.py
@@ -26,7 +26,6 @@
     ResetAgentConfiguration,
     TerminateAllAgents,
 )
-from monkey_island.cc.services.authentication_service.flask_resources import AgentOTP
 from monkey_island.cc.resources.exploitations.monkey_exploitation import MonkeyExploitation
 from monkey_island.cc.resources.island_mode import IslandMode
 from monkey_island.cc.resources.local_run import LocalRun
@@ -91,7 +90,6 @@ def init_api_resources(api: FlaskDIWrapper):
 def init_restful_endpoints(api: FlaskDIWrapper):
     api.add_resource(Root)
 
-    api.add_resource(AgentOTP)
     api.add_resource(Agents)
     api.add_resource(LocalRun)
 

diff --git a/...ey/monkey_island/cc/services/authentication_service/flask_resources/register_resources.py b/...ey/monkey_island/cc/services/authentication_service/flask_resources/register_resources.py
@@ -3,6 +3,7 @@
 from common import DIContainer
 
 from ..authentication_facade import AuthenticationFacade
+from .agent_otp import AgentOTP
 from .login import Login
 from .logout import Logout
 from .register import Register
@@ -18,3 +19,4 @@ def register_resources(api: flask_restful.Api, container: DIContainer):
     )
     api.add_resource(Login, *Login.urls, resource_class_args=(authentication_facade,))
     api.add_resource(Logout, *Logout.urls, resource_class_args=(authentication_facade,))
+    api.add_resource(AgentOTP, *AgentOTP.urls)