Exemple #1
0
class TestSendMessage(TestCase):

    def setUp(self):
        self._server = socket.socket()
        self._server.bind(('localhost', 0))
        self._server.listen(1)

        self._channel = Channel(address=self._server.getsockname())
        self._server_channel = Channel(socket=self._server.accept()[0])

    def tearDown(self):
        self._server_channel.close()
        self._server.close()
        self._channel.close()

    def test_send_message(self):
        for case in encoded_messages:
            self._channel.send(parse_message(MultiplexerMessage,
                case['encoded']))

            assert dict_message(self._server_channel.receive()) == \
                    case['pythonized']