jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

Refactor test_setup_superuser.py

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-01-25 16:16:28 -03:00
commit 8502898032
1 changed files with 12 additions and 21 deletions
Download patch file Download diff file Expand all files Collapse all files

33
tests/test_setup_superuser.py
View file

@ -1,17 +1,12 @@
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,
DEFAULT_SUPERUSER_PASSWORD,
)
from langflow.services.utils import (
teardown_superuser,
)
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.utils.create_super_user")
# @patch("langflow.services.getters.get_settings_service") # @patch("langflow.services.deps.get_settings_service")
# # @patch("langflow.services.utils.verify_password") # # @patch("langflow.services.utils.verify_password")
# def test_setup_superuser( # def test_setup_superuser(
# mock_get_session, mock_create_super_user, mock_get_settings_service # 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) # assert str(actual_expr) == str(expected_expr)
@patch("langflow.services.getters.get_settings_service") @patch("langflow.services.deps.get_settings_service")
@patch("langflow.services.getters.get_session") @patch("langflow.services.deps.get_session")
def test_teardown_superuser_default_superuser( def test_teardown_superuser_default_superuser(mock_get_session, mock_get_settings_service):
mock_get_session, mock_get_settings_service
):
mock_settings_service = MagicMock() mock_settings_service = MagicMock()
mock_settings_service.auth_settings.AUTO_LOGIN = True mock_settings_service.auth_settings.AUTO_LOGIN = True
mock_settings_service.auth_settings.SUPERUSER = DEFAULT_SUPERUSER mock_settings_service.auth_settings.SUPERUSER = DEFAULT_SUPERUSER
@ -120,11 +113,9 @@ def test_teardown_superuser_default_superuser(
mock_session.commit.assert_called_once() mock_session.commit.assert_called_once()
@patch("langflow.services.getters.get_settings_service") @patch("langflow.services.deps.get_settings_service")
@patch("langflow.services.getters.get_session") @patch("langflow.services.deps.get_session")
def test_teardown_superuser_no_default_superuser( def test_teardown_superuser_no_default_superuser(mock_get_session, mock_get_settings_service):
mock_get_session, mock_get_settings_service
):
ADMIN_USER_NAME = "admin_user" ADMIN_USER_NAME = "admin_user"
mock_settings_service = MagicMock() mock_settings_service = MagicMock()
mock_settings_service.auth_settings.AUTO_LOGIN = False mock_settings_service.auth_settings.AUTO_LOGIN = False