Question

我继承了以下Django视图代码，由另一个webservice用于提供可下载的输出数据版本：

def index(request):
    # ... (snip) ...
    data = base64.decodestring(request.POST['data'])
    filename = request.POST['filename']

    wrapper = FileWrapper(StringIO(data))

    response = HttpResponse(wrapper, content_type=guess_type(str(filename))[0])

    response['Content-Length'] = len(data)
    response['Content-Disposition'] = "attachment; filename=" + filename

    return response

函数本身 - 针对Django 1.0编写 - 在升级到1.5后仍能正常工作。不幸的是，涵盖此视图的测试现在失败了：

    def testDownload(self):
        self.client.login(username='test', password='test')

        real = 'abc' * 100
        data = base64.encodestring(real)
        response = self.client.post("/api/v1/download/", {'data': data, 'filename': 'out.jpg'})

        self.assertEqual(real, response.content)
        self.assertEqual(response['Content-Disposition'], 'attachment; filename=out.jpg')

和错误：

Traceback (most recent call last):
  File "/home/fred/.secret_projects/final/gerbils/tests/amf.py", line 548, in testDownload
    self.assertEqual(real, response.content)
  File "/home/fred/.virtualenvs/cunning_plot/lib/python2.7/site-packages/django/http/response.py", line 282, in content
    self._consume_content()
  File "/home/carl/.virtualenvs/cunning_plot/lib/python2.7/site-packages/django/http/response.py", line 278, in _consume_content
    self.content = b''.join(self.make_bytes(e) for e in self._container)
  File "/home/carl/.virtualenvs/cunning_plot/lib/python2.7/site-packages/django/http/response.py", line 278, in <genexpr>
    self.content = b''.join(self.make_bytes(e) for e in self._container)
  File "/usr/lib64/python2.7/wsgiref/util.py", line 30, in next
    data = self.filelike.read(self.blksize)
  File "/usr/lib64/python2.7/StringIO.py", line 127, in read
    _complain_ifclosed(self.closed)
  File "/usr/lib64/python2.7/StringIO.py", line 40, in _complain_ifclosed
    raise ValueError, "I/O operation on closed file"
ValueError: I/O operation on closed file

那么......有什么想法吗？我无法在testDownload()或index()中看到任何必须＆＃34;关闭＆＃34;在需要阅读之前StringIO。如果有什么东西，它也不会影响非测试情况吗？

非常困惑。帮助赞赏。

Answer 1

查看调用close的位置的简单方法是将StringIO子类化，并在close函数中放置断点。

class CustomStringIO(StringIO):
    def close(self):
        import pdb; pdb.set_trace()
        super(CustomStringIO, self).close()

这个堆栈是

-> response = self.client.post("/test/", {'data': data, 'filename': 'out.jpg'})
  ...\venv\lib\site-packages\django\test\client.py(463)post()
-> response = super(Client, self).post(path, data=data, content_type=content_type, **extra)
  ...\venv\lib\site-packages\django\test\client.py(297)post()
-> return self.request(**r)
  ...\venv\lib\site-packages\django\test\client.py(406)request()
-> response = self.handler(environ)
  ...\venv\lib\site-packages\django\test\client.py(119)__call__()
-> response.close()                    # will fire request_finished
  ...\venv\lib\site-packages\django\http\response.py(233)close()
-> closable.close()
> \testapp\views.py(11)close()
-> super(CustomStringIO, self).close()

看起来测试客户端正在关闭响应，然后在FileWrapper上调用close，然后在StringIO上调用close。这是在您实际到达response.content之前的所有内容。

您需要FileWrapper吗？当HttpResponse接收字符串内容而base64.decodestring返回二进制字符串时，您似乎可以直接将data传递给HttpResponse，而不必创建StringIO和FileWrapper。

在Django视图测试中使用StringIO进行“关闭文件的I / O操作”

1 个答案: