class EventTest(unittest.TestCase): Type = "BasicFlow" Version = "v01" Flow_id = "Id01" Owner = "Owner" Enabled = True Priority = "65535" Status = "none" Attributes = {"bandwidth": "10Mbps", "req_bandwidth": "11Mbps", "latency": "10msec", "req_latency": "11msec"} Publisher_id = "Id1" Event_type = "flowchanged" flow_target = Flow(Type, Version, Flow_id, Owner, Enabled, Priority, Status, Attributes) flow_target_packed_object = flow_target.packed_object() def setUp(self): self.target = Event(self.Publisher_id, self.Event_type, self.flow_target) def tearDown(self): self.target = None def test_constructor(self): self.assertEqual(self.target.publisher_id, self.Publisher_id) self.assertEqual(self.target.event_type, self.Event_type) self.assertEqual(self.target.body, self.flow_target) def test_create_from_packed(self): Type = "BasicFlow" Version = "v02" Flow_id = "Id02" Owner = "Owner" Enabled = True Priority = "65535" Status = "none" Attributes = {"bandwidth": "20Mbps", "req_bandwidth": "11Mbps", "latency": "30msec", "req_latency": "11msec"} Publisher_id = "Id1" Event_type = "flowchanged" flow_target = Flow(Type, Version, Flow_id, Owner, Enabled, Priority, Status, Attributes) value = [Publisher_id, Event_type, flow_target] self.result = self.target.create_from_packed(value) self.assertEqual(self.result.publisher_id, self.Publisher_id) self.assertEqual(self.result.event_type, self.Event_type) self.assertEqual(self.result.body, flow_target) def test_packed_object_hasattr_True(self): self.result = self.target.packed_object() self.assertEqual(self.result[0], self.Publisher_id) self.assertEqual(self.result[1], self.Event_type) self.assertEqual(self.result[2], self.flow_target_packed_object) def test_packed_object_hasattr_False(self): self.newtarget = Event(self.Publisher_id, self.Event_type, self.flow_target_packed_object) self.result = self.newtarget.packed_object() self.assertEqual(self.result[0], self.Publisher_id) self.assertEqual(self.result[1], self.Event_type) self.assertEqual(self.result[2], self.flow_target_packed_object)
def __redisSubscriberRunnable(self): for mesg in self.__redisSubscriber.listen(): if mesg['type'] != 'message': continue try: bio = BytesIO() bio.write(mesg['data']) bio.seek(0) upk = msgpack.Unpacker(bio) tp = upk.unpack() sno = upk.unpack() srcid = upk.unpack() if tp == self.TYPE_REQUEST: request = Request.create_from_packed(upk.unpack()) def request_runnable(request, sno, srcid): response = self.dispatch_request(request) pk = msgpack.Packer() resb = bytearray() resb.extend(pk.pack(self.TYPE_RESPONSE)) resb.extend(pk.pack(sno)) resb.extend(pk.pack(request.object_id)) resb.extend(pk.pack(response.packed_object())) self.__pubsqueue.put( MessageDispatcher.PublishData(None, self.TYPE_RESPONSE, sno, srcid, resb)) self.thread_pool.submit(request_runnable, request, sno, srcid) elif tp == self.TYPE_RESPONSE: trans = self.__clients[srcid] response = Response.create_from_packed(upk.unpack()) trans.signalResponse(sno, response) elif tp == self.TYPE_EVENT: event = Event.create_from_packed(upk.unpack()) def event_runnable(event): self.dispatch_event(event) self.thread_pool.submit(event_runnable, event) except: logging.error(traceback.format_exc()) pass
def __redisSubscriberRunnable(self): for mesg in self.__redisSubscriber.listen(): if mesg['type'] != 'message': continue try: bio = BytesIO() bio.write(mesg['data']) bio.seek(0) upk = msgpack.Unpacker(bio) tp = upk.unpack() sno = upk.unpack() srcid = upk.unpack() if tp == self.TYPE_REQUEST: request = Request.create_from_packed(upk.unpack()) def request_runnable(request, sno, srcid): response = self.dispatch_request(request) pk = msgpack.Packer() resb = bytearray() resb.extend(pk.pack(self.TYPE_RESPONSE)) resb.extend(pk.pack(sno)) resb.extend(pk.pack(request.object_id)) resb.extend(pk.pack(response.packed_object())) self.__pubsqueue.put( MessageDispatcher.PublishData( None, self.TYPE_RESPONSE, sno, srcid, resb)) self.thread_pool.submit(request_runnable, request, sno, srcid) elif tp == self.TYPE_RESPONSE: trans = self.__clients[srcid] response = Response.create_from_packed(upk.unpack()) trans.signalResponse(sno, response) elif tp == self.TYPE_EVENT: event = Event.create_from_packed(upk.unpack()) def event_runnable(event): self.dispatch_event(event) self.thread_pool.submit(event_runnable, event) except: logging.error(traceback.format_exc()) pass
class EventTest(unittest.TestCase): Type = "BasicFlow" Version = "v01" Flow_id = "Id01" Owner = "Owner" Enabled = True Priority = "65535" Status = "none" Attributes = { "bandwidth": "10Mbps", "req_bandwidth": "11Mbps", "latency": "10msec", "req_latency": "11msec" } Publisher_id = 'Id1' Event_type = 'flowchanged' flow_target = Flow(Type, Version, Flow_id, Owner, Enabled, Priority, Status, Attributes) flow_target_packed_object = flow_target.packed_object() def setUp(self): self.target = Event(self.Publisher_id, self.Event_type, self.flow_target) def tearDown(self): self.target = None def test_constructor(self): self.assertEqual(self.target.publisher_id, self.Publisher_id) self.assertEqual(self.target.event_type, self.Event_type) self.assertEqual(self.target.body, self.flow_target) def test_create_from_packed(self): Type = "BasicFlow" Version = "v02" Flow_id = "Id02" Owner = "Owner" Enabled = True Priority = "65535" Status = "none" Attributes = { "bandwidth": "20Mbps", "req_bandwidth": "11Mbps", "latency": "30msec", "req_latency": "11msec" } Publisher_id = 'Id1' Event_type = 'flowchanged' flow_target = Flow(Type, Version, Flow_id, Owner, Enabled, Priority, Status, Attributes) value = [Publisher_id, Event_type, flow_target] self.result = self.target.create_from_packed(value) self.assertEqual(self.result.publisher_id, self.Publisher_id) self.assertEqual(self.result.event_type, self.Event_type) self.assertEqual(self.result.body, flow_target) def test_packed_object_hasattr_True(self): self.result = self.target.packed_object() self.assertEqual(self.result[0], self.Publisher_id) self.assertEqual(self.result[1], self.Event_type) self.assertEqual(self.result[2], self.flow_target_packed_object) def test_packed_object_hasattr_False(self): self.newtarget = Event(self.Publisher_id, self.Event_type, self.flow_target_packed_object) self.result = self.newtarget.packed_object() self.assertEqual(self.result[0], self.Publisher_id) self.assertEqual(self.result[1], self.Event_type) self.assertEqual(self.result[2], self.flow_target_packed_object)