Xác thực người dùng trong các thử nghiệm với Django factory_boy

Question

Khi thử đơn vị kiểm tra một phần mã của tôi, tôi cần người dùng đăng nhập. Để giảm số lượng đồ đạc tôi đang sử dụng django_factory_boy Nhà máy người dùng nhưng Người dùng tạo không thể xác thực .

from django_factory_boy.auth import UserF 
from django.contrib.auth import authenticate 

user = UserF() 
user.set_password('password') 

sau đó authenticate(username=user.username, password='password') trả lại None thay vì người dùng. Bất kỳ ý tưởng về những gì còn thiếu ở đây?

Answer 1

Bạn nên gọi user.save() sau user.set_password() vì bản thân set_password không lưu người dùng, chỉ đặt dữ liệu.

Answer 2

Một cách khác để làm điều đó:

import factory 
from django.contrib.auth.hashers import make_password 
from somewhere import YourUserModel 

class UserF(factory.django.DjangoModelFactory): 
    FACTORY_FOR = YourUserModel 
    first_name = factory.Sequence(lambda n: "First%s" % n) 
    last_name = factory.Sequence(lambda n: "Last%s" % n) 
    email = factory.Sequence(lambda n: "email%s@example.com" % n) 
    username = factory.Sequence(lambda n: "email%s@example.com" % n) 
    password = make_password("password") 
    is_staff = False 

>>> u = UserF.create() 
>>> u.check_password("password") 
True 

>>> p = UserF.create(password=make_password("password2")) 
>>> p.check_password("password2") 
True 

Answer 3

@ Andrew-Magee giải pháp hoạt động, nhưng đây là giải pháp describe in the doc:

import factory 
from django.contrib.auth.models import User 
#or 
#from somewhere import CustomUser as User 

class UserFactory(factory.DjangoModelFactory): 
    FACTORY_FOR = User 

    username = 'UserFactory' 
    email = 'test@example.com' 
    password = factory.PostGenerationMethodCall('set_password', 'password') 

Django console:

>>> from tests.factories import UserFactory 
>>> from django.contrib.auth.models import check_password 
>>> user = UserFactory() 
>>> user.email 
'test@example.com' 
>>> check_password('password', user.password) 
True 

>>> user2 = UserFactory(username="SecondUserFactory", email='otheremail@example.com', password="ComplexPasswordMuchLonger!") 
>>> user2.email 
'otheremail@example.com' 
>>> check_password('ComplexPasswordMuchLonger!', user2.password) 
True