fixed tests that were using undefined model fields

This commit is contained in:
Rik
2014-06-27 14:43:56 +02:00
committed by Wilson Júnior
parent 06064decd2
commit 67bf6afc89
4 changed files with 10 additions and 3 deletions

View File

@@ -1492,6 +1492,7 @@ class QuerySetTest(unittest.TestCase):
"""Ensure that atomic updates work properly.
"""
class BlogPost(Document):
name = StringField()
title = StringField()
hits = IntField()
tags = ListField(StringField())
@@ -2878,12 +2879,14 @@ class QuerySetTest(unittest.TestCase):
self.assertEqual(authors, [mark_twain, john_tolkien])
def test_distinct_ListField_ReferenceField(self):
class Foo(Document):
bar_lst = ListField(ReferenceField('Bar'))
class Bar(Document):
text = StringField()
class Foo(Document):
bar = ReferenceField('Bar')
bar_lst = ListField(ReferenceField('Bar'))
Bar.drop_collection()
Foo.drop_collection()