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

Remove unnecessary code in test_setup_superuser.py

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-11-27 17:15:38 -03:00
commit c0ca8bbd04
1 changed files with 2 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

11
tests/test_setup_superuser.py
View file

@ -1,6 +1,5 @@
from unittest.mock import MagicMock, patch 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.settings.constants import DEFAULT_SUPERUSER, DEFAULT_SUPERUSER_PASSWORD
from langflow.services.utils import teardown_superuser from langflow.services.utils import teardown_superuser
@ -104,13 +103,7 @@ def test_teardown_superuser_default_superuser(mock_get_session, mock_get_setting
teardown_superuser(mock_settings_service, mock_session) teardown_superuser(mock_settings_service, mock_session)
mock_session.query.assert_called_once_with(User) mock_session.query.assert_not_called()
actual_expr = mock_session.query.return_value.filter.call_args[0][0]
expected_expr = User.username == DEFAULT_SUPERUSER
assert str(actual_expr) == str(expected_expr)
mock_session.delete.assert_called_once_with(mock_user)
mock_session.commit.assert_called_once()
@patch("langflow.services.deps.get_settings_service") @patch("langflow.services.deps.get_settings_service")
@ -131,6 +124,6 @@ def test_teardown_superuser_no_default_superuser(mock_get_session, mock_get_sett
teardown_superuser(mock_settings_service, mock_session) teardown_superuser(mock_settings_service, mock_session)
mock_session.query.assert_not_called() mock_session.query.assert_called_once()
mock_session.delete.assert_not_called() mock_session.delete.assert_not_called()
mock_session.commit.assert_not_called() mock_session.commit.assert_not_called()