def send(self, s): if len(s) <= self.chunk_size: DatagramHandler.send(self, s) else: chunks = gelf.split(s, self.chunk_size) for chunk in chunks: DatagramHandler.send(self, chunk)
def send(self, s): if len(s) <= self.chunk_size: super(GelfUdpHandler, self).send(s) else: chunks = gelf.split(s, self.chunk_size) for chunk in chunks: super(GelfUdpHandler, self).send(chunk)
def test_split(): message = b'12345' header = b'\x1e\x0f' chunks = list(gelf.split(message, 2)) expected = [(struct.pack('b', 0), struct.pack('b', 3), b'12'), (struct.pack('b', 1), struct.pack('b', 3), b'34'), (struct.pack('b', 2), struct.pack('b', 3), b'5')] assert len(chunks) == len(expected) for index, chunk in enumerate(chunks): expected_index, expected_chunks_count, expected_chunk = expected[index] assert chunk[:2] == header assert chunk[10:11] == expected_index assert chunk[11:12] == expected_chunks_count assert chunk[12:] == expected_chunk
def test_split(): message = b'12345' header = b'\x1e\x0f' chunks = list(gelf.split(message, 2)) expected = [ (struct.pack('b', 0), struct.pack('b', 3), b'12'), (struct.pack('b', 1), struct.pack('b', 3), b'34'), (struct.pack('b', 2), struct.pack('b', 3), b'5') ] assert len(chunks) == len(expected) for index, chunk in enumerate(chunks): expected_index, expected_chunks_count, expected_chunk = expected[index] assert chunk[:2] == header assert chunk[10:11] == expected_index assert chunk[11:12] == expected_chunks_count assert chunk[12:] == expected_chunk