45 lines
1.3 KiB
Python
45 lines
1.3 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 sys
|
|
sys.path[0:0] = [""]
|
|
import unittest
|
|
import warnings
|
|
|
|
from mongoengine import *
|
|
|
|
|
|
__all__ = ('AllWarnings', )
|
|
|
|
|
|
class AllWarnings(unittest.TestCase):
|
|
|
|
def setUp(self):
|
|
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_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())
|