如何将用户名：密码发送到 unittest 的 app.get() 请求？答案

【问题标题】：How to send username:password to unittest's app.get() request?如何将用户名：密码发送到 unittest 的 app.get() 请求？
【发布时间】：2013-10-31 16:32:40
【问题描述】：

这是我在 Flask-RESTful 中的单元测试的一部分。

self.app = application.app.test_client()
rv = self.app.get('api/v1.0/{0}'.format(ios_sync_timestamp))
eq_(rv.status_code,200)

在命令行中，我可以使用 curl 将用户名：密码发送到服务：

curl -d username:password http://localhost:5000/api/v1.0/1234567

如何在单元测试的 get() 中实现同样的效果？

由于我的 get/put/post 需要身份验证，否则测试将失败。

【问题讨论】：

标签： python unit-testing flask basic-authentication flask-restful

【解决方案1】：

来自RFC 1945, Hypertext Transfer Protocol -- HTTP/1.0

11.1 Basic Authentication Scheme

...

为了接收授权，客户端发送用户 ID 和密码，由单个冒号 (":") 字符分隔，在 base64 [5] 内 credentials.string 中的编码字符串。

...

如果用户代理希望发送用户 ID“Aladdin”和密码 open sesame”，它将使用以下头字段：
  Authorization: Basic QWxhZGRpbjpvcGVuIHNlc2FtZQ==

所以如果你真的使用 http 基本身份验证，你可以像下面这样解决，尽管你的 curl 使用建议了其他一些身份验证方案。

from base64 import b64encode

headers = {
    'Authorization': 'Basic ' + b64encode("{0}:{1}".format(username, password)).decode('utf-8')
}

rv = self.app.get('api/v1.0/{0}'.format(ios_sync_timestamp), headers=headers)

【讨论】：

【解决方案2】：

另一种解决方案 - 所有功劳归于 Doug Black

def request(self, method, url, auth=None, **kwargs):
    headers = kwargs.get('headers', {})
    if auth:
        headers['Authorization'] = 'Basic ' + base64.b64encode(auth[0] + ':' + auth[1])

    kwargs['headers'] = headers

    return self.app.open(url, method=method, **kwargs)

然后在你的测试中使用这个方法：

resp = self.request('GET', 'api/v1.0/{0}'.format(ios_sync_timestamp), auth=(username, password))

【讨论】：

【解决方案3】：

对于 Python 3，请尝试以下示例：

from base64 import b64encode
headers = {
    'Authorization': 'Basic %s' % b64encode(b"username:password").decode("ascii")
}

self.app.get("foo/", headers=headers)

如果您想对用户名和密码使用动态变量，请尝试以下操作：

'Basic %s' % b64encode(bytes(username + ':' + password, "utf-8")).decode("ascii")

另请参阅：Python, HTTPS GET with basic authentication

【讨论】：