Populate DB and Factories for tests

[to-sta]

Description

I am trying to achive two things with a factory:

1. Populate the database with dummy data
2. Use Factories for test with pytest

For populating the database, I want to create X users with X organization, hence the query. Because I don't want to have an additional X amount of users, created with the SubFactory.

My attempt was to solve it with a conditional.

created_by = (
    factory.Iterator(UserModel.objects.exclude(username="admin").all())
    if UserModel.objects.exclude(username="admin").exists()
    else factory.SubFactory("authentication.factories.UserFactory")
)

RuntimeError: Database access not allowed, use the "django_db" mark, or the "db" or "transactional_db" fixtures to enable it.

To Reproduce

The bug is occuring whenever I ran pytest.

Model / Factory code

class OrganizationFactory(factory.django.DjangoModelFactory):
    class Meta:
        model = Organization

    name = factory.Faker("word")
    tagline = factory.Faker("word")
    social_links = ["https://www.instagram.com/activist_org/"]
    created_by = (
        factory.Iterator(UserModel.objects.exclude(username="admin").all())
        if UserModel.objects.exclude(username="admin").exists()
        else factory.SubFactory("authentication.factories.UserFactory")
    )
    status = factory.SubFactory("entities.factories.StatusTypeFactory", name="Active")
    is_high_risk = factory.Faker("boolean")
    location = factory.Faker("city")
    acceptance_date = factory.LazyFunction(datetime.datetime.now)

The issue

Add a short description along with your code

# Include the code that provoked the bug, including as full a stack-trace as possible

1. Populating the database:

OrganizationFactory.create_batch(number_of_orgs)

2. Test:

organization = OrganizationFactory.build()

Notes

I haven't seen a solution for my problem in the docs..