def test_process_greenlets():
from shadow.proxy.web import ProxyFlask
from shadow.proxy.web import AbstractResultsLogger
import gevent
svc = mock.Mock()
mock_result_logger = mock.Mock(spec=AbstractResultsLogger)
app = ProxyFlask(svc, ['localhost'], ['localhost'],
result_loggers=[mock_result_logger])
req, expected_req = _mock_request()
resp, expected_resp, elapsed_time = _mock_response()
resp2, expected_resp2, elapsed_time2 = _mock_response()
ident = lambda x: x
greenlets = [
gevent.spawn(ident, x)
for x in [(resp, elapsed_time), (resp2, elapsed_time2)]
]
app.process_greenlets(
app.format_request(req.path, req.method, req.headers, req.args,
req.form, req), greenlets)
mock_result_logger.log_result.assert_called_with({
'request': {
'original': expected_req,
'modified': expected_req
},
'results': [expected_resp, expected_resp2]
})
def test_format_request():
from shadow.proxy.web import ProxyFlask
svc = mock.Mock()
app = ProxyFlask(
svc,
['localhost'],
['localhost']
)
req, expected = _mock_request()
eq_(app.format_request(req.path, req.method, req.headers, req.args, req.form, req), {'original': expected, 'modified': expected})
def test_format_request():
from shadow.proxy.web import ProxyFlask
svc = mock.Mock()
app = ProxyFlask(svc, ['localhost'], ['localhost'])
req, expected = _mock_request()
eq_(
app.format_request(req.path, req.method, req.headers, req.args,
req.form, req), {
'original': expected,
'modified': expected
})
def test_process_greenlets():
from shadow.proxy.web import ProxyFlask
from shadow.proxy.web import AbstractResultsLogger
import gevent
svc = mock.Mock()
mock_result_logger = mock.Mock(spec=AbstractResultsLogger)
app = ProxyFlask(
svc,
['localhost'],
['localhost'],
result_loggers=[mock_result_logger]
)
req, expected_req = _mock_request()
resp, expected_resp, elapsed_time = _mock_response()
resp2, expected_resp2, elapsed_time2 = _mock_response()
ident = lambda x: x
greenlets = [gevent.spawn(ident, x) for x in [(resp, elapsed_time), (resp2, elapsed_time2)]]
app.process_greenlets(app.format_request(req.path, req.method, req.headers, req.args, req.form, req), greenlets)
mock_result_logger.log_result.assert_called_with({
'request': {
'original': expected_req,
'modified': expected_req
},
'results': [
expected_resp,
expected_resp2
]
})
