Python BaseSocksProtocol Examples

Example #1

File: test_protocols.py Project: bitaps-com/aiosocks

async def test_base_reader_limit(loop):
    proto = BaseSocksProtocol(None, None, ('python.org', 80),
                              None, None, reader_limit=10, loop=loop)
    assert proto.reader._limit == 10

    proto = BaseSocksProtocol(None, None, ('python.org', 80),
                              None, None, reader_limit=15, loop=loop)
    assert proto.reader._limit == 15

Example #2

File: test_protocols.py Project: A888R/aiosocks

 def test_incomplete_error(self):
     proto = BaseSocksProtocol(None, None, ('python.org', 80),
                               None, None, reader_limit=10,
                               loop=self.loop)
     proto._stream_reader.readexactly = fake_coroutine(
         asyncio.IncompleteReadError(b'part', 5))
     with self.assertRaises(aiosocks.InvalidServerReply):
         self.loop.run_until_complete(proto.read_response(4))

Example #3

 def test_incomplete_error(self):
     proto = BaseSocksProtocol(None,
                               None, ('python.org', 80),
                               None,
                               None,
                               reader_limit=10,
                               loop=self.loop)
     proto._stream_reader.readexactly = fake_coroutine(
         asyncio.IncompleteReadError(b'part', 5))
     with self.assertRaises(aiosocks.InvalidServerReply):
         self.loop.run_until_complete(proto.read_response(4))

Example #4

File: test_protocols.py Project: bitaps-com/aiosocks

def test_base_ctor(loop):
    with pytest.raises(ValueError):
        BaseSocksProtocol(None, None, None, loop=loop,
                          waiter=None, app_protocol_factory=None)

    with pytest.raises(ValueError):
        BaseSocksProtocol(None, None, 123, loop=loop,
                          waiter=None, app_protocol_factory=None)

    with pytest.raises(ValueError):
        BaseSocksProtocol(None, None, ('python.org',), loop=loop,
                          waiter=None, app_protocol_factory=None)

Example #5

    def test_reader_limit(self):
        proto = BaseSocksProtocol(None,
                                  None, ('python.org', 80),
                                  None,
                                  None,
                                  reader_limit=10,
                                  loop=self.loop)
        self.assertEqual(proto.reader._limit, 10)

        proto = BaseSocksProtocol(None,
                                  None, ('python.org', 80),
                                  None,
                                  None,
                                  reader_limit=15,
                                  loop=self.loop)
        self.assertEqual(proto.reader._limit, 15)

Example #6

File: test_protocols.py Project: bitaps-com/aiosocks

async def test_base_incomplete_error(loop):
    proto = BaseSocksProtocol(None, None, ('python.org', 80),
                              None, None, reader_limit=10, loop=loop)
    proto._stream_reader.readexactly = make_mocked_coro(
        raise_exception=asyncio.IncompleteReadError(b'part', 5))
    with pytest.raises(aiosocks.InvalidServerReply):
        await proto.read_response(4)

Example #7

File: test_protocols.py Project: bitaps-com/aiosocks

def make_base(loop, *, dst=None, waiter=None, ap_factory=None, ssl=None):
    dst = dst or ('python.org', 80)

    proto = BaseSocksProtocol(None, None, dst=dst, ssl=ssl,
                              loop=loop, waiter=waiter,
                              app_protocol_factory=ap_factory)
    return proto

Example #8

    def test_init(self):
        with self.assertRaises(ValueError):
            BaseSocksProtocol(None,
                              None,
                              None,
                              loop=self.loop,
                              waiter=None,
                              app_protocol_factory=None)

        with self.assertRaises(ValueError):
            BaseSocksProtocol(None,
                              None,
                              123,
                              loop=self.loop,
                              waiter=None,
                              app_protocol_factory=None)

        with self.assertRaises(ValueError):
            BaseSocksProtocol(None,
                              None, ('python.org', ),
                              loop=self.loop,
                              waiter=None,
                              app_protocol_factory=None)