def test_with_loss(self):
generator = NackGenerator()
# receive packets: 0, <1 missing>, 2
packets = create_rtp_packets(3, 0)
missing = packets.pop(1)
for packet in packets:
missed = generator.add(packet)
self.assertEqual(missed, packet.sequence_number == 2)
self.assertEqual(generator.missing, set([1]))
# late arrival
missed = generator.add(missing)
self.assertEqual(missed, False)
self.assertEqual(generator.missing, set())
def test_no_loss(self):
generator = NackGenerator()
for packet in create_rtp_packets(20, 0):
missed = generator.add(packet)
self.assertEqual(missed, False)
self.assertEqual(generator.missing, set())
def test_with_loss_truncate(self):
generator = NackGenerator()
packets = create_rtp_packets(259, 0)
generator.add(packets[0])
generator.add(packets[129])
self.assertEqual(generator.missing, set(range(1, 129)))
generator.add(packets[258])
self.assertEqual(generator.missing, set(range(130, 258)))