summaryrefslogtreecommitdiff
path: root/tests
diff options
context:
space:
mode:
authorRaphaël Barrois <raphael.barrois@polyconseil.fr>2012-04-15 09:45:11 +0200
committerRaphaël Barrois <raphael.barrois@polyconseil.fr>2012-04-15 10:06:39 +0200
commitf8a93d6599b71123789541adf9658cd095402f0f (patch)
tree2c1dfe6123775a5591a394d712734b0827f44aef /tests
parent0e7fed312bf2de6d628a61b116bd91e04bf0a9ff (diff)
downloadfactory-boy-f8a93d6599b71123789541adf9658cd095402f0f.tar
factory-boy-f8a93d6599b71123789541adf9658cd095402f0f.tar.gz
Expose and test factory.RelatedFactory.
Signed-off-by: Raphaël Barrois <raphael.barrois@polyconseil.fr>
Diffstat (limited to 'tests')
-rw-r--r--tests/test_using.py44
1 files changed, 44 insertions, 0 deletions
diff --git a/tests/test_using.py b/tests/test_using.py
index 54106a9..13f23dc 100644
--- a/tests/test_using.py
+++ b/tests/test_using.py
@@ -868,6 +868,50 @@ class PostDeclarationHookTestCase(unittest.TestCase):
self.assertEqual(4, obj.one)
self.assertFalse(hasattr(obj, 'incr_one'))
+ def test_related_factory(self):
+ class TestRelatedObject(object):
+ def __init__(self, obj=None, one=None, two=None):
+ obj.related = self
+ self.one = one
+ self.two = two
+ self.three = obj
+
+ class TestRelatedObjectFactory(factory.Factory):
+ FACTORY_FOR = TestRelatedObject
+ one = 1
+ two = factory.LazyAttribute(lambda o: o.one + 1)
+
+ class TestObjectFactory(factory.Factory):
+ FACTORY_FOR = TestObject
+ one = 3
+ two = 2
+ three = factory.RelatedFactory(TestRelatedObjectFactory, name='obj')
+
+ obj = TestObjectFactory.build()
+ # Normal fields
+ self.assertEqual(3, obj.one)
+ self.assertEqual(2, obj.two)
+ # RelatedFactory was built
+ self.assertIsNone(obj.three)
+ self.assertIsNotNone(obj.related)
+ self.assertEqual(1, obj.related.one)
+ self.assertEqual(2, obj.related.two)
+ # RelatedFactory was passed "parent" object
+ self.assertEqual(obj, obj.related.three)
+
+ obj = TestObjectFactory.build(three__one=3)
+ # Normal fields
+ self.assertEqual(3, obj.one)
+ self.assertEqual(2, obj.two)
+ # RelatedFactory was build
+ self.assertIsNone(obj.three)
+ self.assertIsNotNone(obj.related)
+ # three__one was correctly parse
+ self.assertEqual(3, obj.related.one)
+ self.assertEqual(4, obj.related.two)
+ # RelatedFactory received "parent" object
+ self.assertEqual(obj, obj.related.three)
+
if __name__ == '__main__':
unittest.main()