test: Fixture for creating model with random data

ref: #731 #730
This commit is contained in:
2025-04-24 12:41:16 +09:30
parent c4e81ffbed
commit 99211b7505

View File

@ -99,6 +99,231 @@ def create_model(request, django_db_blocker):
@pytest.fixture( scope = 'class')
def create_model_random_data_five(request, django_db_blocker):
"""Create the specified model with random data
**Note:** Dont use this fixture yet. More testing is required
Yields:
models.Model: Created Model
"""
from django.db import models
from django.core.files.base import ContentFile
import random
import os
from faker import Faker
from django.db.models.fields.reverse_related import ManyToOneRel, ManyToManyRel, OneToOneRel
fake = Faker()
_unique_counter = {}
created_instances = {}
def generate_unique_string(length=8):
base = fake.word()
counter = _unique_counter.get(base, 0) + 1
_unique_counter[base] = counter
return f"{base}_{counter}"[:length]
def generate_random_file():
return ContentFile(b"dummydata", name=f"{generate_unique_string(5)}.txt")
def get_random_choice(choices):
valid_choices = [choice[0] for choice in choices if not isinstance(choice[1], (list, tuple))]
return random.choice(valid_choices or [None])
def create_sample_instance(model_class, seen=None, except_fields = [], recurse_depth = 0):
if seen is None:
seen = set()
recurse_depth += 1
if recurse_depth == 3:
return None
if model_class in created_instances:
return created_instances[model_class]
if model_class in seen:
return None
seen.add(model_class)
field_data = {}
m2m_data = {}
# Only iterate over concrete, editable fields
for field in model_class._meta.get_fields():
# if field.auto_created and not field.concrete:
# continue # Skip reverse relations like ManyToOneRel
if isinstance(field, (ManyToOneRel, ManyToManyRel, OneToOneRel)):
continue # Skip reverse relationships only
if getattr(field, 'primary_key', False) or field.name in ['id', 'pk']:
continue
if field.name in except_fields:
continue
if isinstance(field, models.ManyToManyField):
related_model = field.related_model
related_instance = create_sample_instance(related_model, seen)
if related_instance:
m2m_data[field.name] = [related_instance]
elif hasattr(field, 'choices') and field.choices:
field_data[field.name] = get_random_choice(field.choices)
elif isinstance(field, models.CharField):
field_data[field.name] = generate_unique_string(length=field.max_length or 12)
elif isinstance(field, models.TextField):
field_data[field.name] = generate_unique_string(length=200)
elif isinstance(field, models.EmailField):
field_data[field.name] = f"{generate_unique_string(8)}@example.com"
elif isinstance(field, models.IntegerField):
field_data[field.name] = random.randint(1, 99999)
elif isinstance(field, models.FloatField):
field_data[field.name] = random.uniform(1.0, 1000.0)
elif isinstance(field, models.DecimalField):
field_data[field.name] = round(random.uniform(1.0, 1000.0), field.decimal_places)
elif isinstance(field, models.BooleanField):
field_data[field.name] = random.choice([True, False])
elif isinstance(field, models.DateField):
field_data[field.name] = fake.date()
elif isinstance(field, models.DateTimeField):
field_data[field.name] = fake.date_time()
elif isinstance(field, models.TimeField):
field_data[field.name] = fake.time()
elif isinstance(field, models.UUIDField):
field_data[field.name] = fake.uuid4()
elif isinstance(field, models.IPAddressField):
field_data[field.name] = fake.ipv4()
elif isinstance(field, models.GenericIPAddressField):
field_data[field.name] = fake.ipv6()
elif isinstance(field, models.SlugField):
field_data[field.name] = fake.slug()
elif isinstance(field, models.BinaryField):
field_data[field.name] = os.urandom(10)
elif isinstance(field, models.JSONField):
field_data[field.name] = {generate_unique_string(4): random.randint(1, 10)}
elif isinstance(field, (models.FileField, models.ImageField)):
field_data[field.name] = generate_random_file()
elif isinstance(field, (models.ForeignKey, models.OneToOneField)):
related_model = field.related_model
if model_class == related_model:
related_instance = create_sample_instance(related_model, seen=None, except_fields=except_fields, recurse_depth = recurse_depth)
else:
related_instance = create_sample_instance(related_model, seen)
if related_instance:
field_data[field.name] = related_instance
with django_db_blocker.unblock():
instance = model_class.objects.create(**field_data)
created_instances[model_class] = instance
for field_name, related_instances in m2m_data.items():
getattr(instance, field_name).set(related_instances)
seen.remove(model_class)
return instance
yield create_sample_instance
with django_db_blocker.unblock():
for model_class, model in created_instances.items():
model.delete()
@pytest.fixture( scope = 'class')
def organization_one(django_db_blocker):