class DecompressionMiddlewareTest(TestCase): test_formats = ['tar', 'xml.bz2', 'xml.gz', 'zip'] uncompressed_body, test_responses = _test_data(test_formats) def setUp(self): self.mw = DecompressionMiddleware() self.spider = BaseSpider('foo') def test_known_compression_formats(self): for fmt in self.test_formats: rsp = self.test_responses[fmt] new = self.mw.process_response(None, rsp, self.spider) assert isinstance(new, XmlResponse), \ 'Failed %s, response type %s' % (fmt, type(new).__name__) assert_samelines(self, new.body, self.uncompressed_body, fmt) def test_plain_response(self): rsp = Response(url='http://test.com', body=self.uncompressed_body) new = self.mw.process_response(None, rsp, self.spider) assert new is rsp assert_samelines(self, new.body, rsp.body) def test_empty_response(self): rsp = Response(url='http://test.com', body='') new = self.mw.process_response(None, rsp, self.spider) assert new is rsp assert not rsp.body assert not new.body def tearDown(self): del self.mw
class DecompressionMiddlewareTest(TestCase): test_formats = ["tar", "xml.bz2", "xml.gz", "zip"] uncompressed_body, test_responses = _test_data(test_formats) def setUp(self): self.mw = DecompressionMiddleware() self.spider = BaseSpider("foo") def test_known_compression_formats(self): for fmt in self.test_formats: rsp = self.test_responses[fmt] new = self.mw.process_response(None, rsp, self.spider) assert isinstance(new, XmlResponse), "Failed %s, response type %s" % (fmt, type(new).__name__) self.assertEqual(new.body, self.uncompressed_body, fmt) def test_plain_response(self): rsp = Response(url="http://test.com", body=self.uncompressed_body) new = self.mw.process_response(None, rsp, self.spider) assert new is rsp self.assertEqual(new.body, rsp.body) def test_empty_response(self): rsp = Response(url="http://test.com", body="") new = self.mw.process_response(None, rsp, self.spider) assert new is rsp assert not rsp.body assert not new.body def tearDown(self): del self.mw
def setUp(self): self.mw = DecompressionMiddleware() self.spider = BaseSpider('foo')