问题描述
我的 OnetoOneField
模型和 UserTrust
之间有一个 django.contrib.auth.models.User
,我想在新的 User
注册时创建它。我想过使用 UserTrust
信号创建 user_signed_up
实例。
我的 AppConfig
def ready(self):
# importing model classes
from .models import UserTrust
@receiver(user_signed_up)
def create_usertrust(sender,**kwargs):
u = kwargs['user']
UserTrust.objects.create(user=u)
...这是在我的 pytest 测试中
@pytest.fixture
def test_password():
return 'strong-test-pass'
@pytest.fixture
def create_user(db,django_user_model,test_password):
def make_user(**kwargs):
kwargs['password'] = test_password
if 'username' not in kwargs:
kwargs['username'] = str(uuid.uuid4())
return django_user_model.objects.create_user(**kwargs)
return make_user
@pytest.mark.django_db
def test_my_user(create_user):
user = create_user()
assert user.usertrust.available == 1000
仍然,测试失败
django.contrib.auth.models.User.usertrust.RelatedobjectDoesNotExist:用户没有用户信任。
我错过了什么?
解决方法
您通过 django_user_model 创建用户的问题在于它实际上并没有通过实际发送该信号的 allauth 代码。您有两个选择:
- 使用
client
(因为我假设您使用的是 pytest-django)并通过 allauth 提供的链接填写注册以注册新用户。这样,就会发送一个信号,您可以像这样断言属性和模型实例。 - 您可以简单地忽略信号并对函数本身进行单元测试。而已。您信任那个完全不会改变的单一注册视图,并且信任 API 不会改变的包。我仍然可以推荐此选项,但您已收到警告。
- 您可以自己发送信号。如果 allauth 的 API 发生变化,不推荐使用,但您可以从 allauth 导入信号并像这样发送它:
user_signed_up.send(sender=self.__class__,toppings=toppings,size=size)
其中user_signed_up
是信号。参考文档:https://docs.djangoproject.com/en/dev/topics/signals/#sending-signals
再次强调,如果 API 发生变化,绝对推荐第一个。我也可以推荐第二个选项,因为 allauth 非常有信誉,而且你知道如果没有太大的包更改会发生什么,但你永远不知道。