mirror of
https://github.com/webrecorder/pywb.git
synced 2025-03-21 11:02:10 +01:00
72 lines
2.3 KiB
Python
72 lines
2.3 KiB
Python
|
from webagg.inputrequest import DirectWSGIInputRequest, POSTInputRequest
|
||
|
|
from bottle import Bottle, request, response
|
||
|
|
import webtest
|
||
|
|
import traceback
|
||
|
|
|
||
|
|
|
||
|
|
#=============================================================================
|
||
|
|
class InputReqApp(object):
|
||
|
|
def __init__(self):
|
||
|
|
self.application = Bottle()
|
||
|
|
self.application.default_error_handler = self.err_handler
|
||
|
|
|
||
|
|
@self.application.route('/test/<url:re:.*>', 'ANY')
|
||
|
|
def direct_input_request(url=''):
|
||
|
|
inputreq = DirectWSGIInputRequest(request.environ)
|
||
|
|
response['Content-Type'] = 'text/plain; charset=utf-8'
|
||
|
|
return inputreq.reconstruct_request(url)
|
||
|
|
|
||
|
|
@self.application.route('/test-postreq', 'POST')
|
||
|
|
def post_fullrequest():
|
||
|
|
params = dict(request.query)
|
||
|
|
inputreq = POSTInputRequest(request.environ)
|
||
|
|
response['Content-Type'] = 'text/plain; charset=utf-8'
|
||
|
|
return inputreq.reconstruct_request(params.get('url'))
|
||
|
|
|
||
|
|
def err_handler(self, out):
|
||
|
|
print(out)
|
||
|
|
traceback.print_exc()
|
||
|
|
|
||
|
|
|
||
|
|
#=============================================================================
|
||
|
|
class TestInputReq(object):
|
||
|
|
def setup(self):
|
||
|
|
self.app = InputReqApp()
|
||
|
|
self.testapp = webtest.TestApp(self.app.application)
|
||
|
|
|
||
|
|
def test_get_direct(self):
|
||
|
|
res = self.testapp.get('/test/http://example.com/', headers={'Foo': 'Bar'})
|
||
|
|
assert res.text == '\
|
||
|
|
GET /test/http://example.com/ HTTP/1.0\r\n\
|
||
|
|
Host: example.com\r\n\
|
||
|
|
Foo: Bar\r\n\
|
||
|
|
\r\n\
|
||
|
|
'
|
||
|
|
|
||
|
|
def test_post_direct(self):
|
||
|
|
res = self.testapp.post('/test/http://example.com/', headers={'Foo': 'Bar'}, params='ABC')
|
||
|
|
lines = res.text.split('\r\n')
|
||
|
|
assert lines[0] == 'POST /test/http://example.com/ HTTP/1.0'
|
||
|
|
assert 'Host: example.com' in lines
|
||
|
|
assert 'Content-Length: 3' in lines
|
||
|
|
assert 'Content-Type: application/x-www-form-urlencoded' in lines
|
||
|
|
assert 'Foo: Bar' in lines
|
||
|
|
|
||
|
|
assert 'ABC' in lines
|
||
|
|
|
||
|
|
def test_post_req(self):
|
||
|
|
postdata = '\
|
||
|
|
GET /example.html HTTP/1.0\r\n\
|
||
|
|
Foo: Bar\r\n\
|
||
|
|
\r\n\
|
||
|
|
'
|
||
|
|
res = self.testapp.post('/test-postreq?url=http://example.com/', params=postdata)
|
||
|
|
|
||
|
|
assert res.text == '\
|
||
|
|
GET /example.html HTTP/1.0\r\n\
|
||
|
|
Host: example.com\r\n\
|
||
|
|
Foo: Bar\r\n\
|
||
|
|
\r\n\
|
||
|
|
'
|
||
|
|
|