From a5184785b1229e02ec75506db675a2377b32c297 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Rapha=C3=ABl=20Barrois?= Date: Wed, 14 Nov 2012 23:53:26 +0100 Subject: Keep FACTORY_FOR around. MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit And add a test too. Signed-off-by: Raphaël Barrois --- tests/test_base.py | 8 ++++++++ 1 file changed, 8 insertions(+) (limited to 'tests/test_base.py') diff --git a/tests/test_base.py b/tests/test_base.py index 63fc62b..64570cd 100644 --- a/tests/test_base.py +++ b/tests/test_base.py @@ -64,6 +64,14 @@ class SafetyTestCase(unittest.TestCase): class FactoryTestCase(unittest.TestCase): + def test_factory_for(self): + class TestObjectFactory(base.Factory): + FACTORY_FOR = TestObject + + self.assertEqual(TestObject, TestObjectFactory.FACTORY_FOR) + obj = TestObjectFactory.build() + self.assertFalse(hasattr(obj, 'FACTORY_FOR')) + def testDisplay(self): class TestObjectFactory(base.Factory): FACTORY_FOR = FakeDjangoModel -- cgit v1.2.3