Add a PendingDeprecationWarning on associated class auto-discovery.

Signed-off-by: Raphaël Barrois <raphael.barrois@polytechnique.org>

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