Creating multiple objects with a foreign key

Question

I need to create ten user samples ( User ), and each of them should have fifty documents ( Doc ). How to do this in test.py using factoryboy?

 #factories.py from app_name.models import * import factory from datetime import datetime, timedelta, time from django.contrib.auth.models import User class UserFactory(factory.Factory): FACTORY_FOR = User username = factory.Sequence(lambda n: 'User ' + n) email = '[e-mail]' password = '1234567' class DocFactory(factory.Factory): FACTORY_FOR = Doc user = factory.SubFactory(UserFactory) kategories = '1' doc_number = '12345678' date_join = factory.Sequence(lambda n:(datetime.now() + timedelta(days=n)).date(), int) 

in my tests.py:

 from django.test import TestCase from django_dynamic_fixture import G from factories import * 

Answer 1

 users = UserFactory.create_batch(10) for user in users: doc = DocFactory.create(user=user) 

Answer 2

You can use the post_generation decorator:

 class UserFactory(factory.Factory): ... @factory.post_generation def create_docs(self, create, extracted, **kwargs): if not create: return for i in range(50): doc = DocFactory.create(user=self) 

Answer 3

For those of you who work with SQLAlchemy , this can be done with the following recipe (note that I am using the Person / Address models instead of the User / Docs model example above).

 from sqlalchemy import create_engine, Integer, Text, ForeignKey, Column from sqlalchemy.ext.declarative import declarative_base from sqlalchemy.orm import relationship, scoped_session, sessionmaker import factory from factory.alchemy import SQLAlchemyModelFactory as sqla_factory import random engine = create_engine("sqlite:////tmp/factory_boy.sql") session = scoped_session(sessionmaker(bind=engine)) Base = declarative_base() class Person(Base): id = Column(Integer, primary_key=True) name = Column(Text) addresses = relationship("Address", backref="person") class Address(Base): id = Column(Integer, primary_key=True) city = Column(Text) street = Column(Text) street_number = Column(Integer) person_id = Column(Integer, ForeignKey('person.id')) class AddressFactory(sqla_factory): class Meta: model = Address sqlalchemy_session = session street_number = random.randint(0, 10000) street = "Commonwealth Ave" city = "Boston" class PersonFactory(sqla_factory): class Meta: model = Person sqlalchemy_session = session name = "John Doe" Base.metadata.create_all(engine) for i in range(100): person = PersonFactory(addresses=AddressFactory.create_batch(3)) 

This creates 3 workouts for each person created, where each workout refers to a person through person_id FK.