diff --git a/tests/fields/fields.py b/tests/fields/fields.py index 1079b235..d7edea37 100644 --- a/tests/fields/fields.py +++ b/tests/fields/fields.py @@ -1974,14 +1974,14 @@ class FieldTest(unittest.TestCase): def test_recursive_embedding(self): """Ensure that EmbeddedDocumentFields can contain their own documents. """ - class Tree(Document): - name = StringField() - children = ListField(EmbeddedDocumentField('TreeNode')) - class TreeNode(EmbeddedDocument): name = StringField() children = ListField(EmbeddedDocumentField('self')) + class Tree(Document): + name = StringField() + children = ListField(EmbeddedDocumentField('TreeNode')) + Tree.drop_collection() tree = Tree(name="Tree") diff --git a/tests/queryset/queryset.py b/tests/queryset/queryset.py index 40d010c6..703728fe 100644 --- a/tests/queryset/queryset.py +++ b/tests/queryset/queryset.py @@ -512,16 +512,16 @@ class QuerySetTest(unittest.TestCase): def test_updates_can_have_match_operators(self): - class Post(Document): - title = StringField(required=True) - tags = ListField(StringField()) - comments = ListField(EmbeddedDocumentField("Comment")) - class Comment(EmbeddedDocument): content = StringField() name = StringField(max_length=120) vote = IntField() + class Post(Document): + title = StringField(required=True) + tags = ListField(StringField()) + comments = ListField(EmbeddedDocumentField("Comment")) + Post.drop_collection() comm1 = Comment(content="very funny indeed", name="John S", vote=1) diff --git a/tests/test_dereference.py b/tests/test_dereference.py index c37ada59..85588558 100644 --- a/tests/test_dereference.py +++ b/tests/test_dereference.py @@ -318,6 +318,10 @@ class FieldTest(unittest.TestCase): def test_circular_reference(self): """Ensure you can handle circular references """ + class Relation(EmbeddedDocument): + name = StringField() + person = ReferenceField('Person') + class Person(Document): name = StringField() relations = ListField(EmbeddedDocumentField('Relation')) @@ -325,10 +329,6 @@ class FieldTest(unittest.TestCase): def __repr__(self): return "" % self.name - class Relation(EmbeddedDocument): - name = StringField() - person = ReferenceField('Person') - Person.drop_collection() mother = Person(name="Mother") daughter = Person(name="Daughter") @@ -1220,14 +1220,15 @@ class FieldTest(unittest.TestCase): self.assertEqual(page.tags[0], page.posts[0].tags[0]) def test_select_related_follows_embedded_referencefields(self): - class Playlist(Document): - items = ListField(EmbeddedDocumentField("PlaylistItem")) + + class Song(Document): + title = StringField() class PlaylistItem(EmbeddedDocument): song = ReferenceField("Song") - class Song(Document): - title = StringField() + class Playlist(Document): + items = ListField(EmbeddedDocumentField("PlaylistItem")) Playlist.drop_collection() Song.drop_collection()