Ticket #14930: extra_order_by_values_list-with_test.diff

django/db/models/query.py

@@ -1021 +1021 @@
             self.aggregate_names = None
 
         self.query.select = []
-        if self.extra_names is not None:
-            self.query.set_extra_mask(self.extra_names)
+        extra_mask_names = self.extra_names + list(self.query.extra_order_by or [])
+        if extra_mask_names is not None:
+            self.query.set_extra_mask(extra_mask_names)
         self.query.add_fields(self.field_names, True)
         if self.aggregate_names is not None:
             self.query.set_aggregate_mask(self.aggregate_names)
@@ -1097 +1098 @@
     def iterator(self):
         if self.flat and len(self._fields) == 1:
             for row in self.query.get_compiler(self.db).results_iter():
-                yield row[0]
+                yield row[len(self.query.extra_select_mask)]
         elif not self.query.extra_select and not self.query.aggregate_select:
             for row in self.query.get_compiler(self.db).results_iter():
                 yield tuple(row)

tests/regressiontests/queries/tests.py

@@ -1700 +1700 @@
 
 
 class ValuesQuerysetTests(BaseQuerysetTest):
-    def test_flat_values_lits(self):
+    def setUp(self):
         Number.objects.create(num=72)
+
+    def test_flat_values_lits(self):
         qs = Number.objects.values_list("num")
         qs = qs.values_list("num", flat=True)
         self.assertValueQuerysetEqual(
             qs, [72]
         )
 
+    def test_extra_values(self):
+        qs = Number.objects.extra(select={'value_plus_one': 'num+1'}).order_by('value_plus_one')
+        qs = qs.values('num')
+        identity = lambda x:x
+        self.assertQuerysetEqual(qs, [{'num': 72}], identity)
+
+    def test_extra_values_list(self):
+        qs = Number.objects.extra(select={'value_plus_one': 'num+1'}).order_by('value_plus_one')
+        qs = qs.values_list('num')
+        identity = lambda x:x
+        self.assertQuerysetEqual(qs, [(72,)], identity)
+
 
 class WeirdQuerysetSlicingTests(BaseQuerysetTest):
     def setUp(self):