async def test_insert_await_across_transaction_success(self):
tournament = Tournament(name="Test")
query = tournament.save() # pylint: disable=E1111
async with in_transaction():
await query
self.assertEqual(await Tournament.all(), [tournament])
async def test_insert_await_across_transaction_fail(self):
tournament = Tournament(name="Test")
query = tournament.save() # pylint: disable=E1111
try:
async with in_transaction():
await query
raise KeyError("moo")
except KeyError:
pass
self.assertEqual(await Tournament.all(), [])
