Ejemplos de ClientWebSocketResponse en Python

Ejemplo n.º 1

Archivo: test_client_ws.py Proyecto: narzeja/aiohttp

def test_receive_runtime_err(loop):
    resp = ClientWebSocketResponse(mock.Mock(), mock.Mock(), mock.Mock(),
                                   mock.Mock(), 10.0, True, True, loop)
    resp._waiting = True

    with pytest.raises(RuntimeError):
        yield from resp.receive()

Ejemplo n.º 2

Archivo: kucoin_api_user_stream_data_source.py Proyecto: srirambandi/hummingbot

 async def _ping_pong_loop(self, ws: aiohttp.ClientWebSocketResponse):
     while True:
         ping_send_time = time.time()
         await ws.ping()
         await asyncio.sleep(CONSTANTS.WS_PING_HEARTBEAT)
         if self._last_recv_time < ping_send_time:
             ws._pong_not_received()
             raise asyncio.TimeoutError

Ejemplo n.º 3

    def test_receive_runtime_err(self):
        resp = ClientWebSocketResponse(mock.Mock(), mock.Mock(), mock.Mock(),
                                       mock.Mock(), 10.0, True, True,
                                       self.loop)
        resp._waiting = True

        self.assertRaises(RuntimeError, self.loop.run_until_complete,
                          resp.receive())

Ejemplo n.º 4

Archivo: test_client_ws.py Proyecto: cr0hn/aiohttp

def test_receive_runtime_err(loop):
    resp = ClientWebSocketResponse(
        mock.Mock(), mock.Mock(), mock.Mock(), mock.Mock(), 10.0,
        True, True, loop)
    resp._waiting = True

    with pytest.raises(RuntimeError):
        yield from resp.receive()

Ejemplo n.º 5

Archivo: test_client_ws.py Proyecto: przerull/aiohttp

    def test_receive_runtime_err(self):
        resp = ClientWebSocketResponse(
            mock.Mock(), mock.Mock(), mock.Mock(), mock.Mock(), 10.0,
            True, True, self.loop)
        resp._waiting = True

        self.assertRaises(
            RuntimeError, self.loop.run_until_complete, resp.receive())

Ejemplo n.º 6

Archivo: common.py Proyecto: natekspencer/pylitterbot

async def get_account(logged_in: bool = False,
                      load_robots: bool = False) -> Account:
    """Get an account that has the underlying API patched."""
    with patch(
            "pylitterbot.session.ClientSession.ws_connect",
            return_value=ClientWebSocketResponse(Mock(), Mock(), Mock(),
                                                 Mock(), Mock(), Mock(),
                                                 Mock(), Mock()),
    ):
        account = Account()
        if logged_in:
            await account.connect(username=USERNAME,
                                  password=PASSWORD,
                                  load_robots=load_robots)
        return account

Ejemplo n.º 7

Archivo: mexc_api_user_stream_data_source.py Proyecto: sniperswang/hummingbot

 async def _inner_messages(
         self, ws: aiohttp.ClientWebSocketResponse) -> AsyncIterable[str]:
     try:
         while True:
             msg = await asyncio.wait_for(ws.receive(),
                                          timeout=self.MESSAGE_TIMEOUT)
             if msg.type == aiohttp.WSMsgType.CLOSED:
                 raise ConnectionError
             yield json.loads(msg.data)
     except asyncio.TimeoutError:
         return
     except ConnectionClosed:
         return
     except ConnectionError:
         return

Ejemplo n.º 8

 async def _subscribe_to_streams(self, ws: ClientWebSocketResponse):
     coros = (ws.send_json(key) for key in self._subscription_keys())
     await asyncio.gather(*coros)