Skip to content
Snippets Groups Projects
Commit 33fe7a18 authored by Nicolas Mérigot's avatar Nicolas Mérigot
Browse files

feature: add profile factory

parent b7b997cd
No related branches found
No related tags found
1 merge request!2Feature: add model factory
import factory
from .models import Profile
from djangoldp.factories import UserFactory
from django.db.models.signals import post_save
@factory.django.mute_signals(post_save)
class ProfileFactory(factory.django.DjangoModelFactory):
class Meta:
model = Profile
user = factory.SubFactory(UserFactory)
available = factory.Faker('null_boolean')
bio = factory.Faker('text', max_nb_chars=250)
city = factory.Faker('city')
country = factory.Faker('country')
phone = factory.Faker('phone_number')
website = factory.Faker('url')
from django.core.management.base import BaseCommand, CommandError
from djangoldp_profile.factories import ProfileFactory
from django.contrib.auth.models import User
class Command(BaseCommand):
help = 'Mock data'
def handle(self, *args, **options):
for user in User.objects.filter(profile__isnull=True):
ProfileFactory.create(user=user);
self.stdout.write(self.style.SUCCESS('Successful data mock install'))
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment