为Django RESTful API编写功能测试



我正在尝试为使用Django REST框架编写的REST API编写一些功能测试。不过,它并不是很具体,因为它主要是Django的通用内容。

这就是我想做的

  1. 在测试类的setUp方法中创建用户
  2. 使用测试客户端从API请求用户令牌

测试.py

from django.test import LiveServerTestCase
from django.contrib.auth.models import User
from django.test.client import Client
from rest_framework.authtoken.models import Token
class TokenAuthentication(LiveServerTestCase):
    def setUp(self):
        user = User.objects.create(username='foo', password='password', email="foo@example.com")
        user.save()
        self.c = Client()
    def test_get_auth_token(self):
        user = User.objects.get(username="foo")
        print user # this outputs foo
        print Token.objects.get(user_id = user.pk) # this outputs a normal looking token
        response = self.c.post("/api-token-auth/", {'username': 'foo', 'password': 'password'})
        print response.status_code # this outputs 400
        self.assertEqual(response.status_code, 200, "User couldn't log in")

当我运行测试时,它返回的状态是400而不是200,所以用户没有经过身份验证。如果我输入数据库中已经存在的用户的凭据,它就会通过。所以我假设在测试类中创建的记录只能在它自己的方法中访问,这可能是因为它是为单元测试而设计的。但我使用数据库中的数据来执行测试,如果数据发生变化,它将失败。

像这样的功能测试,在运行测试之前需要创建数据,应该如何在Django中执行?

您创建的用户不正确。User.objects.create以纯文本而不是通过哈希机制设置密码。您应该使用User.objects.create_user,它将正确设置密码,以便您可以使用用户名和密码进行身份验证。

最新更新