Вложенный serilailzer и трудности с подфакторами в модульном тесте django

Я получаю неудачный тест, потому что (я думаю) я не могу добавить подфактурное поле.

У меня есть сериализатор, который вызывает другой сериализатор. Мне трудно воспроизвести это в тесте. Проблема вызвана моим тестом: _expected_restuarant_response. Он ожидает ответ, включающий поле "сотрудник". Проблема в том, что я не могу добавить это подполе в RestaurantFactory, потому что RestaurantFactory находится выше EmployeeFactory в файле factories.py (EmployeeFactory не вызывается в этот момент в файле). Я также не могу поместить EmployeeFactory выше, чем RestaurantFactory, потому что оно ссылается на RestaurantFactory.

Может кто-нибудь направить меня в правильном направлении? Спасибо.

Упрощенная версия моих файлов:

models.py:

class Restaurant(models.Model):
    name = models.CharField(max_length=20)

class Employee(models.Model):
    badge_id = models.CharField(max_length=20)
    restaurant = models.ForeignKey(Restaurant)

serializers.py:

class EmployeeSerializer(serializers.ModelSerializer):
    class Meta:
        model = Employee
        fields = [
            'badge_id']

class RestaurantSerializer(serializers.ModelSerializer):
    employee = EmployeeSerializer(many=True, read_only=True)
    class Meta:
        model = Restaurant
        fields = [
            'name',
            'employee']

factories.py:

class RestaurantFactory(factory.django.DjangoModelFactory):
    name = factory.Faker('company')
         < ----------------------------------- do I need to add an Employee subfactory here? If so how?

class EmployeeFactory(factory.django.DjangoModelFactory):
    badge_id = factory.Faker('id')
    restaurant = factory.SubFactory(RestaurantFactory)

test.py:

def _expected_restuarant_response(restaurant):
    return {
        restaurant.name,
        restaurant.employee_set.badge_id     <---- the test fails here because the RestaurantFactory does not include this field
         }

assert response_json == [_expected_restaurant_response(RestaurantFactory())]

В плане создания фабрики вы, вероятно, могли бы сделать что-то вроде этого:

class RestaurantFactory(factory.django.DjangoModelFactory):
    name = factory.Faker('company')
    employee = factory.SubFactory('path.to.EmployeeFactory', employee=None)

then on EmployeeFactory do this:

class EmployeeFactory(factory.django.DjangoModelFactory):
    badge_id = factory.Faker('id')
    restaurant = factory.RelatedFactory(RestaurantFactory, factory_related_name='employee')
Вернуться на верх