Ticket #22414: patch

django/test/testcases.py

@@ -1116 +1116 @@
             self.httpd.set_app(handler)
             self.is_ready.set()
             self.httpd.serve_forever()
+            for alias in connections:
+                connections[alias].close()
         except Exception as e:
             self.error = e
             self.is_ready.set()

tests/servers/tests.py

@@ -185 +185 @@
             ['jane', 'robert', 'emily'],
             lambda b: b.name
         )
+
+
+class LiveServerPersistentDatabase(LiveServerBase):
+
+    def test_connection_closing(self):
+        self.connection_closed = False
+
+        from django.db import connections, DEFAULT_DB_ALIAS
+        connection = connections[DEFAULT_DB_ALIAS]
+
+        # Mark connection as persistent
+        connection.close_at = None
+
+        # Monkey-patch to check when is it closed
+        original_close = connection.close
+        def wrapped_close():
+            self.connection_closed = True
+            return original_close()
+        connection.close = wrapped_close
+
+        # Force the LiveServerThread to open a connection
+        self.urlopen('/example_view/')
+
+        # Stop the server
+        self.server_thread.terminate()
+        self.server_thread.join()
+
+        # The connection must have been closed by the server
+        self.assertTrue(self.connection_closed)