我正在尝试为使用Django REST框架编写的REST API编写一些功能测试。不过,它并不是很具体,因为它主要是Django的通用内容。
这就是我想做的
- 在测试类的
setUp
方法中创建用户 - 使用测试客户端从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
,它将正确设置密码,以便您可以使用用户名和密码进行身份验证。