def acquire_source(self, msg, connector_type, direction, push_hops=True): """Determine the `str(go_connector)` value that a msg came in on by looking at the connector_type and fetching the appropriate values from the `msg` helper_metadata. Raises `UnroutableMessageError` if the connector_type has a value not appropriate for the direction. Note: `str(go_connector)` is what is stored in Go routing tables. """ msg_mdh = self.get_metadata_helper(msg) if direction == self.INBOUND: allowed_types = (self.TRANSPORT_TAG, self.ROUTER, self.BILLING) else: allowed_types = (self.CONVERSATION, self.ROUTER, self.OPT_OUT, self.BILLING) if connector_type not in allowed_types: raise UnroutableMessageError( "Source connector of invalid type: %s" % connector_type, msg) if connector_type == self.CONVERSATION: conv_info = msg_mdh.get_conversation_info() src_conn = str(GoConnector.for_conversation( conv_info['conversation_type'], conv_info['conversation_key'])) elif connector_type == self.ROUTER: router_info = msg_mdh.get_router_info() src_conn = str(GoConnector.for_router( router_info['router_type'], router_info['router_key'], self.router_direction(direction))) elif connector_type == self.TRANSPORT_TAG: src_conn = str(GoConnector.for_transport_tag(*msg_mdh.tag)) elif connector_type == self.OPT_OUT: src_conn = str(GoConnector.for_opt_out()) elif connector_type == self.BILLING: # when the source is a billing router, outbound messages # are always received from the inbound billing connector # and inbound messages are always received from the outbound # billing connector. src_conn = str( GoConnector.for_billing(self.router_direction(direction))) else: raise UnroutableMessageError( "Serious error. Reached apparently unreachable state" " in which source connector type is both valid" " but unknown. Bad connector type is: %s" % connector_type, msg) src_conn_str = str(src_conn) if push_hops: rmeta = RoutingMetadata(msg) rmeta.push_source(src_conn_str, msg.get_routing_endpoint()) return src_conn_str
def test_connector_direction(self): def assert_inbound(conn): self.assertEqual(GoConnector.INBOUND, conn.direction) def assert_outbound(conn): self.assertEqual(GoConnector.OUTBOUND, conn.direction) assert_inbound(GoConnector.for_opt_out()) assert_inbound(GoConnector.for_conversation("conv_type_1", "12345")) assert_outbound(GoConnector.for_transport_tag("tagpool_1", "tag_1")) assert_inbound( GoConnector.for_router("rb_type_1", "12345", GoConnector.INBOUND)) assert_outbound( GoConnector.for_router("rb_type_1", "12345", GoConnector.OUTBOUND))
def test_connector_direction(self): def assert_inbound(conn): self.assertEqual(GoConnector.INBOUND, conn.direction) def assert_outbound(conn): self.assertEqual(GoConnector.OUTBOUND, conn.direction) assert_inbound(GoConnector.for_opt_out()) assert_inbound(GoConnector.for_conversation("conv_type_1", "12345")) assert_outbound(GoConnector.for_transport_tag("tagpool_1", "tag_1")) assert_inbound( GoConnector.for_router("rb_type_1", "12345", GoConnector.INBOUND)) assert_outbound( GoConnector.for_router("rb_type_1", "12345", GoConnector.OUTBOUND))
def test_create_opt_out_connector(self): c = GoConnector.for_opt_out() self.assertEqual(c.ctype, GoConnector.OPT_OUT) self.assertEqual(str(c), "OPT_OUT")
def publish_inbound_optout(self, config, msg): """Publish an inbound opt-out request to the opt-out worker.""" target = [str(GoConnector.for_opt_out()), 'default'] dst_connector_name, dst_endpoint = yield self.set_destination( msg, target, self.INBOUND) yield self.publish_inbound(msg, dst_connector_name, dst_endpoint)
def test_create_opt_out_connector(self): c = GoConnector.for_opt_out() self.assertEqual(c.ctype, GoConnector.OPT_OUT) self.assertEqual(str(c), "OPT_OUT")
def publish_inbound_optout(self, config, msg): """Publish an inbound opt-out request to the opt-out worker.""" target = [str(GoConnector.for_opt_out()), 'default'] dst_connector_name, dst_endpoint = yield self.set_destination( msg, target, self.INBOUND) yield self.publish_inbound(msg, dst_connector_name, dst_endpoint)
def acquire_source(self, msg, connector_type, direction, push_hops=True): """Determine the `str(go_connector)` value that a msg came in on by looking at the connector_type and fetching the appropriate values from the `msg` helper_metadata. Raises `UnroutableMessageError` if the connector_type has a value not appropriate for the direction. Note: `str(go_connector)` is what is stored in Go routing tables. """ msg_mdh = self.get_metadata_helper(msg) if direction == self.INBOUND: allowed_types = (self.TRANSPORT_TAG, self.ROUTER, self.BILLING) else: allowed_types = (self.CONVERSATION, self.ROUTER, self.OPT_OUT, self.BILLING) if connector_type not in allowed_types: raise UnroutableMessageError( "Source connector of invalid type: %s" % connector_type, msg) if connector_type == self.CONVERSATION: conv_info = msg_mdh.get_conversation_info() src_conn = str( GoConnector.for_conversation(conv_info['conversation_type'], conv_info['conversation_key'])) elif connector_type == self.ROUTER: router_info = msg_mdh.get_router_info() src_conn = str( GoConnector.for_router(router_info['router_type'], router_info['router_key'], self.router_direction(direction))) elif connector_type == self.TRANSPORT_TAG: src_conn = str(GoConnector.for_transport_tag(*msg_mdh.tag)) elif connector_type == self.OPT_OUT: src_conn = str(GoConnector.for_opt_out()) elif connector_type == self.BILLING: # when the source is a billing router, outbound messages # are always received from the inbound billing connector # and inbound messages are always received from the outbound # billing connector. src_conn = str( GoConnector.for_billing(self.router_direction(direction))) else: raise UnroutableMessageError( "Serious error. Reached apparently unreachable state" " in which source connector type is both valid" " but unknown. Bad connector type is: %s" % connector_type, msg) src_conn_str = str(src_conn) if push_hops: rmeta = RoutingMetadata(msg) rmeta.push_source(src_conn_str, msg.get_routing_endpoint()) return src_conn_str