Deprecate the extract_prefix option to PostGeneration.

Introduces a new, call-less syntax for the @post_generation decorator.

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

1 files changed, 20 insertions, 1 deletions

diff --git a/tests/test_using.py b/tests/test_using.py
index 2bab8c0..fd8befb 100644
--- a/tests/test_using.py
+++ b/tests/test_using.py
@@ -1173,6 +1173,25 @@ class PostGenerationTestCase(unittest.TestCase):
 
             one = 1
 
+            @factory.post_generation
+            def incr_one(self, _create, _increment):
+                self.one += 1
+
+        obj = TestObjectFactory.build()
+        self.assertEqual(2, obj.one)
+        self.assertFalse(hasattr(obj, 'incr_one'))
+
+        obj = TestObjectFactory.build(one=2)
+        self.assertEqual(3, obj.one)
+        self.assertFalse(hasattr(obj, 'incr_one'))
+
+    @tools.disable_warnings
+    def test_post_generation_calling(self):
+        class TestObjectFactory(factory.Factory):
+            FACTORY_FOR = TestObject
+
+            one = 1
+
             @factory.post_generation()
             def incr_one(self, _create, _increment):
                 self.one += 1
@@ -1191,7 +1210,7 @@ class PostGenerationTestCase(unittest.TestCase):
 
             one = 1
 
-            @factory.post_generation()
+            @factory.post_generation
             def incr_one(self, _create, increment=1):
                 self.one += increment