Add a new type of attribute, SelfAttribute.

Should be used as a shortcut for 'lambda o: o.some_attr' ; most useful in SubFactories.

Signed-off-by: Raphaël Barrois <raphael.barrois@polyconseil.fr>

1 files changed, 8 insertions, 0 deletions

diff --git a/factory/test_base.py b/factory/test_base.py
index 689b56d..b324680 100644
--- a/factory/test_base.py
+++ b/factory/test_base.py
@@ -115,6 +115,14 @@ class FactoryTestCase(unittest.TestCase):
         test_object = TestObjectFactory.build()
         self.assertEqual(test_object.one, 'one')
 
+    def testSelfAttribute(self):
+        class TestObjectFactory(Factory):
+            one = 'xx'
+            two = declarations.SelfAttribute('one')
+
+        test_object = TestObjectFactory.build(one=1)
+        self.assertEqual(1, test_object.two)
+
     def testSequenceDecorator(self):
         class TestObjectFactory(Factory):
             @declarations.sequence