chore: update gitignore; feat: 新增customer_agent业务模块与api路由
This commit is contained in:
@@ -0,0 +1,24 @@
|
||||
import unittest
|
||||
from unittest.mock import AsyncMock
|
||||
|
||||
from service.customer_agent.audit import write_anonymous_sensitive_audit
|
||||
|
||||
|
||||
class AuditTests(unittest.IsolatedAsyncioTestCase):
|
||||
async def test_writes_sensitive_anonymous_action_with_trace_id(self):
|
||||
db = AsyncMock()
|
||||
|
||||
await write_anonymous_sensitive_audit(
|
||||
db, session_id="s1", trace_id="trace-1"
|
||||
)
|
||||
|
||||
db.execute.assert_awaited_once()
|
||||
db.commit.assert_awaited_once()
|
||||
params = db.execute.await_args.args[1]
|
||||
self.assertEqual(params["action"], "anon_sensitive_input")
|
||||
self.assertEqual(params["trace_id"], "trace-1")
|
||||
self.assertEqual(params["target"], "s1")
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
unittest.main()
|
||||
Reference in New Issue
Block a user