def test_register():
importlib.reload(web)
importlib.reload(sprunge)
sprunge.register()
assert web.pastebins.get('sprunge') is not None
sprunge.unregister()
assert web.pastebins.get('sprunge') is None
def test_paste_bytes(mock_requests):
importlib.reload(web)
importlib.reload(sprunge)
sprunge.register()
paster = web.pastebins['sprunge']
mock_requests.add(
'POST', 'http://sprunge.us', body='http://sprunge.us/foobar'
)
assert paster.paste(b'test data', 'txt') == 'http://sprunge.us/foobar?txt'
def test_paste_error(mock_requests):
importlib.reload(web)
importlib.reload(sprunge)
sprunge.register()
paster = web.pastebins['sprunge']
with pytest.raises(web.ServiceError):
paster.paste('test data', 'txt')
mock_requests.add(
'POST', 'http://sprunge.us', status=500,
)
with pytest.raises(web.ServiceHTTPError):
paster.paste('test data', 'txt')
def test_data_params(mock_requests):
importlib.reload(web)
importlib.reload(sprunge)
sprunge.register()
body = None
def req_cb(req):
nonlocal body
body = req.body
return 200, {}, 'http://sprunge.us/foobar\n'
paster = web.pastebins['sprunge']
mock_requests.add_callback(
'POST', 'http://sprunge.us', callback=req_cb
)
assert paster.paste('test data', 'txt') == 'http://sprunge.us/foobar?txt'
assert body == 'sprunge=test+data'