Fixed #24894 -- Added contrib.postgres.functions.TransactionNow

1 files changed, 28 insertions, 0 deletions

diff --git a/tests/postgres_tests/test_functions.py b/tests/postgres_tests/test_functions.py
new file mode 100644
index 0000000000..620b561325
--- /dev/null
+++ b/tests/postgres_tests/test_functions.py
@@ -0,0 +1,28 @@
+from datetime import datetime
+from time import sleep
+
+from django.contrib.postgres.functions import TransactionNow
+
+from . import PostgreSQLTestCase
+from .models import NowTestModel
+
+
+class TestTransactionNow(PostgreSQLTestCase):
+
+    def test_transaction_now(self):
+        """
+        The test case puts everything under a transaction, so two models
+        updated with a short gap should have the same time.
+        """
+        m1 = NowTestModel.objects.create()
+        m2 = NowTestModel.objects.create()
+
+        NowTestModel.objects.filter(id=m1.id).update(when=TransactionNow())
+        sleep(0.1)
+        NowTestModel.objects.filter(id=m2.id).update(when=TransactionNow())
+
+        m1.refresh_from_db()
+        m2.refresh_from_db()
+
+        self.assertIsInstance(m1.when, datetime)
+        self.assertEqual(m1.when, m2.when)