fixes in unit tests

This commit is contained in:
Bastien Gérard 2016-07-02 22:51:54 +02:00
parent e394c8f0f2
commit df9ed835ca
5 changed files with 15 additions and 7 deletions

View File

@ -13,7 +13,7 @@ from datetime import datetime
from bson import DBRef, ObjectId from bson import DBRef, ObjectId
from tests import fixtures from tests import fixtures
from tests.fixtures import (PickleEmbedded, PickleTest, PickleSignalsTest, from tests.fixtures import (PickleEmbedded, PickleTest, PickleSignalsTest,
PickleDyanmicEmbedded, PickleDynamicTest) PickleDynamicEmbedded, PickleDynamicTest)
from mongoengine import * from mongoengine import *
from mongoengine.errors import (NotRegistered, InvalidDocumentError, from mongoengine.errors import (NotRegistered, InvalidDocumentError,
@ -2317,7 +2317,7 @@ class InstanceTest(unittest.TestCase):
pickle_doc = PickleDynamicTest( pickle_doc = PickleDynamicTest(
name="test", number=1, string="One", lists=['1', '2']) name="test", number=1, string="One", lists=['1', '2'])
pickle_doc.embedded = PickleDyanmicEmbedded(foo="Bar") pickle_doc.embedded = PickleDynamicEmbedded(foo="Bar")
pickled_doc = pickle.dumps(pickle_doc) # make sure pickling works even before the doc is saved pickled_doc = pickle.dumps(pickle_doc) # make sure pickling works even before the doc is saved
pickle_doc.save() pickle_doc.save()

View File

@ -26,7 +26,7 @@ class NewDocumentPickleTest(Document):
new_field = StringField() new_field = StringField()
class PickleDyanmicEmbedded(DynamicEmbeddedDocument): class PickleDynamicEmbedded(DynamicEmbeddedDocument):
date = DateTimeField(default=datetime.now) date = DateTimeField(default=datetime.now)

View File

@ -1,8 +1,11 @@
import unittest
from convert_to_new_inheritance_model import * from convert_to_new_inheritance_model import *
from decimalfield_as_float import * from decimalfield_as_float import *
from refrencefield_dbref_to_object_id import * from referencefield_dbref_to_object_id import *
from turn_off_inheritance import * from turn_off_inheritance import *
from uuidfield_to_binary import * from uuidfield_to_binary import *
if __name__ == '__main__': if __name__ == '__main__':
unittest.main() unittest.main()

View File

@ -12,9 +12,13 @@ from mongoengine.context_managers import query_counter
class FieldTest(unittest.TestCase): class FieldTest(unittest.TestCase):
def setUp(self): @classmethod
connect(db='mongoenginetest') def setUpClass(cls):
self.db = get_db() cls.db = connect(db='mongoenginetest')
@classmethod
def tearDownClass(cls):
cls.db.drop_database('mongoenginetest')
def test_list_item_dereference(self): def test_list_item_dereference(self):
"""Ensure that DBRef items in ListFields are dereferenced. """Ensure that DBRef items in ListFields are dereferenced.
@ -304,6 +308,7 @@ class FieldTest(unittest.TestCase):
User.drop_collection() User.drop_collection()
Post.drop_collection() Post.drop_collection()
SimpleList.drop_collection()
u1 = User.objects.create(name='u1') u1 = User.objects.create(name='u1')
u2 = User.objects.create(name='u2') u2 = User.objects.create(name='u2')