From f87f03b583c22e810dced73ed277910d9bc8232a Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Rapha=C3=ABl=20Barrois?= Date: Sun, 15 Apr 2012 10:46:08 +0200 Subject: Rename post_declaration to post_generation. MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Signed-off-by: Raphaël Barrois --- factory/__init__.py | 3 ++- factory/declarations.py | 2 +- 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 -- cgit v1.2.3