def test_after_receive_response(statsd_hook, request):
response = Response(code=StatusCode.ok)
statsd_hook.after_receive_response(request, response)
statsd_hook._statsd.count.assert_called_with(
"tchannel.outbound.calls.success.no-service.test.endpoint1", 1)
response = Response(code=StatusCode.error)
statsd_hook.after_receive_response(request, response)
statsd_hook._statsd.count.assert_called_with(
"tchannel.outbound.calls.app-errors.no-service.test.endpoint1", 1)
def test_build_raw_response_message():
message_factory = MessageFactory()
resp = Response(
flags=FlagsType.none,
code=StatusCode.ok,
headers={},
id=1111,
)
resp.state = StreamState.init
message = message_factory.build_raw_response_message(resp, None, True)
assert message.code == resp.code
assert message.flags == resp.flags
assert message.id == resp.id
assert message.headers == resp.headers
def test_response_exception():
resp = Response()
yield resp.write_body("aaa")
with pytest.raises(StreamingError):
yield resp.write_header("aaa")
resp.flush()
with pytest.raises(TChannelError):
yield resp.write_body("aaaa")
