wutta/wutta-continuum
3
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

feat: add TransactionMetaPlugin to save comments when applicable

Browse source
This commit is contained in:
Lance Edgar 2025-12-18 23:02:08 -06:00
parent 09e4ef6a1e
commit b19f565aa1
3 changed files with 87 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

19
tests/test_conf.py
View file

@ -2,7 +2,7 @@
import socket
from unittest.mock import patch
from unittest.mock import patch, Mock
from wuttjamaican.testing import ConfigTestCase, DataTestCase
@ -71,3 +71,20 @@ class TestWuttaContinuumPlugin(DataTestCase):
plugin.transaction_args(None, self.session),
{"remote_addr": "127.0.0.1", "user_id": "some-random-uuid"},
)
def test_before_flush(self):
plugin = self.make_plugin()
meta = {}
txn = Mock(meta=meta)
uow = Mock(current_transaction=txn)
# no comment in session or transaction
plugin.before_flush(uow, self.session)
self.assertNotIn("comment", meta)
# transaction comment matches session
self.session.info["continuum_comment"] = "whaddyaknow"
plugin.before_flush(uow, self.session)
self.assertIn("comment", meta)
self.assertEqual(meta["comment"], "whaddyaknow")