Added missing tests for transforms usage with subquery for PostgreSQL fields

3 files changed, 20 insertions, 0 deletions

diff --git a/tests/postgres_tests/test_array.py b/tests/postgres_tests/test_array.py
index a893a95406..d152f76b3d 100644
--- a/tests/postgres_tests/test_array.py
+++ b/tests/postgres_tests/test_array.py
@@ -228,6 +228,14 @@ class TestQuerying(PostgreSQLTestCase):
             [instance]
         )
 
+    def test_usage_in_subquery(self):
+        self.assertSequenceEqual(
+            NullableIntegerArrayModel.objects.filter(
+                id__in=NullableIntegerArrayModel.objects.filter(field__len=3)
+            ),
+            [self.objs[3]]
+        )
+
 
 class TestChecks(PostgreSQLTestCase):
 
diff --git a/tests/postgres_tests/test_hstore.py b/tests/postgres_tests/test_hstore.py
index 398f444167..34229bb15c 100644
--- a/tests/postgres_tests/test_hstore.py
+++ b/tests/postgres_tests/test_hstore.py
@@ -132,6 +132,12 @@ class TestQuerying(PostgreSQLTestCase):
             self.objs[:2]
         )
 
+    def test_usage_in_subquery(self):
+        self.assertSequenceEqual(
+            HStoreModel.objects.filter(id__in=HStoreModel.objects.filter(field__a='b')),
+            self.objs[:2]
+        )
+
 
 class TestSerialization(PostgreSQLTestCase):
     test_data = '[{"fields": {"field": "{\\"a\\": \\"b\\"}"}, "model": "postgres_tests.hstoremodel", "pk": null}]'
diff --git a/tests/postgres_tests/test_json.py b/tests/postgres_tests/test_json.py
index 9637d47ec6..b4f0dd4ecf 100644
--- a/tests/postgres_tests/test_json.py
+++ b/tests/postgres_tests/test_json.py
@@ -204,6 +204,12 @@ class TestQuerying(TestCase):
             [self.objs[7], self.objs[8]]
         )
 
+    def test_usage_in_subquery(self):
+        self.assertSequenceEqual(
+            JSONModel.objects.filter(id__in=JSONModel.objects.filter(field__c=1)),
+            self.objs[7:9]
+        )
+
 
 @skipUnlessPG94
 class TestSerialization(TestCase):