summaryrefslogtreecommitdiff
path: root/tests/test_base.py
diff options
context:
space:
mode:
Diffstat (limited to 'tests/test_base.py')
-rw-r--r--tests/test_base.py47
1 files changed, 47 insertions, 0 deletions
diff --git a/tests/test_base.py b/tests/test_base.py
index b445cb3..aaaa65f 100644
--- a/tests/test_base.py
+++ b/tests/test_base.py
@@ -367,6 +367,28 @@ class FactoryDefaultStrategyTestCase(unittest.TestCase):
self.assertEqual('x0x', test_model.two.one)
self.assertEqual('x0xx0x', test_model.two.two)
+ def testSubFactoryOverriding(self):
+ class TestObject(object):
+ def __init__(self, **kwargs):
+ for k, v in kwargs.iteritems():
+ setattr(self, k, v)
+
+ class TestObjectFactory(base.Factory):
+ FACTORY_FOR = TestObject
+
+ class WrappingTestObjectFactory(base.Factory):
+ FACTORY_FOR = TestObject
+
+ wrapped = declarations.SubFactory(TestObjectFactory, two=2, four=4)
+ wrapped__two = 4
+ wrapped__three = 3
+
+ wrapping = WrappingTestObjectFactory.build()
+ self.assertEqual(wrapping.wrapped.two, 4)
+ self.assertEqual(wrapping.wrapped.three, 3)
+ self.assertEqual(wrapping.wrapped.four, 4)
+
+
def testNestedSubFactory(self):
"""Test nested sub-factories."""
@@ -421,6 +443,31 @@ class FactoryDefaultStrategyTestCase(unittest.TestCase):
self.assertEqual(outer.wrap.wrapped.two.four, 4)
self.assertEqual(outer.wrap.friend, 5)
+ def testSubFactoryAndInheritance(self):
+ """Test inheriting from a factory with subfactories, overriding."""
+ class TestObject(object):
+ def __init__(self, **kwargs):
+ for k, v in kwargs.iteritems():
+ setattr(self, k, v)
+
+ class TestObjectFactory(base.Factory):
+ FACTORY_FOR = TestObject
+ two = 'two'
+
+ class WrappingTestObjectFactory(base.Factory):
+ FACTORY_FOR = TestObject
+
+ wrapped = declarations.SubFactory(TestObjectFactory)
+ friend = declarations.LazyAttribute(lambda o: o.wrapped.two + 1)
+
+ class ExtendedWrappingTestObjectFactory(WrappingTestObjectFactory):
+ wrapped__two = 4
+
+ wrapping = ExtendedWrappingTestObjectFactory.build()
+ self.assertEqual(wrapping.wrapped.two, 4)
+ self.assertEqual(wrapping.friend, 5)
+
+
def testStubStrategy(self):
base.Factory.default_strategy = base.STUB_STRATEGY