def test_payload_build(self): type = PayloadType.UNK event = PayloadEvent.UNK data = "test_data" role = "test_role" p = Payload(data, event, type, role) x = build_payload(p.content()) y = self.param_association(x, data, type, event, role) self.assertEqual(x.content(), y.content()) print("Build: DICT :Passed") y = build_payload(json.dumps(x.content())) self.assertEqual(x.content(), y.content()) print("Build: JSON :Passed") p = Payload(data, event, type) self.assertEqual(BLANK_FIELD, p.role) print("Build: Default Role :Passed")
def data_handler(self, payload: Payload, delegation_event_whitelist=None): if delegation_event_whitelist is None: delegation_event_whitelist = [] notify_all = payload.role == WILDCARD correct_payload = payload.role == ROLE is_request = payload.type in REQUEST_TYPES white_listed = payload.event in delegation_event_whitelist # 1. Determines if the payload is intended for the current unit if not correct_payload and is_request and not white_listed: # 2 If the current payload is NOT intended for the current unit # it will be delegated, and hopefully send it to the appropriate # unit. if IS_DELEGATOR and payload.role is not BLANK_FIELD: # 2.1 If delegation is enabled, it then gets the ip address of the intended # unit and sends the payload, if the wildcard has been specified then all # the units will be notified addresses = PiDiscovery.get_ip_addresses(payload.role) transactions = [] res_payload = PayloadEventMessages.ADDRESS_NOT_FOUND.value for address in addresses: client = LANClient(address, self.port, timeout=5, ignore_errors=True) print("\t%s\n\t|\t\tRelaying -> %s:%s" % (print_payload(payload), address, self.port)) res_payload = client.send(payload) transactions.append((res_payload.data, res_payload.event.name)) if len(transactions) > 1: payload.data = transactions payload.type = PayloadType.RSP return payload return res_payload else: # 2.2 If delegation is disabled, and the role can't be handled, then # a error response will be sent telling the client 'Wrong Node'. if notify_all: return self.process_payload(payload) return PayloadEventMessages.WRONG_NODE return self.process_payload(payload)