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