async def test_delete_await_across_transaction_success(self):
await Tournament.create(name='Test1')
query = Tournament.filter(id=1).delete()
async with in_transaction():
await query
self.assertEqual(await Tournament.all(), [])
async def test_update_await_across_transaction_success(self):
await Tournament.create(name='Test1')
query = Tournament.filter(id=1).update(name='Test2')
async with in_transaction():
await query
self.assertEqual(await Tournament.all().values('id', 'name'), [{'id': 1, 'name': 'Test2'}])
async def test_delete_await_across_transaction_fail(self):
await Tournament.create(name='Test1')
query = Tournament.filter(id=1).delete()
try:
async with in_transaction():
await query
raise KeyError('moo')
except KeyError:
pass
self.assertEqual(await Tournament.all().values('id', 'name'), [{'id': 1, 'name': 'Test1'}])
async def test_update_await_across_transaction_success(self):
await Tournament.create(name="Test1")
query = Tournament.filter(id=1).update(name="Test2")
async with in_transaction():
await query
self.assertEqual(await Tournament.all().values("id", "name"),
[{
"id": 1,
"name": "Test2"
}])
async def test_delete_await_across_transaction_fail(self):
await Tournament.create(name="Test1")
query = Tournament.filter(id=1).delete()
try:
async with in_transaction():
await query
raise KeyError("moo")
except KeyError:
pass
self.assertEqual(await Tournament.all().values("id", "name"),
[{
"id": 1,
"name": "Test1"
}])
