async def _acquire(self) -> NoReturn:
await self.connection.acquire()
register(self.connection, self)
self._multiplexer = Multiplexer(self.connection)
await self._multiplexer.acquire()
register(self, self._multiplexer)
self._channel = self._multiplexer.connect("control")
await self._channel.acquire()
self._control = self.create_endpoint(self._channel)
register(self, self._control)
await self._control.acquire()
async def test_multiplexer_illegal(self):
c_faked_networking, c_multiplexed = pipe_bidi()
muliplexer = Multiplexer(c_multiplexed)
async with c_faked_networking, muliplexer:
async with muliplexer.connect('existing') as f:
await c_faked_networking.write(
Message({
'target': 'existing',
'type': 'message'
}))
msg: Message = await wait_for(c_faked_networking.read(), 5)
self.assertIsInstance(msg.values, dict)
self.assertIn('type', msg.values)
self.assertEqual('illegal', msg.values['type'])
class Multiplexed(Resource, ABC):
def __init__(self, connection: Connection):
super().__init__()
self.connection = connection
self._multiplexer = None
self._channel = None
self._control = None
@abstractmethod
def create_endpoint(self, connection: Connection) -> Resource:
pass
async def _acquire(self) -> NoReturn:
await self.connection.acquire()
register(self.connection, self)
self._multiplexer = Multiplexer(self.connection)
await self._multiplexer.acquire()
register(self, self._multiplexer)
self._channel = self._multiplexer.connect("control")
await self._channel.acquire()
self._control = self.create_endpoint(self._channel)
register(self, self._control)
await self._control.acquire()
async def _release(self) -> NoReturn:
await self.connection.release(force=False)
async def test_multiplexer_delivered(self):
value = {'v': 'test'}
c_faked_networking, c_multiplexed = pipe_bidi()
muliplexer = Multiplexer(c_multiplexed)
async with c_faked_networking, muliplexer:
async with muliplexer.connect('existing') as f:
await c_faked_networking.write(
Message({
'target': 'existing',
'type': 'message',
'payload': value
}))
msg: Message = await wait_for(f.read(), 5)
self.assertIs(value, msg.values)
async def test_multiplexer_send(self):
values = {}
c_faked_networking, c_multiplexed = pipe_bidi()
muliplexer = Multiplexer(c_multiplexed)
async with c_faked_networking, muliplexer:
async with muliplexer.connect('existing') as f:
await f.write(Message(values, []))
msg: Message = await wait_for(c_faked_networking.read(), 5)
self.assertIsInstance(msg.values, dict)
self.assertIn('target', msg.values)
self.assertEqual('existing', msg.values['target'])
self.assertIn('type', msg.values)
self.assertEqual('message', msg.values['type'])
self.assertIn('payload', msg.values)
self.assertIs(values, msg.values['payload'])
async def test_multiplexer_channel_death(self):
c_faked_networking, c_multiplexed = pipe_bidi()
muliplexer = Multiplexer(c_multiplexed)
async with c_faked_networking, muliplexer:
async with muliplexer.connect('temporary'):
pass
msg: Message = await wait_for(c_faked_networking.read(), 5)
self.assertIsInstance(msg.values, dict)
self.assertIn('type', msg.values)
self.assertEqual('close', msg.values['type'])
await c_faked_networking.write(
Message({
'target': 'temporary',
'payload': {}
}))
msg: Message = await wait_for(c_faked_networking.read(), 5)
self.assertIsInstance(msg.values, dict)
self.assertIn('type', msg.values)
self.assertEqual('close', msg.values['type'])
async def test_multiplexer_reject(self):
c_faked_networking, c_multiplexed = pipe_bidi()
muliplexer = Multiplexer(c_multiplexed)
async with c_faked_networking, muliplexer:
await c_faked_networking.write(
Message({
'target': 'non-existent',
'payload': {}
}))
msg: Message = await wait_for(c_faked_networking.read(), 5)
self.assertIsInstance(msg.values, dict)
self.assertIn('type', msg.values)
self.assertEqual('close', msg.values['type'])
async def test_multiplexer_forward_close(self):
c_m1, c_m2 = pipe_bidi()
m1 = Multiplexer(c_m1)
m2 = Multiplexer(c_m2)
async with m1, m2:
m1_ch = m1.connect('channel')
m2_ch = m2.connect('channel')
async with m1_ch, m2_ch:
await m1_ch.close()
self.assertTrue(m1_ch.closed)
await wait_for(m2_ch.read(), 5)
self.assertTrue(m2_ch.closed)
async def test_bidi_multiplexer_comm(self):
c_m1, c_m2 = pipe_bidi()
m1 = Multiplexer(c_m1)
m2 = Multiplexer(c_m2)
async with m1, m2:
m1_ch = m1.connect('channel')
m2_ch = m2.connect('channel')
async with m1_ch, m2_ch:
await m1_ch.write(Message({'someval': 123456}))
msg: Message = await m2_ch.read()
self.assertIsInstance(msg.values, dict)
self.assertIn('someval', msg.values)
self.assertEqual(123456, msg.values['someval'])
self.assertTrue(m1_ch.closed)
self.assertTrue(m2_ch.closed)