Merge pull request #393 from wpjunior/queryset_select

select method in Queryset

tests/queryset.py

@@ -2909,6 +2909,160 @@ class QueryFieldListTest(unittest.TestCase):
 
         ak = list(Bar.objects(foo__match={'shape': "square", "color": "purple"}))
         self.assertEqual([b1], ak)
+    
+    def test_values_list(self):
+        class TestDoc(Document):
+            x = IntField()
+            y = BooleanField()
+
+        TestDoc.drop_collection()
+
+        TestDoc(x=10, y=True).save()
+        TestDoc(x=20, y=False).save()
+        TestDoc(x=30, y=True).save()
+
+        plist = list(TestDoc.objects.values_list('x', 'y'))
+
+        self.assertEqual(len(plist), 3)
+        self.assertEqual(plist[0], [10, True])
+        self.assertEqual(plist[1], [20, False])
+        self.assertEqual(plist[2], [30, True])
+
+        class UserDoc(Document):
+            name = StringField()
+            age = IntField()
+
+        UserDoc.drop_collection()
+
+        UserDoc(name="Wilson Jr", age=19).save()
+        UserDoc(name="Wilson", age=43).save()
+        UserDoc(name="Eliana", age=37).save()
+        UserDoc(name="Tayza", age=15).save()
+
+        ulist = list(UserDoc.objects.values_list('name', 'age'))
+
+        self.assertEqual(ulist, [
+                [u'Wilson Jr', 19],
+                [u'Wilson', 43],
+                [u'Eliana', 37],
+                [u'Tayza', 15]])
+
+        ulist = list(UserDoc.objects.order_by('age').values_list('name'))
+
+        self.assertEqual(ulist, [
+                [u'Tayza'],
+                [u'Wilson Jr'],
+                [u'Eliana'],
+                [u'Wilson']])
+
+    def test_values_list_embedded(self):
+        class Profile(EmbeddedDocument):
+            name = StringField()
+            age = IntField()
+
+        class Locale(EmbeddedDocument):
+            city = StringField()
+            country = StringField()
+
+        class Person(Document):
+            profile = EmbeddedDocumentField(Profile)
+            locale = EmbeddedDocumentField(Locale)
+
+        Person.drop_collection()
+
+        Person(profile=Profile(name="Wilson Jr", age=19),
+               locale=Locale(city="Corumba-GO", country="Brazil")).save()
+
+        Person(profile=Profile(name="Gabriel Falcao", age=23),
+               locale=Locale(city="New York", country="USA")).save()
+
+        Person(profile=Profile(name="Lincoln de souza", age=28),
+               locale=Locale(city="Belo Horizonte", country="Brazil")).save()
+
+        Person(profile=Profile(name="Walter cruz", age=30),
+               locale=Locale(city="Brasilia", country="Brazil")).save()
+
+        self.assertEqual(
+            list(Person.objects.order_by('profile.age').values_list('profile.name')),
+            [[u'Wilson Jr'], [u'Gabriel Falcao'],
+             [u'Lincoln de souza'], [u'Walter cruz']])
+
+        ulist = list(Person.objects.order_by('locale.city')
+                     .values_list('profile.name', 'profile.age', 'locale.city'))
+        self.assertEqual(ulist,
+                         [[u'Lincoln de souza', 28, u'Belo Horizonte'],
+                          [u'Walter cruz', 30, u'Brasilia'],
+                          [u'Wilson Jr', 19, u'Corumba-GO'],
+                          [u'Gabriel Falcao', 23, u'New York']])
+
+    def test_values_list_decimal(self):
+        from decimal import Decimal
+        class Person(Document):
+            name = StringField()
+            rating = DecimalField()
+            
+        Person.drop_collection()
+        Person(name="Wilson Jr", rating=Decimal('1.0')).save()
+
+        ulist = list(Person.objects.values_list('name', 'rating'))
+        self.assertEqual(ulist, [[u'Wilson Jr', Decimal('1.0')]])
+
+
+    def test_values_list_reference_field(self):
+        class State(Document):
+            name = StringField()
+
+        class Person(Document):
+            name = StringField()
+            state = ReferenceField(State)
+
+        State.drop_collection()
+        Person.drop_collection()
+
+        s1 = State(name="Goias")
+        s1.save()
+
+        Person(name="Wilson JR", state=s1).save()
+
+        plist = list(Person.objects.values_list('name', 'state'))
+        self.assertEqual(plist, [[u'Wilson JR', s1]])
+
+    def test_values_list_generic_reference_field(self):
+        class State(Document):
+            name = StringField()
+
+        class Person(Document):
+            name = StringField()
+            state = GenericReferenceField()
+
+        State.drop_collection()
+        Person.drop_collection()
+
+        s1 = State(name="Goias")
+        s1.save()
+
+        Person(name="Wilson JR", state=s1).save()
+
+        plist = list(Person.objects.values_list('name', 'state'))
+        self.assertEqual(plist, [[u'Wilson JR', s1]])
+
+    def test_values_list_db_field(self):
+        class TestDoc(Document):
+            x = IntField(db_field="y")
+            y = BooleanField(db_field="x")
+
+        TestDoc.drop_collection()
+
+        TestDoc(x=10, y=True).save()
+        TestDoc(x=20, y=False).save()
+        TestDoc(x=30, y=True).save()
+
+        plist = list(TestDoc.objects.values_list('x', 'y'))
+
+        self.assertEqual(len(plist), 3)
+        self.assertEqual(plist[0], [10, True])
+        self.assertEqual(plist[1], [20, False])
+        self.assertEqual(plist[2], [30, True])
 
 if __name__ == '__main__':
     unittest.main()