Beta Shell v2.0

File Manager

~ /usr/local/python-3.12/lib/python3.12/test/test_asyncio

Name	Size	Perms	Modified	Actions
[ .. / .. ]
__init__.py	353 B	-rw-r--r--	2024-04-12 10:29:39	Edit Del
__main__.py	58 B	-rw-r--r--	2024-04-12 10:29:39	Edit Del
__pycache__	-	drwxr-xr-x	2024-04-12 10:29:55	Del
echo.py	148 B	-rw-r--r--	2024-04-12 10:29:39	Edit Del
echo2.py	123 B	-rw-r--r--	2024-04-12 10:29:39	Edit Del
echo3.py	276 B	-rw-r--r--	2024-04-12 10:29:39	Edit Del
functional.py	7.3 KB	-rw-r--r--	2024-04-12 10:29:39	Edit Del
test_base_events.py	79.58 KB	-rw-r--r--	2024-04-12 10:29:39	Edit Del
test_buffered_proto.py	2.28 KB	-rw-r--r--	2024-04-12 10:29:39	Edit Del
test_context.py	1.04 KB	-rw-r--r--	2024-04-12 10:29:39	Edit Del
test_eager_task_factory.py	10.35 KB	-rw-r--r--	2024-04-12 10:29:39	Edit Del
test_events.py	108.35 KB	-rw-r--r--	2024-04-12 10:29:39	Edit Del
test_futures.py	29.44 KB	-rw-r--r--	2024-04-12 10:29:39	Edit Del
test_futures2.py	2.77 KB	-rw-r--r--	2024-04-12 10:29:39	Edit Del
test_locks.py	46.64 KB	-rw-r--r--	2024-04-12 10:29:39	Edit Del
test_pep492.py	5.4 KB	-rw-r--r--	2024-04-12 10:29:39	Edit Del
test_proactor_events.py	37.84 KB	-rw-r--r--	2024-04-12 10:29:39	Edit Del
test_protocols.py	2.24 KB	-rw-r--r--	2024-04-12 10:29:39	Edit Del
test_queues.py	14.74 KB	-rw-r--r--	2024-04-12 10:29:39	Edit Del
test_runners.py	14.62 KB	-rw-r--r--	2024-04-12 10:29:39	Edit Del
test_selector_events.py	52.19 KB	-rw-r--r--	2024-04-12 10:29:39	Edit Del
test_sendfile.py	21.15 KB	-rw-r--r--	2024-04-12 10:29:39	Edit Del
test_server.py	5.49 KB	-rw-r--r--	2024-04-12 10:29:39	Edit Del
test_sock_lowlevel.py	24.65 KB	-rw-r--r--	2024-04-12 10:29:39	Edit Del
test_ssl.py	53.41 KB	-rw-r--r--	2024-04-12 10:29:39	Edit Del
test_sslproto.py	26.65 KB	-rw-r--r--	2024-04-12 10:29:39	Edit Del
test_streams.py	42.74 KB	-rw-r--r--	2024-04-12 10:29:39	Edit Del
test_subprocess.py	36.56 KB	-rw-r--r--	2024-04-12 10:29:39	Edit Del
test_taskgroups.py	23.04 KB	-rw-r--r--	2024-04-12 10:29:39	Edit Del
test_tasks.py	107.78 KB	-rw-r--r--	2024-04-12 10:29:39	Edit Del
test_threads.py	1.59 KB	-rw-r--r--	2024-04-12 10:29:39	Edit Del
test_timeouts.py	10.79 KB	-rw-r--r--	2024-04-12 10:29:39	Edit Del
test_transports.py	3.73 KB	-rw-r--r--	2024-04-12 10:29:39	Edit Del
test_unix_events.py	66.66 KB	-rw-r--r--	2024-04-12 10:29:39	Edit Del
test_waitfor.py	10.57 KB	-rw-r--r--	2024-04-12 10:29:39	Edit Del
test_windows_events.py	11.94 KB	-rw-r--r--	2024-04-12 10:29:39	Edit Del
test_windows_utils.py	4.14 KB	-rw-r--r--	2024-04-12 10:29:39	Edit Del
utils.py	18.53 KB	-rw-r--r--	2024-04-12 10:29:39	Edit Del

Editing: test_buffered_proto.py (2.28 KB)

import asyncio
import unittest

from test.test_asyncio import functional as func_tests


def tearDownModule():
    asyncio.set_event_loop_policy(None)


class ReceiveStuffProto(asyncio.BufferedProtocol):
    def __init__(self, cb, con_lost_fut):
        self.cb = cb
        self.con_lost_fut = con_lost_fut

    def get_buffer(self, sizehint):
        self.buffer = bytearray(100)
        return self.buffer

    def buffer_updated(self, nbytes):
        self.cb(self.buffer[:nbytes])

    def connection_lost(self, exc):
        if exc is None:
            self.con_lost_fut.set_result(None)
        else:
            self.con_lost_fut.set_exception(exc)


class BaseTestBufferedProtocol(func_tests.FunctionalTestCaseMixin):

    def new_loop(self):
        raise NotImplementedError

    def test_buffered_proto_create_connection(self):

        NOISE = b'12345678+' * 1024

        async def client(addr):
            data = b''

            def on_buf(buf):
                nonlocal data
                data += buf
                if data == NOISE:
                    tr.write(b'1')

            conn_lost_fut = self.loop.create_future()

            tr, pr = await self.loop.create_connection(
                lambda: ReceiveStuffProto(on_buf, conn_lost_fut), *addr)

            await conn_lost_fut

        async def on_server_client(reader, writer):
            writer.write(NOISE)
            await reader.readexactly(1)
            writer.close()
            await writer.wait_closed()

        srv = self.loop.run_until_complete(
            asyncio.start_server(
                on_server_client, '127.0.0.1', 0))

        addr = srv.sockets[0].getsockname()
        self.loop.run_until_complete(
            asyncio.wait_for(client(addr), 5))

        srv.close()
        self.loop.run_until_complete(srv.wait_closed())


class BufferedProtocolSelectorTests(BaseTestBufferedProtocol,
                                    unittest.TestCase):

    def new_loop(self):
        return asyncio.SelectorEventLoop()


@unittest.skipUnless(hasattr(asyncio, 'ProactorEventLoop'), 'Windows only')
class BufferedProtocolProactorTests(BaseTestBufferedProtocol,
                                    unittest.TestCase):

    def new_loop(self):
        return asyncio.ProactorEventLoop()


if __name__ == '__main__':
    unittest.main()