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(), [])