Як я можу встановити групи користувачів за допомогою factory_boy - django, django-testing, factory-boy

Я досить новий для Джанго. Я намагаюся встановити groups поле для User використовуючи factory_boy. За умовчанням User клас має поле _groups. Я спробував встановити це, але це не допомагає.

class GroupFactory(factory.django.DjangoModelFactory):
class Meta:
model = Group

name = Sequence(lambda n: "group_{0}".format(n))


class UserFactory(factory.django.DjangoModelFactory):
class Meta:
model = User

username = factory.Sequence(lambda n: "user_{0}".format(n))
password = "test"
first_name = u"ßhamra"
last_name = u"ßhamra"
_groups = factory.SubFactory(GroupFactory)


@classmethod
def _create(cls, model_class, *args, **kwargs):
"""
Override the default ``_create`` with our custom call.
Due to internal behavior or create user create method that                forces is_staff kwarg


"""
g = GroupFactory("abc")
manager = cls._get_manager(model_class)
is_staff = kwargs.pop("is_staff", None)
user = manager.create_user(*args, **kwargs)


if is_staff:
user.is_staff = is_staff
user.save()
return user

Відповіді:

У наведеному нижче коді використовуються стандартні групи django. Вам потрібно використовувати чергову публікацію фабричного хлопця, щоб додати групи до користувача.

Фабрики

import factory
import django.contrib.auth.models as auth_models
from django.contrib.auth.hashers import make_password

user_password = "password"

class SubscribedGroupFactory(factory.django.DjangoModelFactory):
class Meta:
model = auth_models.Group

name = "subscribed"

class UserFactory(factory.django.DjangoModelFactory):
class Meta:
model = auth_models.User

first_name = "Sophia"
last_name = "Ball"
username = "clowngirl@heaven.com"
password = make_password(user_password)
email = "clowngirl@heaven.com"
is_active = True

@factory.post_generation
def groups(self, create, extracted, **kwargs):
if not create:
return

if extracted:
for group in extracted:
self.groups.add(group)

Тест

from app.factories import UserFactory, SubscribedGroupFactory, user_password
class TestCenterTest(TestCase):

def test_test_center_redirect(self):
user = UserFactory.create(groups=(SubscribedGroupFactory.create(),))
self.client.login(username=user.email, password=user_password)
response = self.client.get("/test-center/")
self.assertEqual(302, response.status_code)

З тих пір Group має ManyToManyField відношення до User, ви не можете використовувати SubFactory (що підходить тільки для ForeignKey поле): з a SubFactory, factory_boy спочатку створить Group потім передайте його UserFactory який використовує його як User.objects.create(_groups=Group). Це не вийшло.

Погляньте на Recipes розділ документації: http://factoryboy.readthedocs.org/en/latest/recipes.html#simple-manytomany

Тут повинен працювати наступний код:

  class UserFactory(factory.django.DjangoModelFactory):
# All other declarations here

groups = factory.List([
factory.RelatedFactory(GroupFactory),
])