Django测试未实现对象

Question

我有一个 django rest 框架测试，它只是常规 django 测试的包装器，其工作方式完全相同。代码如下所示：

user_created = User.objects.create_user(first_name="Wally", username="farseer@gmail.com", password="1234",
                                            email="farseer@gmail.com")

client_created = Client.objects.create(user=user_created, cart=cart)

data_client_profile["user"]["first_name"] = "Apoc"

response = self.client.put(reverse("misuper:client_profile"), data_client_profile, format="json")

client_created.refresh_from_db()  # Tried this too

self.assertEqual(response.status_code, status.HTTP_200_OK)

self.assertEqual(client_created.user.first_name, data_client_profile["user"]["first_name"])

所以，我要更新client_created与在字典一些数据对象data_client_profile，那么assertEqual该client.user.first_name是“载脂蛋白C”。

这是视图中的代码，我添加了两个pdb.set_trace()不仅有助于粘贴所有代码：

        pdb.set_trace()

        client_existing_user_obj.phone = phone
        client_existing_user_obj.user.email = email
        client_existing_user_obj.user.first_name = first_name # Updating here!
        client_existing_user_obj.user.last_name = last_name
        client_existing_user_obj.user.save()
        client_existing_user_obj.save()
        pdb.set_trace()

第一个 pdb 中断显示了这一点：

(Pdb) client_existing_user_obj.user.username
u'farseer@gmail.com'  # Make sure I'm updating the created object
(Pdb) client_existing_user_obj.user.first_name
u'Wally'  # First name is not updated yet

第二个 pdb 中断显示了这一点：

(Pdb) client_existing_user_obj.user.first_name
u'Apoc'  # Looks like the first name has being updated!

但是，当测试运行时，我收到错误：

self.assertEqual(client_created.user.first_name, data_client_profile["user"]["first_name"])
AssertionError: 'Wally' != 'Apoc'

为什么会失败？我什至打电话refresh_from_db()。我确认它已在视图中更新，但在测试中它看起来没有。我不明白。

Answer 1

请注意，refresh_from_dbsay that 的文档client_created.user不会被刷新client_created.refresh_from_db()，因为client_created.user_id保持不变：

先前加载的关系值不再有效的相关实例将从重新加载的实例中删除。例如，如果你从重装比如一个外键与名称的另一个模型Author，那么如果obj.author_id != obj.author.id，obj.author就会被扔掉，并在下次访问将与价值被重新加载obj.author_id。

因此你需要刷新client_created.user：

client_created.user.refresh_from_db()

或重新获取client_created自己：

client_created = Client.objects.get(pk=client_created.pk)