def test_handshakeFailed_does_not_log_when_connection_is_closed(self):
server = RegionServer()
with TwistedLoggerFixture() as logger:
server.handshakeFailed(Failure(ConnectionClosed()))
# Nothing was logged.
self.assertEqual("", logger.output)
def test_connection_closed(test_client, mock_treq):
mock_treq.post.side_effect = ConnectionClosed()
output = yield test_client.request("POST", "/test/uri", SOME_JSON)
assert output == ConnectionClosed