diff --git a/monkey/monkey_island/cc/app.py b/monkey/monkey_island/cc/app.py index 570d1bee8..fc5b2b44a 100644 --- a/monkey/monkey_island/cc/app.py +++ b/monkey/monkey_island/cc/app.py @@ -23,9 +23,7 @@ from monkey_island.cc.resources.AbstractResource import AbstractResource from monkey_island.cc.resources.agent_configuration import AgentConfiguration from monkey_island.cc.resources.agent_controls import StopAgentCheck, StopAllAgents from monkey_island.cc.resources.attack.attack_report import AttackReport -from monkey_island.cc.resources.auth.authenticate import Authenticate, init_jwt -from monkey_island.cc.resources.auth.register import Register -from monkey_island.cc.resources.auth.registration_status import RegistrationStatus +from monkey_island.cc.resources.auth import Authenticate, Register, RegistrationStatus, init_jwt from monkey_island.cc.resources.blackbox.log_blackbox_endpoint import LogBlackboxEndpoint from monkey_island.cc.resources.blackbox.monkey_blackbox_endpoint import MonkeyBlackboxEndpoint from monkey_island.cc.resources.blackbox.telemetry_blackbox_endpoint import ( diff --git a/monkey/monkey_island/cc/resources/auth/__init__.py b/monkey/monkey_island/cc/resources/auth/__init__.py index e69de29bb..d73489248 100644 --- a/monkey/monkey_island/cc/resources/auth/__init__.py +++ b/monkey/monkey_island/cc/resources/auth/__init__.py @@ -0,0 +1,3 @@ +from .authenticate import Authenticate, init_jwt +from .register import Register +from .registration_status import RegistrationStatus diff --git a/monkey/tests/unit_tests/monkey_island/cc/resources/auth/test_authenticate.py b/monkey/tests/unit_tests/monkey_island/cc/resources/auth/test_authenticate.py index 6d5af8a94..ab0184262 100644 --- a/monkey/tests/unit_tests/monkey_island/cc/resources/auth/test_authenticate.py +++ b/monkey/tests/unit_tests/monkey_island/cc/resources/auth/test_authenticate.py @@ -4,7 +4,7 @@ from unittest.mock import MagicMock import pytest from common.utils.exceptions import IncorrectCredentialsError -from monkey_island.cc.resources.auth.authenticate import Authenticate +from monkey_island.cc.resources.auth import Authenticate USERNAME = "test_user" PASSWORD = "test_password" diff --git a/monkey/tests/unit_tests/monkey_island/cc/resources/auth/test_register.py b/monkey/tests/unit_tests/monkey_island/cc/resources/auth/test_register.py index aee054f84..a759c2fc3 100644 --- a/monkey/tests/unit_tests/monkey_island/cc/resources/auth/test_register.py +++ b/monkey/tests/unit_tests/monkey_island/cc/resources/auth/test_register.py @@ -4,7 +4,7 @@ from unittest.mock import MagicMock import pytest from common.utils.exceptions import AlreadyRegisteredError, InvalidRegistrationCredentialsError -from monkey_island.cc.resources.auth.register import Register +from monkey_island.cc.resources.auth import Register REGISTRATION_URL = Register.urls[0] diff --git a/monkey/tests/unit_tests/monkey_island/cc/resources/auth/test_registration_status.py b/monkey/tests/unit_tests/monkey_island/cc/resources/auth/test_registration_status.py index 350590df8..76d6406a9 100644 --- a/monkey/tests/unit_tests/monkey_island/cc/resources/auth/test_registration_status.py +++ b/monkey/tests/unit_tests/monkey_island/cc/resources/auth/test_registration_status.py @@ -2,7 +2,7 @@ from unittest.mock import MagicMock import pytest -from monkey_island.cc.resources.auth.registration_status import RegistrationStatus +from monkey_island.cc.resources.auth import RegistrationStatus REGISTRATION_STATUS_URL = RegistrationStatus.urls[0] diff --git a/monkey/tests/unit_tests/monkey_island/cc/resources/conftest.py b/monkey/tests/unit_tests/monkey_island/cc/resources/conftest.py index 4fb2e7be3..1b914b3e0 100644 --- a/monkey/tests/unit_tests/monkey_island/cc/resources/conftest.py +++ b/monkey/tests/unit_tests/monkey_island/cc/resources/conftest.py @@ -7,7 +7,7 @@ from tests.monkey_island import OpenErrorFileRepository from tests.unit_tests.monkey_island.conftest import init_mock_app import monkey_island.cc.app -import monkey_island.cc.resources.auth.authenticate +import monkey_island.cc.resources.auth import monkey_island.cc.resources.island_mode from monkey_island.cc.repository import IFileRepository