diff options
Diffstat (limited to 'tests/test_base.py')
-rw-r--r-- | tests/test_base.py | 16 |
1 files changed, 16 insertions, 0 deletions
diff --git a/tests/test_base.py b/tests/test_base.py index 4f77421..8da655e 100644 --- a/tests/test_base.py +++ b/tests/test_base.py @@ -21,6 +21,7 @@ # THE SOFTWARE. import unittest +import warnings from factory import base from factory import declarations @@ -153,6 +154,21 @@ class FactoryCreationTestCase(unittest.TestCase): self.assertTrue(isinstance(TestObjectFactory.build(), TestObject)) + def testDeprecationWarning(self): + """Make sure the 'auto-discovery' deprecation warning is issued.""" + + with warnings.catch_warnings(record=True) as w: + # Clear the warning registry. + if hasattr(base, '__warningregistry__'): + base.__warningregistry__.clear() + + warnings.simplefilter('always') + class TestObjectFactory(base.Factory): + pass + + self.assertEqual(1, len(w)) + self.assertIn('deprecated', str(w[0].message)) + def testStub(self): class TestFactory(base.StubFactory): pass |