Tests: use "data" parameter in Request() to send body.

This commit is contained in:
Andrey Zelenkov
2018-01-15 19:59:47 +03:00
parent 263741fb44
commit 819b43fc2d
2 changed files with 5 additions and 8 deletions

View File

@@ -59,9 +59,8 @@ def application(environ, start_response):
r = unit.TestUnitHTTP.post(headers={
'Host': 'localhost',
'Content-Type': 'text/html',
'Content-Length': str(len(body)),
'Custom-Header': 'blah'
}, body=body)
}, data=body)
self.assertEqual(r.status_code, 200, 'status')
self.assertEqual(r.headers['Content-Length'], str(len(body)),

View File

@@ -215,15 +215,13 @@ class TestUnitHTTP():
host = '127.0.0.1:7080' if 'host' not in kwargs else kwargs['host']
uri = '/' if 'uri' not in kwargs else kwargs['uri']
sess = Session() if 'sess' not in kwargs else kwargs['sess']
body = None if 'body' not in kwargs else kwargs['body']
data = None if 'data' not in kwargs else kwargs['data']
headers = None if 'headers' not in kwargs else kwargs['headers']
req = Request('POST', 'http://' + host + uri, headers=headers)
prepped = req.prepare()
req = Request(method, 'http://' + host + uri, data=data,
headers=headers)
prepped.body = body
r = sess.send(prepped)
r = sess.send(req.prepare())
if 'keep' not in kwargs:
sess.close()