mirror of
https://github.com/webrecorder/pywb.git
synced 2025-03-22 14:24:27 +01:00
97182b71b7
- merge pywb.urlrewrite -> pywb.rewrite, remove obsolete stuff (rewrite_content.py, rewrite_live.py, dsrules.py) - move wbrequestresponse -> pywb.apps - move pywb.webapp.handlers -> pywb.apps.static_handler - remove pywb.webapp, pywb.framework packages - disable old header_rewriter, content_rewriter tests - finish renaming from previous warcserver refactor - all other tests passing!
47 lines
1.6 KiB
Python
47 lines
1.6 KiB
Python
from pywb.apps.wbrequestresponse import WbResponse
|
|
from warcio.statusandheaders import StatusAndHeaders
|
|
|
|
|
|
def test_resp_1():
|
|
resp = vars(WbResponse.text_response('Test'))
|
|
|
|
expected = {'body': [b'Test'], 'status_headers': StatusAndHeaders(protocol = '', statusline = '200 OK',
|
|
headers = [('Content-Type', 'text/plain; charset=utf-8'), ('Content-Length', '4')])}
|
|
|
|
assert(resp == expected)
|
|
|
|
|
|
def test_resp_2():
|
|
resp = vars(WbResponse.bin_stream([b'Test', b'Another'], content_type='text/plain; charset=utf-8', status='404'))
|
|
|
|
expected = {'body': [b'Test', b'Another'], 'status_headers': StatusAndHeaders(protocol = '', statusline = '404',
|
|
headers = [('Content-Type', 'text/plain; charset=utf-8')])}
|
|
|
|
assert(resp == expected)
|
|
|
|
def test_resp_3():
|
|
|
|
resp = vars(WbResponse.redir_response('http://example.com/otherfile'))
|
|
|
|
expected = {'body': [], 'status_headers': StatusAndHeaders(protocol = '', statusline = '302 Redirect',
|
|
headers = [('Location', 'http://example.com/otherfile'), ('Content-Length', '0')])}
|
|
|
|
assert(resp == expected)
|
|
|
|
def test_resp_4():
|
|
resp = vars(WbResponse.text_response('Test').add_range(10, 4, 100))
|
|
|
|
expected = {'body': [b'Test'], 'status_headers': StatusAndHeaders(protocol = '', statusline = '206 Partial Content',
|
|
headers = [ ('Content-Type', 'text/plain; charset=utf-8'),
|
|
('Content-Length', '4'),
|
|
('Content-Range', 'bytes 10-13/100'),
|
|
('Accept-Ranges', 'bytes')])}
|
|
|
|
assert(resp == expected)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
import doctest
|
|
doctest.testmod()
|
|
|