def test_connect(self):
gatherer_1 = RTCIceGatherer()
transport_1 = RTCIceTransport(gatherer_1)
gatherer_2 = RTCIceGatherer()
transport_2 = RTCIceTransport(gatherer_2)
# gather candidates
run(asyncio.gather(gatherer_1.gather(), gatherer_2.gather()))
for candidate in gatherer_2.getLocalCandidates():
run(transport_1.addRemoteCandidate(candidate))
for candidate in gatherer_1.getLocalCandidates():
run(transport_2.addRemoteCandidate(candidate))
self.assertEqual(transport_1.state, "new")
self.assertEqual(transport_2.state, "new")
# connect
run(
asyncio.gather(
transport_1.start(gatherer_2.getLocalParameters()),
transport_2.start(gatherer_1.getLocalParameters()),
))
self.assertEqual(transport_1.state, "completed")
self.assertEqual(transport_2.state, "completed")
# cleanup
run(asyncio.gather(transport_1.stop(), transport_2.stop()))
self.assertEqual(transport_1.state, "closed")
self.assertEqual(transport_2.state, "closed")
def test_connect_fail(self):
gatherer_1 = RTCIceGatherer()
transport_1 = RTCIceTransport(gatherer_1)
gatherer_2 = RTCIceGatherer()
transport_2 = RTCIceTransport(gatherer_2)
# gather candidates
run(asyncio.gather(gatherer_1.gather(), gatherer_2.gather()))
for candidate in gatherer_2.getLocalCandidates():
transport_1.addRemoteCandidate(candidate)
for candidate in gatherer_1.getLocalCandidates():
transport_2.addRemoteCandidate(candidate)
self.assertEqual(transport_1.state, 'new')
self.assertEqual(transport_2.state, 'new')
# connect
run(transport_2.stop())
run(transport_1.start(gatherer_2.getLocalParameters()))
self.assertEqual(transport_1.state, 'failed')
self.assertEqual(transport_2.state, 'closed')
# cleanup
run(asyncio.gather(transport_1.stop(), transport_2.stop()))
self.assertEqual(transport_1.state, 'closed')
self.assertEqual(transport_2.state, 'closed')
def test_connect_when_closed(self):
gatherer = RTCIceGatherer()
transport = RTCIceTransport(gatherer)
# stop transport
run(transport.stop())
self.assertEqual(transport.state, "closed")
# try to start it
with self.assertRaises(InvalidStateError) as cm:
run(
transport.start(
RTCIceParameters(usernameFragment="foo", password="******")))
self.assertEqual(str(cm.exception), "RTCIceTransport is closed")
def test_send_connection_error(self):
gatherer = RTCIceGatherer()
transport = RTCIceTransport(gatherer)
self.assertEqual(transport.state, "new")
# fake connection
gatherer._connection.connect = noop
run(
transport.start(
RTCIceParameters(usernameFragment="foo", password="******")))
self.assertEqual(transport.state, "completed")
with self.assertRaises(ConnectionError):
run(transport._send(b"foo"))
self.assertEqual(transport.state, "failed")
def test_send_connection_error(self):
gatherer = RTCIceGatherer()
transport = RTCIceTransport(gatherer)
self.assertEqual(transport.state, 'new')
# fake connection
gatherer._connection.connect = noop
run(
transport.start(
RTCIceParameters(usernameFragment='foo', password='******')))
self.assertEqual(transport.state, 'completed')
with self.assertRaises(ConnectionError):
run(transport._send(b'foo'))
self.assertEqual(transport.state, 'failed')
def test_connection_closed(self):
gatherer = RTCIceGatherer()
# mock out methods
gatherer._connection.connect = mock_connect
gatherer._connection.get_event = mock_get_event
transport = RTCIceTransport(gatherer)
self.assertEqual(transport.state, "new")
run(
transport.start(
RTCIceParameters(usernameFragment="foo", password="******")))
self.assertEqual(transport.state, "completed")
run(asyncio.sleep(1))
self.assertEqual(transport.state, "failed")
run(transport.stop())
self.assertEqual(transport.state, "closed")
