Question

我的 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：用户没有用户信任。

我错过了什么？

Answer 1

您通过 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 非常有信誉，而且你知道如果没有太大的包更改会发生什么，但你永远不知道。

如何使用 pytest 触发 allauth.account.signals.user_signed_up 信号？

1 个答案: