def test_addTrack_bogus(self):
pc = RTCPeerConnection()
# try adding a bogus track
with self.assertRaises(InternalError) as cm:
pc.addTrack(BogusStreamTrack)
self.assertEqual(str(cm.exception), 'Invalid track kind "bogus"')
def test_close(self):
pc = RTCPeerConnection()
pc_states = track_states(pc)
# close once
run(pc.close())
# close twice
run(pc.close())
self.assertEqual(pc_states['signalingState'], ['stable', 'closed'])
def test_setRemoteDescription_unexpected_offer(self):
pc = RTCPeerConnection()
pc.addTrack(AudioStreamTrack())
offer = run(pc.createOffer())
run(pc.setLocalDescription(offer))
with self.assertRaises(InvalidStateError) as cm:
run(pc.setRemoteDescription(RTCSessionDescription(sdp='', type='offer')))
self.assertEqual(str(cm.exception),
'Cannot handle offer in signaling state "have-local-offer"')
async def offer(request):
offer = await request.json()
offer = RTCSessionDescription(sdp=offer['sdp'], type=offer['type'])
pc = RTCPeerConnection()
await pc.setRemoteDescription(offer)
answer = await pc.createAnswer()
await pc.setLocalDescription(answer)
return web.Response(content_type='application/json',
text=json.dumps({
'sdp': pc.localDescription.sdp,
'type': pc.localDescription.type
}))
def test_addTrack_video(self):
pc = RTCPeerConnection()
# add video track
track = VideoStreamTrack()
sender = pc.addTrack(track)
self.assertIsNotNone(sender)
self.assertEqual(sender.track, track)
self.assertEqual(pc.getSenders(), [sender])
# try to add same track again
with self.assertRaises(InvalidAccessError) as cm:
pc.addTrack(track)
self.assertEqual(str(cm.exception), 'Track already has a sender')
# try adding an audio track
with self.assertRaises(InternalError) as cm:
pc.addTrack(AudioStreamTrack())
self.assertEqual(str(cm.exception), 'Only a single media track is supported for now')
def test_addTrack_closed(self):
pc = RTCPeerConnection()
run(pc.close())
with self.assertRaises(InvalidStateError) as cm:
pc.addTrack(AudioStreamTrack())
self.assertEqual(str(cm.exception), 'RTCPeerConnection is closed')
def test_setRemoteDescription_unexpected_answer(self):
pc = RTCPeerConnection()
with self.assertRaises(InvalidStateError) as cm:
run(pc.setRemoteDescription(RTCSessionDescription(sdp='', type='answer')))
self.assertEqual(str(cm.exception), 'Cannot handle answer in signaling state "stable"')
def test_createOffer_closed(self):
pc = RTCPeerConnection()
run(pc.close())
with self.assertRaises(InvalidStateError) as cm:
run(pc.createOffer())
self.assertEqual(str(cm.exception), 'RTCPeerConnection is closed')
def test_createAnswer_without_offer(self):
pc = RTCPeerConnection()
with self.assertRaises(InvalidStateError) as cm:
run(pc.createAnswer())
self.assertEqual(str(cm.exception), 'Cannot create answer in signaling state "stable"')
def test_connect_datachannel(self):
pc1 = RTCPeerConnection()
pc1_states = track_states(pc1)
pc2 = RTCPeerConnection()
pc2_states = track_states(pc2)
# create offer
dc = pc1.createDataChannel('chat')
dc.send('hello')
offer = run(pc1.createOffer())
self.assertEqual(offer.type, 'offer')
self.assertTrue('m=application ' in offer.sdp)
self.assertFalse('a=candidate:' in offer.sdp)
run(pc1.setLocalDescription(offer))
self.assertEqual(pc1.iceConnectionState, 'new')
self.assertEqual(pc1.iceGatheringState, 'complete')
self.assertTrue('m=application ' in pc1.localDescription.sdp)
self.assertTrue('a=candidate:' in pc1.localDescription.sdp)
self.assertTrue('a=sctpmap:5000 webrtc-datachannel 256' in pc1.localDescription.sdp)
# handle offer
run(pc2.setRemoteDescription(pc1.localDescription))
self.assertEqual(pc2.remoteDescription, pc1.localDescription)
self.assertEqual(len(pc2.getReceivers()), 0)
self.assertEqual(len(pc2.getSenders()), 0)
self.assertEqual(len(pc2.getSenders()), 0)
# create answer
answer = run(pc2.createAnswer())
self.assertEqual(answer.type, 'answer')
self.assertTrue('m=application ' in answer.sdp)
self.assertFalse('a=candidate:' in answer.sdp)
run(pc2.setLocalDescription(answer))
self.assertEqual(pc2.iceConnectionState, 'checking')
self.assertEqual(pc2.iceGatheringState, 'complete')
self.assertTrue('m=application ' in pc2.localDescription.sdp)
self.assertTrue('a=candidate:' in pc2.localDescription.sdp)
self.assertTrue('a=sctpmap:5000 webrtc-datachannel 256' in pc2.localDescription.sdp)
# handle answer
run(pc1.setRemoteDescription(pc2.localDescription))
self.assertEqual(pc1.remoteDescription, pc2.localDescription)
self.assertEqual(pc1.iceConnectionState, 'checking')
# check outcome
run(asyncio.sleep(1))
self.assertEqual(pc1.iceConnectionState, 'completed')
self.assertEqual(pc2.iceConnectionState, 'completed')
# close
run(pc1.close())
run(pc2.close())
self.assertEqual(pc1.iceConnectionState, 'closed')
self.assertEqual(pc2.iceConnectionState, 'closed')
# check state changes
self.assertEqual(pc1_states['iceConnectionState'], [
'new', 'checking', 'completed', 'closed'])
self.assertEqual(pc1_states['iceGatheringState'], [
'new', 'gathering', 'complete'])
self.assertEqual(pc1_states['signalingState'], [
'stable', 'have-local-offer', 'stable', 'closed'])
self.assertEqual(pc2_states['iceConnectionState'], [
'new', 'checking', 'completed', 'closed'])
self.assertEqual(pc2_states['iceGatheringState'], [
'new', 'gathering', 'complete'])
self.assertEqual(pc2_states['signalingState'], [
'stable', 'have-remote-offer', 'stable', 'closed'])
def test_connect_audio_bidirectional(self):
pc1 = RTCPeerConnection()
pc1_states = track_states(pc1)
pc2 = RTCPeerConnection()
pc2_states = track_states(pc2)
self.assertEqual(pc1.iceConnectionState, 'new')
self.assertEqual(pc1.iceGatheringState, 'new')
self.assertIsNone(pc1.localDescription)
self.assertIsNone(pc1.remoteDescription)
self.assertEqual(pc2.iceConnectionState, 'new')
self.assertEqual(pc2.iceGatheringState, 'new')
self.assertIsNone(pc2.localDescription)
self.assertIsNone(pc2.remoteDescription)
# create offer
pc1.addTrack(AudioStreamTrack())
offer = run(pc1.createOffer())
self.assertEqual(offer.type, 'offer')
self.assertTrue('m=audio ' in offer.sdp)
self.assertFalse('a=candidate:' in offer.sdp)
run(pc1.setLocalDescription(offer))
self.assertEqual(pc1.iceConnectionState, 'new')
self.assertEqual(pc1.iceGatheringState, 'complete')
self.assertTrue('m=audio ' in pc1.localDescription.sdp)
self.assertTrue('a=candidate:' in pc1.localDescription.sdp)
self.assertTrue('a=sendrecv' in pc1.localDescription.sdp)
# handle offer
run(pc2.setRemoteDescription(pc1.localDescription))
self.assertEqual(pc2.remoteDescription, pc1.localDescription)
self.assertEqual(len(pc2.getReceivers()), 1)
# create answer
pc2.addTrack(AudioStreamTrack())
answer = run(pc2.createAnswer())
self.assertEqual(answer.type, 'answer')
self.assertTrue('m=audio ' in answer.sdp)
self.assertFalse('a=candidate:' in answer.sdp)
run(pc2.setLocalDescription(answer))
self.assertEqual(pc2.iceConnectionState, 'checking')
self.assertEqual(pc2.iceGatheringState, 'complete')
self.assertTrue('m=audio ' in pc2.localDescription.sdp)
self.assertTrue('a=candidate:' in pc2.localDescription.sdp)
self.assertTrue('a=sendrecv' in pc1.localDescription.sdp)
# handle answer
run(pc1.setRemoteDescription(pc2.localDescription))
self.assertEqual(pc1.remoteDescription, pc2.localDescription)
self.assertEqual(pc1.iceConnectionState, 'checking')
# check outcome
run(asyncio.sleep(1))
self.assertEqual(pc1.iceConnectionState, 'completed')
self.assertEqual(pc2.iceConnectionState, 'completed')
# close
run(pc1.close())
run(pc2.close())
self.assertEqual(pc1.iceConnectionState, 'closed')
self.assertEqual(pc2.iceConnectionState, 'closed')
# check state changes
self.assertEqual(pc1_states['iceConnectionState'], [
'new', 'checking', 'completed', 'closed'])
self.assertEqual(pc1_states['iceGatheringState'], [
'new', 'gathering', 'complete'])
self.assertEqual(pc1_states['signalingState'], [
'stable', 'have-local-offer', 'stable', 'closed'])
self.assertEqual(pc2_states['iceConnectionState'], [
'new', 'checking', 'completed', 'closed'])
self.assertEqual(pc2_states['iceGatheringState'], [
'new', 'gathering', 'complete'])
self.assertEqual(pc2_states['signalingState'], [
'stable', 'have-remote-offer', 'stable', 'closed'])