我使用Robot Framework和robotframework-requests运行测试时遇到问题。我需要在正文中发送POST请求和二进制数据。我已经看了this question,但它没有真正回答。以下是我的测试用例的样子:
Upload ${filename} file
Create Session mysession http://${ADDRESS}
${data} = Get Binary File ${filename}
&{headers} = Create Dictionary Content-Type=application/octet-stream Accept=application/octet-stream
${resp} = Post Request mysession ${CGIPath} data=${data} headers=&{headers}
[Return] ${resp.status_code} ${resp.text}
问题是我的二进制数据大约是250MB。当用Get Binary File
读取数据时,我发现内存消耗高达2.x GB。几秒钟后,当Post Request
被触发时,我的测试被OOM杀死。我已经看过files
参数,但似乎它使用了多部分编码上传,这不是我需要的。
我的另一个想法是将开放文件处理程序直接传递给底层请求库,但我想这需要修改robotframework-request。另一个想法是只回到卷曲这个测试。
我在测试中遗漏了什么吗?解决这个问题的更好方法是什么?
我继续进行机器人框架 - 请求修改的想法,并添加了这种方法
def post_request_binary(
self,
alias,
uri,
path=None,
params=None,
headers=None,
allow_redirects=None,
timeout=None):
session = self._cache.switch(alias)
redir = True if allow_redirects is None else allow_redirects
self._capture_output()
method_name = "post"
method = getattr(session, method_name)
with open(path, 'rb') as f:
resp = method(self._get_url(session, uri),
data=f,
params=self._utf8_urlencode(params),
headers=headers,
allow_redirects=allow_redirects,
timeout=self._get_timeout(timeout),
cookies=self.cookies,
verify=self.verify)
self._print_debug()
# Store the last session object
session.last_resp = resp
self.builtin.log(method_name + ' response: ' + resp.text, 'DEBUG')
return resp
我想我可以稍微改进一下并创建一个拉取请求。