89 lines
2.6 KiB
Python
89 lines
2.6 KiB
Python
"""
|
|
This test has been put into a module. This is because it tests warnings that
|
|
only get triggered on first hit. This way we can ensure its imported into the
|
|
top level and called first by the test suite.
|
|
"""
|
|
|
|
import unittest
|
|
import warnings
|
|
|
|
from mongoengine import *
|
|
|
|
|
|
__all__ = ('AllWarnings', )
|
|
|
|
|
|
class AllWarnings(unittest.TestCase):
|
|
|
|
def setUp(self):
|
|
conn = connect(db='mongoenginetest')
|
|
self.warning_list = []
|
|
self.showwarning_default = warnings.showwarning
|
|
warnings.showwarning = self.append_to_warning_list
|
|
|
|
def append_to_warning_list(self, message, category, *args):
|
|
self.warning_list.append({"message": message,
|
|
"category": category})
|
|
|
|
def tearDown(self):
|
|
# restore default handling of warnings
|
|
warnings.showwarning = self.showwarning_default
|
|
|
|
def test_dbref_reference_field_future_warning(self):
|
|
|
|
class Person(Document):
|
|
name = StringField()
|
|
parent = ReferenceField('self')
|
|
|
|
Person.drop_collection()
|
|
|
|
p1 = Person()
|
|
p1.parent = None
|
|
p1.save()
|
|
|
|
p2 = Person(name="Wilson Jr")
|
|
p2.parent = p1
|
|
p2.save(cascade=False)
|
|
|
|
self.assertTrue(len(self.warning_list) > 0)
|
|
warning = self.warning_list[0]
|
|
self.assertEqual(FutureWarning, warning["category"])
|
|
self.assertTrue("ReferenceFields will default to using ObjectId"
|
|
in str(warning["message"]))
|
|
|
|
def test_document_save_cascade_future_warning(self):
|
|
|
|
class Person(Document):
|
|
name = StringField()
|
|
parent = ReferenceField('self')
|
|
|
|
Person.drop_collection()
|
|
|
|
p1 = Person(name="Wilson Snr")
|
|
p1.parent = None
|
|
p1.save()
|
|
|
|
p2 = Person(name="Wilson Jr")
|
|
p2.parent = p1
|
|
p2.parent.name = "Poppa Wilson"
|
|
p2.save()
|
|
|
|
self.assertEqual(len(self.warning_list), 1)
|
|
warning = self.warning_list[0]
|
|
self.assertEqual(FutureWarning, warning["category"])
|
|
self.assertTrue("Cascading saves will default to off in 0.8"
|
|
in str(warning["message"]))
|
|
|
|
def test_document_collection_syntax_warning(self):
|
|
|
|
class NonAbstractBase(Document):
|
|
meta = {'allow_inheritance': True}
|
|
|
|
class InheritedDocumentFailTest(NonAbstractBase):
|
|
meta = {'collection': 'fail'}
|
|
|
|
warning = self.warning_list[0]
|
|
self.assertEqual(SyntaxWarning, warning["category"])
|
|
self.assertEqual('non_abstract_base',
|
|
InheritedDocumentFailTest._get_collection_name())
|