58

对于我的一些 Django 视图，我创建了一个装饰器来执行基本 HTTP 访问身份验证。然而，在 Django 中编写测试用例时，我花了一段时间才弄清楚如何对视图进行身份验证。这就是我的做法。我希望有人觉得这很有用。

4

6 回答 6

89

我是这样做的：

from django.test import Client
import base64
auth_headers = {
    'HTTP_AUTHORIZATION': 'Basic ' + base64.b64encode('username:password'),
}
c = Client()
response = c.get('/my-protected-url/', **auth_headers)

注意：您还需要创建一个用户。

于 2011-03-31T04:50:51.353 回答
34

在您的 Django TestCase 中，您可以更新客户端默认值以包含您的 HTTP 基本身份验证凭据。

import base64
from django.test import TestCase

class TestMyStuff(TestCase):

    def setUp(self):
        credentials = base64.b64encode('username:password')
        self.client.defaults['HTTP_AUTHORIZATION'] = 'Basic ' + credentials
于 2012-01-31T23:22:02.287 回答
6

对于 python3，您可以对username:password字符串进行 base64 编码：

base64.b64encode(b'username:password')

这将返回字节，因此您需要将其转换为 ASCII 字符串.decode('ascii')：

完整示例：

import base64

from django.test import TestCase

class TestClass(TestCase):
   def test_authorized(self):
       headers = {
           'HTTP_AUTHORIZATION': 'Basic ' + 
                base64.b64encode(b'username:password').decode("ascii")
       }
       response = self.client.get('/', **headers)
       self.assertEqual(response.status_code, 200)
于 2017-05-30T15:56:22.330 回答
2

假设我有一个登录表单，我使用以下技术通过测试框架登录：

    client = Client()
    client.post('/login/', {'username': 'john.smith', 'password': 'secret'})

然后我在其他测试中随身携带，client因为它已经过身份验证。你对这篇文章有什么疑问？

于 2011-03-31T15:34:24.913 回答
2

（python3）我在测试中使用它：

credentials_string = '%s:%s' % ('invalid', 'invalid')
credentials = base64.b64encode(credentials_string.encode())
self.client.defaults['HTTP_AUTHORIZATION'] = 'Basic ' + credentials.decode()

以及以下观点：

import base64
[...]
type, auth = request.META['HTTP_AUTHORIZATION'].split(' ', 1)
auth = base64.b64decode(auth.strip()).decode()
于 2019-10-11T13:29:07.273 回答
-1

另一种方法是绕过 Django Client() 并使用 Requests 代替。

class MyTest(TestCase):
    def setUp(self):
        AUTH = requests.auth.HTTPBasicAuth("username", "password")

    def some_test(self):
        resp = requests.get(BASE_URL + 'endpoint/', auth=AUTH)
        self.assertEqual(resp.status_code, 200)
于 2014-10-30T15:24:32.170 回答