def test_css_detect(self):
self.assertTrue(
CSSReader.is_file(
io.BytesIO('body { color: white }'.encode('utf-16le'))))
self.assertFalse(
CSSReader.is_file(io.BytesIO('hello world!'.encode('utf-16le'))))
self.assertFalse(CSSReader.is_file(io.BytesIO(b'<html><body>hello')))
self.assertTrue(
CSSReader.is_file(io.BytesIO(b'<html><body>hello')) is VeryFalse)
self.assertTrue(
CSSReader.is_file(io.BytesIO(b'h1 { background-color: red }')))
self.assertTrue(CSSReader.is_file(io.BytesIO(b'@import url.css;')))
self.assertTrue(
CSSReader.is_url(URLInfo.parse('example.com/index.css')))
self.assertFalse(
CSSReader.is_url(URLInfo.parse('example.com/image.jpg')))
self.assertTrue(CSSReader.is_request(Request('example.com/index.css')))
self.assertFalse(CSSReader.is_request(
Request('example.com/image.jpg')))
response = Response(200, 'OK')
response.fields['Content-Type'] = 'text/css'
self.assertTrue(CSSReader.is_response(response))
response = Response(200, 'OK')
response.fields['Content-Type'] = 'image/png'
self.assertFalse(CSSReader.is_response(response))
def test_css_detect(self):
self.assertTrue(CSSReader.is_file(
io.BytesIO('body { color: white }'.encode('utf-16le'))
))
self.assertFalse(CSSReader.is_file(
io.BytesIO('hello world!'.encode('utf-16le'))
))
self.assertFalse(CSSReader.is_file(
io.BytesIO(b'<html><body>hello')
))
self.assertTrue(CSSReader.is_file(
io.BytesIO(b'<html><body>hello')
) is VeryFalse)
self.assertTrue(CSSReader.is_file(
io.BytesIO(b'h1 { background-color: red }')
))
self.assertTrue(CSSReader.is_file(
io.BytesIO(b'@import url.css;')
))
self.assertTrue(
CSSReader.is_url(URLInfo.parse('example.com/index.css'))
)
self.assertFalse(
CSSReader.is_url(URLInfo.parse('example.com/image.jpg'))
)
self.assertTrue(
CSSReader.is_request(Request('example.com/index.css'))
)
self.assertFalse(
CSSReader.is_request(Request('example.com/image.jpg'))
)
response = Response(200, 'OK')
response.fields['Content-Type'] = 'text/css'
self.assertTrue(CSSReader.is_response(response))
response = Response(200, 'OK')
response.fields['Content-Type'] = 'image/png'
self.assertFalse(CSSReader.is_response(response))
def _append_filename_extension(self, response):
'''Append an HTML/CSS file suffix as needed.'''
if not self._filename:
return
if response.request.url_info.scheme not in ('http', 'https'):
return
if not re.search(r'\.[hH][tT][mM][lL]?$', self._filename) and \
HTMLReader.is_response(response):
self._filename += '.html'
elif not re.search(r'\.[cC][sS][sS]$', self._filename) and \
CSSReader.is_response(response):
self._filename += '.css'
def _append_filename_extension(self, response: BaseResponse):
'''Append an HTML/CSS file suffix as needed.'''
if not self._filename:
return
if response.request.url_info.scheme not in ('http', 'https'):
return
if not re.search(r'\.[hH][tT][mM][lL]?$', self._filename) and \
HTMLReader.is_response(response):
self._filename += '.html'
elif not re.search(r'\.[cC][sS][sS]$', self._filename) and \
CSSReader.is_response(response):
self._filename += '.css'
