def test_route_rtp(self):
receiver1 = object()
receiver2 = object()
router = RtpRouter()
router.register_receiver(receiver1, ssrcs=[1234, 2345], payload_types=[96, 97])
router.register_receiver(receiver2, ssrcs=[3456, 4567], payload_types=[98, 99])
# known SSRC and payload type
self.assertEqual(
router.route_rtp(RtpPacket(ssrc=1234, payload_type=96)), receiver1
)
self.assertEqual(
router.route_rtp(RtpPacket(ssrc=2345, payload_type=97)), receiver1
)
self.assertEqual(
router.route_rtp(RtpPacket(ssrc=3456, payload_type=98)), receiver2
)
self.assertEqual(
router.route_rtp(RtpPacket(ssrc=4567, payload_type=99)), receiver2
)
# unknown SSRC, known payload type
self.assertEqual(
router.route_rtp(RtpPacket(ssrc=5678, payload_type=96)), receiver1
)
self.assertEqual(router.ssrc_table[5678], receiver1)
# unknown SSRC and payload type
self.assertEqual(router.route_rtp(RtpPacket(ssrc=6789, payload_type=100)), None)
def test_route_rtp_ambiguous_payload_type(self):
receiver1 = object()
receiver2 = object()
router = RtpRouter()
router.register_receiver(receiver1, ssrcs=[1234, 2345], payload_types=[96, 97])
router.register_receiver(receiver2, ssrcs=[3456, 4567], payload_types=[96, 97])
# known SSRC and payload type
self.assertEqual(
router.route_rtp(RtpPacket(ssrc=1234, payload_type=96)), receiver1
)
self.assertEqual(
router.route_rtp(RtpPacket(ssrc=2345, payload_type=97)), receiver1
)
self.assertEqual(
router.route_rtp(RtpPacket(ssrc=3456, payload_type=96)), receiver2
)
self.assertEqual(
router.route_rtp(RtpPacket(ssrc=4567, payload_type=97)), receiver2
)
# unknown SSRC, ambiguous payload type
self.assertEqual(router.route_rtp(RtpPacket(ssrc=5678, payload_type=96)), None)
self.assertEqual(router.route_rtp(RtpPacket(ssrc=5678, payload_type=97)), None)
def test_route_rtcp(self):
receiver = object()
sender = object()
router = RtpRouter()
router.register_receiver(receiver,
ssrcs=[1234, 2345],
payload_types=[96, 97])
router.register_sender(sender, ssrc=3456)
# BYE
packet = RtcpByePacket(sources=[1234, 2345])
self.assertEqual(router.route_rtcp(packet), set([receiver]))
# RR
packet = RtcpRrPacket(ssrc=1234,
reports=[
RtcpReceiverInfo(ssrc=3456,
fraction_lost=0,
packets_lost=0,
highest_sequence=630,
jitter=1906,
lsr=0,
dlsr=0)
])
self.assertEqual(router.route_rtcp(packet), set([sender]))
# SR
packet = RtcpSrPacket(ssrc=1234,
sender_info=RtcpSenderInfo(ntp_timestamp=0,
rtp_timestamp=0,
packet_count=0,
octet_count=0),
reports=[
RtcpReceiverInfo(ssrc=3456,
fraction_lost=0,
packets_lost=0,
highest_sequence=630,
jitter=1906,
lsr=0,
dlsr=0)
])
self.assertEqual(router.route_rtcp(packet), set([receiver, sender]))
# PSFB
packet = RtcpPsfbPacket(fmt=RTCP_PSFB_PLI, ssrc=1234, media_ssrc=3456)
self.assertEqual(router.route_rtcp(packet), set([sender]))
# RTPFB
packet = RtcpRtpfbPacket(fmt=RTCP_RTPFB_NACK,
ssrc=1234,
media_ssrc=3456)
self.assertEqual(router.route_rtcp(packet), set([sender]))