25 lines
739 B
Python
25 lines
739 B
Python
|
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()
|