tests: fix usage of transaction.non_atomic_requests (#6043)

blueyed · Ryan P Kilby · commit a44cb679888e · 2018-06-21T14:44:58.000-04:00
diff --git a/tests/test_atomic_requests.py b/tests/test_atomic_requests.py
@@ -6,7 +6,6 @@
 from django.db import connection, connections, transaction
 from django.http import Http404
 from django.test import TestCase, TransactionTestCase, override_settings
-from django.utils.decorators import method_decorator
 
 from rest_framework import status
 from rest_framework.exceptions import APIException
@@ -37,7 +36,7 @@ def post(self, request, *args, **kwargs):
 
 
 class NonAtomicAPIExceptionView(APIView):
-    @method_decorator(transaction.non_atomic_requests)
+    @transaction.non_atomic_requests
     def dispatch(self, *args, **kwargs):
         return super(NonAtomicAPIExceptionView, self).dispatch(*args, **kwargs)
 
