python接口测试框架

python接口测试框架：

发送请求

import requests
r = requests.get(‘https://api.github.com/events’)
r = requests.post(‘http://httpbin.org/post’, data = {‘key’:‘value’})
r = requests.put(‘http://httpbin.org/put’, data = {‘key’:‘value’})
r = requests.delete(‘http://httpbin.org/delete’)
r = requests.head(‘http://httpbin.org/get’)
r = requests.options(‘http://httpbin.org/get’)

传递 URL 参数

payload = {‘key1’: ‘value1’, ‘key2’: ‘value2’}
r = requests.get(“http://httpbin.org/get”, params=payload)
print(r.url)
http://httpbin.org/get?key2=value2&key1=value1

payload = {‘key1’: ‘value1’, ‘key2’: [‘value2’, ‘value3’]}

r = requests.get(‘http://httpbin.org/get’, params=payload)
print(r.url)
http://httpbin.org/get?key1=value1&key2=value2&key2=value3

响应内容

import requests
r = requests.get(‘https://api.github.com/events’)
r.text
u’[{“repository”:{“open_issues”:0,“url”:"https://github.com/…

r.encoding
‘utf-8’

r.encoding = ‘ISO-8859-1’

二进制响应内容

r.content
b’[{“repository”:{“open_issues”:0,“url”:"https://github.com/…

from PIL import Image
from io import BytesIO

i = Image.open(BytesIO(r.content))

JSON 响应内容

import requests

r = requests.get(‘https://api.github.com/events’)
r.json()
[{u’repository’: {u’open_issues’: 0, u’url’: 'https://github.com/…

原始响应内容

想获取来自服务器的原始套接字响应,问 r.raw. 需要确保在初始请求中设置了 stream=True。

r = requests.get(‘https://api.github.com/events’, stream=True)
r.raw
<requests.packages.urllib3.response.HTTPResponse object at 0x101194810>

r.raw.read(10)
‘\x1f\x8b\x08\x00\x00\x00\x00\x00\x00\x03’

with open(filename, ‘wb’) as fd:
for chunk in r.iter_content(chunk_size):
fd.write(chunk)

定制请求头

url = ‘https://api.github.com/some/endpoint’
headers = {‘user-agent’: ‘my-app/0.0.1’}

r = requests.get(url, headers=headers)

更加复杂的 POST 请求

payload = {‘key1’: ‘value1’, ‘key2’: ‘value2’}

r = requests.post(“http://httpbin.org/post”, data=payload)
print(r.text)
{
…
“form”: {
“key2”: “value2”,
“key1”: “value1”
},
…
}

data 参数传入一个元组列表,更多适用于表单中多个元素使用同一 key

payload = ((‘key1’, ‘value1’), (‘key1’, ‘value2’))
r = requests.post(‘http://httpbin.org/post’, data=payload)
print(r.text)
{
…
“form”: {
“key1”: [
“value1”,
“value2”
]
},
…
}

发送的数据并非编码为表单形式的。如果你传递一个 string 而不是一个 dict,那么数据会被直接发布出去。

import json

url = ‘https://api.github.com/some/endpoint’
payload = {‘some’: ‘data’}

r = requests.post(url, data=json.dumps(payload))

使用 json 参数直接传递,会被自动编码

url = ‘https://api.github.com/some/endpoint’
payload = {‘some’: ‘data’}

r = requests.post(url, json=payload)

响应状态码

r = requests.get(‘http://httpbin.org/get’)
r.status_code
200
Requests还附带了一个内置的状态码查询对象

r.status_code == requests.codes.ok
True

如果发送了一个错误请求(一个 4XX 客户端错误，或者 5XX 服务器错误响应)，我们可以通过 Response.raise_for_status() 来抛出异常：

bad_r = requests.get(‘http://httpbin.org/status/404’)
bad_r.status_code
404

bad_r.raise_for_status()
Traceback (most recent call last):
File “requests/models.py”, line 832, in raise_for_status
raise http_error
requests.exceptions.HTTPError: 404 Client Error

响应头

r.headers
{
‘content-encoding’: ‘gzip’,
‘transfer-encoding’: ‘chunked’,
‘connection’: ‘close’,
‘server’: ‘nginx/1.0.4’,
‘x-runtime’: ‘148ms’,
‘etag’: ‘“e1ca502697e5c9317743dc078f67693f”’,
‘content-type’: ‘application/json’
}

Cookie

url = ‘http://example.com/some/cookie/setting/url’
r = requests.get(url)

r.cookies[‘example_cookie_name’]
‘example_cookie_value’

url = ‘http://httpbin.org/cookies’
cookies = dict(cookies_are=‘working’)

r = requests.get(url, cookies=cookies)
r.text
‘{“cookies”: {“cookies_are”: “working”}}’

Cookie 的返回对象为 RequestsCookieJar，它的行为和字典类似，但接口更为完整，适合跨域名跨路径使用。你还可以把 Cookie Jar 传到 Requests 中：

jar = requests.cookies.RequestsCookieJar()
jar.set(‘tasty_cookie’, ‘yum’, domain=‘httpbin.org’, path=’/cookies’)
jar.set(‘gross_cookie’, ‘blech’, domain=‘httpbin.org’, path=’/elsewhere’)
url = ‘http://httpbin.org/cookies’
r = requests.get(url, cookies=jar)
r.text
‘{“cookies”: {“tasty_cookie”: “yum”}}’

重定向与请求历史
默认情况下，除了 HEAD, Requests 会自动处理所有重定向。

可以使用响应对象的 history 方法来追踪重定向。
Response.history 是一个 Response 对象的列表，为了完成请求而创建了这些对象。这个对象列表按照从最老到最近的请求进行排序。

r = requests.get(‘http://github.com’)

r.url
‘https://github.com/’

r.status_code
200

r.history
[<Response [301]>]

如果使用的是GET、OPTIONS、POST、PUT、PATCH 或者 DELETE，那么你可以通过 allow_redirects 参数禁用重定向处理：

r = requests.get(‘http://github.com’, allow_redirects=False)
r.status_code
301

r.history
[]
如果你使用了 HEAD，你也可以启用重定向:

r = requests.head(‘http://github.com’, allow_redirects=True)
r.url
‘https://github.com/’

r.history
[<Response [301]>]

超时
告诉 requests 在经过以 timeout 参数设定的秒数时间之后停止等待响应

requests.get(‘http://github.com’, timeout=0.001)
Traceback (most recent call last):
File “”, line 1, in
requests.exceptions.Timeout: HTTPConnectionPool(host=‘github.com’, port=80): Request timed out. (timeout=0.001)

错误与异常
遇到网络问题（如：DNS 查询失败、拒绝连接等）时，Requests 会抛出一个 ConnectionError 异常。

如果 HTTP 请求返回了不成功的状态码， Response.raise_for_status() 会抛出一个 HTTPError 异常。

若请求超时，则抛出一个 Timeout 异常。

若请求超过了设定的最大重定向次数，则会抛出一个 TooManyRedirects 异常。

所有Requests显式抛出的异常都继承自 requests.exceptions.RequestException 。