def test_post_commit_funcs_error(self):
def func():
raise ValueError()
with self.assertRaises(RuntimeError):
with TRN:
TRN.add("SELECT 42")
TRN.add_post_commit_func(func)
def test_post_commit_funcs(self):
fd, fp = mkstemp()
close(fd)
self._files_to_remove.append(fp)
def func(fp):
with open(fp, "w") as f:
f.write("\n")
with TRN:
TRN.add("SELECT 42")
TRN.add_post_commit_func(func, fp)
self.assertTrue(exists(fp))
