opensource/dify
2
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-07-25 13:38:31 +08:00
Code Issues Packages Projects Releases Wiki Activity

test: use caplog in ext login tests (#39542)

Browse Source
This commit is contained in:
Evan 2026-07-24 21:27:05 +08:00 committed by GitHub
parent 799c7eea3a
commit a0b917425a
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
api/tests/unit_tests/extensions/test_ext_login.py
View File

@ -52,25 +52,25 @@ def test_on_user_logged_in_sets_end_user_logging_identity() -> None:
assert get_identity_context() == ("tenant-id", "end-user-id", "browser")
def test_on_user_logged_in_does_not_break_auth_when_identity_is_unavailable() -> None:
def test_on_user_logged_in_does_not_break_auth_when_identity_is_unavailable(caplog: pytest.LogCaptureFixture) -> None:
account = mock.Mock(spec=ext_login.Account)
type(account).current_tenant_id = mock.PropertyMock(side_effect=RuntimeError("unavailable"))
account.id = "account-id"
clear_request_context()
with mock.patch.object(ext_login.logger, "exception") as logger_exception:
with caplog.at_level("ERROR", logger=ext_login.logger.name):
ext_login.on_user_logged_in(None, account)
assert get_identity_context() == ("", "", "")
logger_exception.assert_called_once_with("Failed to set logging identity context")
assert "Failed to set logging identity context" in caplog.text
def test_on_user_logged_in_logs_unsupported_user_type() -> None:
def test_on_user_logged_in_logs_unsupported_user_type(caplog: pytest.LogCaptureFixture) -> None:
unsupported_user = cast(ext_login.LoginUser, object())
clear_request_context()
with mock.patch.object(ext_login.logger, "exception") as logger_exception:
with caplog.at_level("ERROR", logger=ext_login.logger.name):
ext_login.on_user_logged_in(None, unsupported_user)
assert get_identity_context() == ("", "", "")
logger_exception.assert_called_once_with("Failed to set logging identity context")
assert "Failed to set logging identity context" in caplog.text