From 2def76283112fe73245ce4f8158461ad60605279 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Tue, 14 Nov 2023 21:57:48 -0300 Subject: [PATCH] =?UTF-8?q?=F0=9F=93=A6=20chore(getters.py):=20add=20gette?= =?UTF-8?q?rs=20for=20various=20services=20and=20session=20in=20langflow?= =?UTF-8?q?=20backend=20=F0=9F=94=A8=20refactor(test=5Fsetup=5Fsuperuser.p?= =?UTF-8?q?y):=20refactor=20imports=20and=20replace=20deprecated=20depende?= =?UTF-8?q?ncies=20in=20test=5Fsetup=5Fsuperuser.py?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../langflow/services/{deps.py => getters.py} | 0 tests/test_setup_superuser.py | 16 ++++++++-------- 2 files changed, 8 insertions(+), 8 deletions(-) rename src/backend/langflow/services/{deps.py => getters.py} (100%) diff --git a/src/backend/langflow/services/deps.py b/src/backend/langflow/services/getters.py similarity index 100% rename from src/backend/langflow/services/deps.py rename to src/backend/langflow/services/getters.py diff --git a/tests/test_setup_superuser.py b/tests/test_setup_superuser.py index 1d3ed5d68..5c1629708 100644 --- a/tests/test_setup_superuser.py +++ b/tests/test_setup_superuser.py @@ -1,4 +1,5 @@ -from unittest.mock import patch, MagicMock +from unittest.mock import MagicMock, patch + from langflow.services.database.models.user.user import User from langflow.services.settings.constants import ( DEFAULT_SUPERUSER, @@ -8,10 +9,9 @@ from langflow.services.utils import ( teardown_superuser, ) - -# @patch("langflow.services.deps.get_session") +# @patch("langflow.services.getters.get_session") # @patch("langflow.services.utils.create_super_user") -# @patch("langflow.services.deps.get_settings_service") +# @patch("langflow.services.getters.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,8 +92,8 @@ from langflow.services.utils import ( # assert str(actual_expr) == str(expected_expr) -@patch("langflow.services.deps.get_settings_service") -@patch("langflow.services.deps.get_session") +@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): mock_settings_service = MagicMock() mock_settings_service.auth_settings.AUTO_LOGIN = True @@ -118,8 +118,8 @@ def test_teardown_superuser_default_superuser(mock_get_session, mock_get_setting mock_session.commit.assert_called_once() -@patch("langflow.services.deps.get_settings_service") -@patch("langflow.services.deps.get_session") +@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): ADMIN_USER_NAME = "admin_user" mock_settings_service = MagicMock()