diff options
author | Raphaël Barrois <raphael.barrois@polyconseil.fr> | 2012-04-15 10:46:08 +0200 |
---|---|---|
committer | Raphaël Barrois <raphael.barrois@polyconseil.fr> | 2012-04-15 10:46:08 +0200 |
commit | f87f03b583c22e810dced73ed277910d9bc8232a (patch) | |
tree | ae1684e3ec09d4cbe79df336924cc134e7a1e146 | |
parent | f8a93d6599b71123789541adf9658cd095402f0f (diff) | |
download | factory-boy-f87f03b583c22e810dced73ed277910d9bc8232a.tar factory-boy-f87f03b583c22e810dced73ed277910d9bc8232a.tar.gz |
Rename post_declaration to post_generation.
Signed-off-by: Raphaël Barrois <raphael.barrois@polyconseil.fr>
-rw-r--r-- | factory/__init__.py | 3 | ||||
-rw-r--r-- | factory/declarations.py | 2 | ||||
-rw-r--r-- | tests/test_using.py | 10 |
3 files changed, 8 insertions, 7 deletions
diff --git a/factory/__init__.py b/factory/__init__.py index 179a633..73425aa 100644 --- a/factory/__init__.py +++ b/factory/__init__.py @@ -60,6 +60,7 @@ from declarations import ( SelfAttribute, ContainerAttribute, SubFactory, + PostGeneration, RelatedFactory, lazy_attribute, @@ -68,6 +69,6 @@ from declarations import ( sequence, lazy_attribute_sequence, container_attribute, - post_declaration, + post_generation, ) diff --git a/factory/declarations.py b/factory/declarations.py index ddc6c78..4d6e767 100644 --- a/factory/declarations.py +++ b/factory/declarations.py @@ -309,7 +309,7 @@ class PostGeneration(PostGenerationDeclaration): self.function(obj, create, extracted, **kwargs) -def post_declaration(extract_prefix=None): +def post_generation(extract_prefix=None): def decorator(fun): return PostGeneration(fun, extract_prefix=extract_prefix) return decorator diff --git a/tests/test_using.py b/tests/test_using.py index 13f23dc..20acd79 100644 --- a/tests/test_using.py +++ b/tests/test_using.py @@ -835,12 +835,12 @@ class IteratorTestCase(unittest.TestCase): self.assertEqual(i % 5, obj.one) -class PostDeclarationHookTestCase(unittest.TestCase): - def test_post_declaration(self): +class PostGenerationDeclarationTestCase(unittest.TestCase): + def test_post_generation(self): class TestObjectFactory(factory.Factory): one = 1 - @factory.post_declaration() + @factory.post_generation() def incr_one(self, _create, _increment): self.one += 1 @@ -852,11 +852,11 @@ class PostDeclarationHookTestCase(unittest.TestCase): self.assertEqual(3, obj.one) self.assertFalse(hasattr(obj, 'incr_one')) - def test_post_declaration_extraction(self): + def test_post_generation_extraction(self): class TestObjectFactory(factory.Factory): one = 1 - @factory.post_declaration() + @factory.post_generation() def incr_one(self, _create, increment=1): self.one += increment |