encode/django-rest-framework
1
1
Fork 0
mirror of https://github.com/encode/django-rest-framework.git synced 2025-02-19 21:11:04 +03:00
Code Issues Packages Projects Releases Wiki Activity

converted asserts of atomic requests test to pytest

Browse Source
This commit is contained in:
Asif Saifuddin Auvi 2016-11-30 16:17:30 +06:00
parent e03d88ced7
commit 504f4b44c6
1 changed files with 6 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
tests/test_atomic_requests.py
View File

@ -67,8 +67,8 @@ class DBTransactionTests(TestCase):
with self.assertNumQueries(1): with self.assertNumQueries(1):
response = self.view(request) response = self.view(request)
self.assertFalse(transaction.get_rollback()) assert not transaction.get_rollback()
self.assertEqual(response.status_code, status.HTTP_200_OK) assert response.status_code == status.HTTP_200_OK
assert BasicModel.objects.count() == 1 assert BasicModel.objects.count() == 1
@ -98,7 +98,7 @@ class DBTransactionErrorTests(TestCase):
# 3 - release savepoint # 3 - release savepoint
with transaction.atomic(): with transaction.atomic():
self.assertRaises(Exception, self.view, request) self.assertRaises(Exception, self.view, request)
self.assertFalse(transaction.get_rollback()) assert not transaction.get_rollback()
assert BasicModel.objects.count() == 1 assert BasicModel.objects.count() == 1
@ -128,9 +128,8 @@ class DBTransactionAPIExceptionTests(TestCase):
# 4 - release savepoint (django>=1.8 only) # 4 - release savepoint (django>=1.8 only)
with transaction.atomic(): with transaction.atomic():
response = self.view(request) response = self.view(request)
self.assertTrue(transaction.get_rollback()) assert transaction.get_rollback()
self.assertEqual(response.status_code, assert response.status_code == status.HTTP_500_INTERNAL_SERVER_ERROR
status.HTTP_500_INTERNAL_SERVER_ERROR)
assert BasicModel.objects.count() == 0 assert BasicModel.objects.count() == 0
@ -151,5 +150,4 @@ class NonAtomicDBTransactionAPIExceptionTests(TransactionTestCase):
# without checking connection.in_atomic_block view raises 500 # without checking connection.in_atomic_block view raises 500
# due attempt to rollback without transaction # due attempt to rollback without transaction
self.assertEqual(response.status_code, assert response.status_code == status.HTTP_404_NOT_FOUND
status.HTTP_404_NOT_FOUND)