From 85028980321283936d30f2da9e2f761a4a3b870b Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Thu, 25 Jan 2024 16:16:28 -0300 Subject: [PATCH] Refactor test_setup_superuser.py --- tests/test_setup_superuser.py | 33 ++++++++++++--------------------- 1 file changed, 12 insertions(+), 21 deletions(-) diff --git a/tests/test_setup_superuser.py b/tests/test_setup_superuser.py index 8cdcfc0c8..d3033b728 100644 --- a/tests/test_setup_superuser.py +++ b/tests/test_setup_superuser.py @@ -1,17 +1,12 @@ -from unittest.mock import patch, MagicMock -from langflow.services.database.models.user.user import User -from langflow.services.settings.constants import ( - DEFAULT_SUPERUSER, - DEFAULT_SUPERUSER_PASSWORD, -) -from langflow.services.utils import ( - teardown_superuser, -) +from unittest.mock import MagicMock, patch +from langflow.services.database.models.user.model import User +from langflow.services.settings.constants import DEFAULT_SUPERUSER, DEFAULT_SUPERUSER_PASSWORD +from langflow.services.utils import teardown_superuser -# @patch("langflow.services.getters.get_session") +# @patch("langflow.services.deps.get_session") # @patch("langflow.services.utils.create_super_user") -# @patch("langflow.services.getters.get_settings_service") +# @patch("langflow.services.deps.get_settings_service") # # @patch("langflow.services.utils.verify_password") # def test_setup_superuser( # mock_get_session, mock_create_super_user, mock_get_settings_service @@ -92,11 +87,9 @@ from langflow.services.utils import ( # assert str(actual_expr) == str(expected_expr) -@patch("langflow.services.getters.get_settings_service") -@patch("langflow.services.getters.get_session") -def test_teardown_superuser_default_superuser( - mock_get_session, mock_get_settings_service -): +@patch("langflow.services.deps.get_settings_service") +@patch("langflow.services.deps.get_session") +def test_teardown_superuser_default_superuser(mock_get_session, mock_get_settings_service): mock_settings_service = MagicMock() mock_settings_service.auth_settings.AUTO_LOGIN = True mock_settings_service.auth_settings.SUPERUSER = DEFAULT_SUPERUSER @@ -120,11 +113,9 @@ def test_teardown_superuser_default_superuser( mock_session.commit.assert_called_once() -@patch("langflow.services.getters.get_settings_service") -@patch("langflow.services.getters.get_session") -def test_teardown_superuser_no_default_superuser( - mock_get_session, mock_get_settings_service -): +@patch("langflow.services.deps.get_settings_service") +@patch("langflow.services.deps.get_session") +def test_teardown_superuser_no_default_superuser(mock_get_session, mock_get_settings_service): ADMIN_USER_NAME = "admin_user" mock_settings_service = MagicMock() mock_settings_service.auth_settings.AUTO_LOGIN = False