Merge pull request #90 from Mogost/deferred-loading

Fix deferred loading

Author: millerdev

django_cte/cte.py

@@ -109,6 +109,7 @@ def queryset(self):
         query = CTEQuery(cte_query.model)
         query.join(BaseTable(self.name, None))
         query.default_cols = cte_query.default_cols
+        query.deferred_loading = cte_query.deferred_loading
         if cte_query.annotations:
             for alias, value in cte_query.annotations.items():
                 col = CTEColumnRef(alias, self.name, value.output_field)

tests/test_manager.py

@@ -104,3 +104,12 @@ def test_cte_queryset_with_custom_queryset(self):
             ('venus', 22, 'sun'),
             ('venus', 23, 'sun'),
         ])
+
+    def test_cte_queryset_with_deferred_loading(self):
+        cte = With(
+            OrderCustomManagerNQuery.objects.order_by("id").only("id")[:1]
+        )
+        orders = cte.queryset().with_cte(cte)
+        print(orders.query)
+
+        self.assertEqual([x.id for x in orders], [1])