def setUp(self): self.caster = MagicMock() self.handler = transport.SyslogToZeroMQHandler(self.caster) self.msg_head = SyslogMessageHead() self.msg_part_1 = "Part 1 " self.msg_part_2 = "Part 2 " self.msg_part_3 = "Part 3 " self.test_message = self.msg_part_1 + self.msg_part_2 + self.msg_part_3 self.msg_length = 127 self.final_message = self.msg_head.as_dict() self.final_message['message'] = self.test_message self.final_message['msg_length'] = self.msg_length
def setUp(self): self.host = '127.0.0.1' self.port = '5000' self.host_tuple = (self.host, self.port) self.connect_host_tuples = [self.host_tuple] self.msg_head = SyslogMessageHead() self.test_message = "test_message" self.msg_length = 127 self.final_message = self.msg_head.as_dict() self.final_message['message'] = self.test_message self.final_message['msg_length'] = self.msg_length self.final_message_json = simplejson.dumps(self.final_message)
def build_header(self, app_name, host_name, message_id, priority, process_id, version, timestamp, sd): """ Builds a Syslog Message Header Object Also handles replacing passed None's with syslog "nil" values. """ head = SyslogMessageHead() head.appname = app_name or '-' head.hostname = host_name or '-' head.messageid = message_id or '-' head.priority = priority or '-' head.processid = process_id or '-' head.timestamp = timestamp or '-' head.version = version or '-' head.sd = sd or {} return head
class WhenIntegrationTestingTransport(unittest.TestCase): def setUp(self): self.host = '127.0.0.1' self.port = '5000' self.host_tuple = (self.host, self.port) self.connect_host_tuples = [self.host_tuple] self.msg_head = SyslogMessageHead() self.test_message = "test_message" self.msg_length = 127 self.final_message = self.msg_head.as_dict() self.final_message['message'] = self.test_message self.final_message['msg_length'] = self.msg_length self.final_message_json = simplejson.dumps(self.final_message) def test_message_transport_over_zmq(self): self.caster = transport.ZeroMQCaster(self.host_tuple) self.handler = transport.SyslogToZeroMQHandler(self.caster) self.receiver = transport.ZeroMQReceiver(self.connect_host_tuples) self.receiver.connect() self.handler.on_msg_head(self.msg_head) self.handler.on_msg_part(self.test_message) self.handler.on_msg_complete(self.msg_length) rcvd_msg = self.receiver.get() self.assertEqual(rcvd_msg, self.final_message_json) def tearDown(self): self.caster.close() self.receiver.close()
class WhenTestingSyslogToZeroMQHandler(unittest.TestCase): def setUp(self): self.caster = MagicMock() self.handler = transport.SyslogToZeroMQHandler(self.caster) self.msg_head = SyslogMessageHead() self.msg_part_1 = "Part 1 " self.msg_part_2 = "Part 2 " self.msg_part_3 = "Part 3 " self.test_message = self.msg_part_1 + self.msg_part_2 + self.msg_part_3 self.msg_length = 127 self.final_message = self.msg_head.as_dict() self.final_message['message'] = self.test_message self.final_message['msg_length'] = self.msg_length def test_constructor(self): self.assertIsInstance(self.handler.msg, bytearray) self.assertIsNone(self.handler.msg_head) self.assertEqual(self.handler.caster, self.caster) self.assertTrue(self.handler.caster.bound) def test_on_msg_head(self): self.handler.on_msg_head(self.msg_head) self.assertEqual(self.handler.msg_head, self.msg_head) def test_on_msg_part(self): self.handler.on_msg_part(self.msg_part_1) self.assertEqual( self.handler.msg, bytearray(self.msg_part_1) ) self.handler.on_msg_part(self.msg_part_2) self.assertEqual( self.handler.msg, bytearray(self.msg_part_1 + self.msg_part_2) ) self.handler.on_msg_part(self.msg_part_3) self.assertEqual( self.handler.msg, bytearray(self.msg_part_1 + self.msg_part_2 + self.msg_part_3) ) def test_on_msg_complete(self): self.handler.on_msg_head(self.msg_head) self.handler.on_msg_part(self.msg_part_1) self.handler.on_msg_part(self.msg_part_2) self.handler.on_msg_part(self.msg_part_3) self.handler.on_msg_complete(self.msg_length) self.caster.cast.assert_called_once_with( simplejson.dumps(self.final_message)) self.assertEqual(self.handler.msg, b'')