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)