async def test_select_await_across_transaction_success(self):
query = Tournament.all().values('id', 'name')
async with in_transaction():
await Tournament.create(name='Test1')
result = await query
self.assertEqual(result, [{'id': 1, 'name': 'Test1'}])
self.assertEqual(await Tournament.all().values('id', 'name'), [{'id': 1, 'name': 'Test1'}])
async def test_select_await_across_transaction_success(self):
query = Tournament.all().values("id", "name")
async with in_transaction():
await Tournament.create(name="Test1")
result = await query
self.assertEqual(result, [{"id": 1, "name": "Test1"}])
self.assertEqual(await Tournament.all().values("id", "name"),
[{
"id": 1,
"name": "Test1"
}])
async def test_select_await_across_transaction_fail(self):
query = Tournament.all().values('id', 'name')
try:
async with in_transaction():
await Tournament.create(name='Test1')
result = await query
raise KeyError('moo')
except KeyError:
pass
self.assertEqual(result, [{'id': 1, 'name': 'Test1'}])
self.assertEqual(await Tournament.all(), [])
async def test_select_await_across_transaction_fail(self):
query = Tournament.all().values("id", "name")
try:
async with in_transaction():
await Tournament.create(name="Test1")
result = await query
raise KeyError("moo")
except KeyError:
pass
self.assertEqual(result, [{"id": 1, "name": "Test1"}])
self.assertEqual(await Tournament.all(), [])
