def test_replaced_sequence(self):
seq = declarations.Sequence(lambda n: 'xx%d' % n)
seq2 = declarations.Sequence(lambda n: 'yy%d' % n)
FakeFactory = self.make_fake_factory({'one': seq})
ab = containers.AttributeBuilder(FakeFactory, extra={'one': seq2})
self.assertEqual({'one': 'yy1'}, ab.build(create=False))
class TestObjectFactory(Factory):
@classmethod
def _setup_next_sequence(cls):
return 42
one = declarations.Sequence(lambda n: 'one' + n)
two = declarations.Sequence(lambda n: 'two' + n)
def test_replaced_sequence(self):
seq = declarations.Sequence(lambda n: 'xx%d' % n)
seq2 = declarations.Sequence(lambda n: 'yy%d' % n)
class FakeFactory(object):
@classmethod
def declarations(cls, extra):
d = {'one': seq}
d.update(extra)
return d
@classmethod
def _generate_next_sequence(cls):
return 1
ab = containers.AttributeBuilder(FakeFactory, extra={'one': seq2})
self.assertEqual({'one': 'yy1'}, ab.build(create=False))
def test_additionnal_sequence(self):
seq = declarations.Sequence(lambda n: 'xx' + n)
class FakeFactory(object):
@classmethod
def declarations(cls, extra):
d = {'one': 1}
d.update(extra)
return d
@classmethod
def _generate_next_sequence(cls):
return 1
ab = containers.AttributeBuilder(FakeFactory, extra={'two': seq})
self.assertEqual({'one': 1, 'two': 'xx1'}, ab.build(create=False))
class TestObjectFactory(base.Factory):
class Meta:
model = TestObject
one = declarations.Sequence(lambda n: n)
class TestObjectFactory(base.Factory):
FACTORY_FOR = TestObject
one = declarations.Sequence(lambda n: n)
def test_additionnal_sequence(self):
seq = declarations.Sequence(lambda n: 'xx%d' % n)
FakeFactory = self.make_fake_factory({'one': 1})
ab = containers.AttributeBuilder(FakeFactory, extra={'two': seq})
self.assertEqual({'one': 1, 'two': 'xx1'}, ab.build(create=False))
def test_factory_defined_sequence(self):
seq = declarations.Sequence(lambda n: 'xx%d' % n)
FakeFactory = self.make_fake_factory({'one': seq})
ab = containers.AttributeBuilder(FakeFactory)
self.assertEqual({'one': 'xx1'}, ab.build(create=False))
class TestObjectFactory(Factory):
one = declarations.Sequence(lambda n: 'one' + n)
two = declarations.Sequence(lambda n: 'two' + n)
class TestModel2Factory(Factory):
FACTORY_FOR = TestModel2
two = declarations.SubFactory(TestModelFactory,
one=declarations.Sequence(lambda n: 'x%sx' % n),
two=declarations.LazyAttribute(
lambda o: '%s%s' % (o.one, o.one)))
class TestObjectFactory(Factory):
one = declarations.Sequence(lambda a: a)
