.

Author: alexander-alderman-webb

tests/integrations/django/myapp/urls.py

@@ -61,6 +61,16 @@ def path(path, *args, **kwargs):
     path("template-test4", views.template_test4, name="template_test4"),
     path("postgres-select", views.postgres_select, name="postgres_select"),
     path("postgres-select-slow", views.postgres_select_orm, name="postgres_select_orm"),
+    path(
+        "postgres-select-no-autocommit",
+        views.postgres_select_orm_no_autocommit,
+        name="postgres_select_orm_no_autocommit",
+    ),
+    path(
+        "postgres-select-atomic",
+        views.postgres_select_orm_atomic,
+        name="postgres_select_orm_atomic",
+    ),
     path(
         "postgres-select-slow-from-supplement",
         helper_views.postgres_select_orm,

tests/integrations/django/myapp/views.py

@@ -2,6 +2,7 @@
 import json
 import threading
 
+from django.db import transaction
 from django.contrib.auth import login
 from django.contrib.auth.models import User
 from django.core.exceptions import PermissionDenied
@@ -246,6 +247,21 @@ def postgres_select_orm(request, *args, **kwargs):
     return HttpResponse("ok {}".format(user))
 
 
+@csrf_exempt
+def postgres_select_orm_no_autocommit(request, *args, **kwargs):
+    transaction.set_autocommit(False)
+    user = User.objects.using("postgres").all().first()
+    transaction.commit()
+    return HttpResponse("ok {}".format(user))
+
+
+@csrf_exempt
+def postgres_select_orm_atomic(request, *args, **kwargs):
+    with transaction.atomic():
+        user = User.objects.using("postgres").all().first()
+    return HttpResponse("ok {}".format(user))
+
+
 @csrf_exempt
 def permission_denied_exc(*args, **kwargs):
     raise PermissionDenied("bye")