mark.tian/ai-talk-callback
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

保存请求失败返回500

Browse Source 
调整表结构
This commit is contained in:
mark.tian
2025-12-04 08:14:21 +08:00
parent f1f694bd63
commit 6ac998ebdf
5 changed files with 158 additions and 79 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
test_callback.py
View File

@@ -306,7 +306,7 @@ class TestAITalkCallback:
sample_callback_request
):
"""测试remote_address和server_ip字段是否正确保存"""
from app.database import CallbackLog
from app.database import CallbackFailureLog
# 设置模拟对象
mock_db = AsyncMock()
@@ -333,9 +333,9 @@ class TestAITalkCallback:
assert mock_db.add.called
assert mock_db.commit.called
# 获取传递给add的CallbackLog对象
# 获取传递给add的CallbackFailureLog对象
call_args = mock_db.add.call_args[0][0]
assert isinstance(call_args, CallbackLog)
assert isinstance(call_args, CallbackFailureLog)
# 验证新字段存在（可能为None，因为测试环境）
assert hasattr(call_args, 'remote_address')
@@ -356,7 +356,7 @@ class TestAITalkCallback:
sample_callback_request
):
"""测试JSON格式日志记录功能"""
from app.database import CallbackLog
from app.database import CallbackFailureLog
# 设置模拟对象
mock_db = AsyncMock()
@@ -382,9 +382,9 @@ class TestAITalkCallback:
# 验证log_callback_request被调用
assert mock_db.add.called
# 获取传递给add的CallbackLog对象
# 获取传递给add的CallbackFailureLog对象
call_args = mock_db.add.call_args[0][0]
assert isinstance(call_args, CallbackLog)
assert isinstance(call_args, CallbackFailureLog)
# 验证数据结构适合JSON序列化
import json