在 Django 测试框架中使用基本 HTTP 访问身份验证

Question

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

Answer 1

我是这样做的：

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)

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

Answer 2

在 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

Answer 3

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

base64.b64encode(b'username:password')

这会返回字节，因此您需要使用 .decode('ascii') 将其转换为 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)

Answer 4

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

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

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

Answer 5

（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()

Answer 6

另一种方法是绕过 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)