def test_log_backpressure_info(self): """ When backpressure is hit, DEBUG and INFO logs will be shed. """ handler = RemoteHandler("127.0.0.1", 9000, maximum_buffer=10, _reactor=self.reactor) logger = self.get_logger(handler) # Send some debug messages for i in range(0, 3): logger.debug("debug %s" % (i, )) # Send a bunch of useful messages for i in range(0, 10): logger.warning("warn %s" % (i, )) # Send a bunch of info messages for i in range(0, 3): logger.info("info %s" % (i, )) # The last debug message pushes it past the maximum buffer logger.debug("too much debug") # Allow the reconnection client, server = connect_logging_client(self.reactor, 0) client.transport.flush() # The 10 warnings made it through, the debugs and infos were elided logs = server.data.splitlines() self.assertEqual(len(logs), 10) self.assertNotIn(b"debug", server.data) self.assertNotIn(b"info", server.data)
def test_cancel_connection(self): """ Gracefully handle the connection being cancelled. """ handler = RemoteHandler("127.0.0.1", 9000, maximum_buffer=10, _reactor=self.reactor) logger = self.get_logger(handler) # Send a message. logger.info("Hello there, %s!", "wally") # Do not accept the connection and shutdown. This causes the pending # connection to be cancelled (and should not raise any exceptions). handler.close()
def test_log_backpressure_cut_middle(self): """ When backpressure is hit, and no more DEBUG and INFOs cannot be culled, it will cut the middle messages out. """ handler = RemoteHandler("127.0.0.1", 9000, maximum_buffer=10, _reactor=self.reactor) logger = self.get_logger(handler) # Send a bunch of useful messages for i in range(0, 20): logger.warning("warn %s" % (i, )) # Allow the reconnection client, server = connect_logging_client(self.reactor, 0) client.transport.flush() # The first five and last five warnings made it through, the debugs and # infos were elided logs = server.data.decode("utf8").splitlines() self.assertEqual( ["warn %s" % (i, ) for i in range(5)] + ["warn %s" % (i, ) for i in range(15, 20)], logs, )
def test_log_output(self): """ The remote handler delivers logs over TCP. """ handler = RemoteHandler("127.0.0.1", 9000, _reactor=self.reactor) logger = self.get_logger(handler) logger.info("Hello there, %s!", "wally") # Trigger the connection client, server = connect_logging_client(self.reactor, 0) # Trigger data being sent client.transport.flush() # One log message, with a single trailing newline logs = server.data.decode("utf8").splitlines() self.assertEqual(len(logs), 1) self.assertEqual(server.data.count(b"\n"), 1) # Ensure the data passed through properly. self.assertEqual(logs[0], "Hello there, wally!")