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()