pub/mongoengine
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Added test for overriding objects

Browse Source
This commit is contained in:
Harry Marr 2010-08-30 13:54:20 +01:00
parent 2635e41f69
commit 4fb6fcabef
1 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
tests/queryset.py
View File

@ -992,10 +992,15 @@ class QuerySetTest(unittest.TestCase):
""" """
class BlogPost(Document): class BlogPost(Document):
tags = ListField(StringField()) tags = ListField(StringField())
deleted = BooleanField(default=False)
@queryset_manager
def objects(doc_cls, queryset):
return queryset(deleted=False)
@queryset_manager @queryset_manager
def music_posts(doc_cls, queryset): def music_posts(doc_cls, queryset):
return queryset(tags='music') return queryset(tags='music', deleted=False)
BlogPost.drop_collection() BlogPost.drop_collection()
@ -1005,6 +1010,8 @@ class QuerySetTest(unittest.TestCase):
post2.save() post2.save()
post3 = BlogPost(tags=['film', 'actors']) post3 = BlogPost(tags=['film', 'actors'])
post3.save() post3.save()
post4 = BlogPost(tags=['film', 'actors'], deleted=True)
post4.save()
self.assertEqual([p.id for p in BlogPost.objects], self.assertEqual([p.id for p in BlogPost.objects],
[post1.id, post2.id, post3.id]) [post1.id, post2.id, post3.id])