def test_client(self): sqs = Mock(name='sqs') x = AsyncConnection(sqs) assert x._httpclient is http.get_client() client = Mock(name='client') y = AsyncConnection(sqs, http_client=client) assert y._httpclient is client
def test_get_http_connection(self): sqs = Mock(name='sqs') x = AsyncConnection(sqs) assert isinstance( x.get_http_connection(), AsyncHTTPSConnection, ) conn = x.get_http_connection() assert conn.http_client is x._httpclient
async def fake_server(event_loop): server = FakeConnectionHmip(loop=event_loop, base_url='test.homematic.com', port=None) connector = await server.start() async with aiohttp.ClientSession(connector=connector, loop=event_loop) as session: connection = AsyncConnection(event_loop, session=session) connection.headers[ATTR_AUTH_TOKEN] = '' connection.headers[ATTR_CLIENT_AUTH] = '' yield connection await server.stop()
def test_get_http_connection(self): x = AsyncConnection(client=Mock(name='client')) self.assertIsInstance( x.get_http_connection('aws.vandelay.com', 80, False), AsyncHTTPConnection, ) self.assertIsInstance( x.get_http_connection('aws.vandelay.com', 443, True), AsyncHTTPSConnection, ) conn = x.get_http_connection('aws.vandelay.com', 80, False) self.assertIs(conn.http_client, x._httpclient) self.assertEqual(conn.host, 'aws.vandelay.com') self.assertEqual(conn.port, 80)
def test_get_http_connection(self): x = AsyncConnection(client=Mock(name='client')) assert isinstance( x.get_http_connection('aws.vandelay.com', 80, False), AsyncHTTPConnection, ) assert isinstance( x.get_http_connection('aws.vandelay.com', 443, True), AsyncHTTPSConnection, ) conn = x.get_http_connection('aws.vandelay.com', 80, False) assert conn.http_client is x._httpclient assert conn.host == 'aws.vandelay.com' assert conn.port == 80
async def fake_lookup_server(event_loop): server = FakeLookupHmip(loop=event_loop, base_url='lookup.homematic.com', port=48335) connector = await server.start() async with aiohttp.ClientSession(connector=connector, loop=event_loop) as session: connection = AsyncConnection(event_loop, session=session) yield connection await server.stop()
async def main(loop): logging.basicConfig(level=logging.DEBUG) fake_ws = FakeWebsocketHmip(loop=loop, base_url='ws.homematic.com') connector = await fake_ws.start() incoming = {} def parser(*args, **kwargs): incoming['test'] = None async with aiohttp.ClientSession(connector=connector, loop=loop) as session: connection = AsyncConnection(loop, session) connection.headers[ATTR_AUTH_TOKEN] = 'auth_token' connection.headers[ATTR_CLIENT_AUTH] = 'client_auth' connection._urlWebSocket = 'wss://ws.homematic.com/' try: ws_loop = await connection.ws_connect(parser) await ws_loop except Exception as err: pass print(incoming) await fake_ws.stop()
async def main(loop): logging.basicConfig(level=logging.DEBUG) fake_ws = FakeWebsocketHmip(loop=loop, base_url='ws.homematic.com') connector = await fake_ws.start() async with aiohttp.ClientSession(connector=connector, loop=loop) as session: connection = AsyncConnection(loop, session) # async with session.get('https://lookup.homematic.com:48335/getHost', # params={'access_token': token}) as resp: # print(await resp.json()) connection.headers[ATTR_AUTH_TOKEN] = 'auth_token' connection.headers[ATTR_CLIENT_AUTH] = 'client_auth' connection._urlWebSocket = 'wss://ws.homematic.com/' # await connection._listen_for_incoming_websocket_data( # lambda x: print(x)) connection.listen_for_websocket_data(lambda x: print(x)) # await asyncio.sleep(1) # await fake_ws.close_connection() await asyncio.sleep(600) await fake_ws.stop()
def __init__(self, loop, websession=None): super().__init__(connection=AsyncConnection(loop, websession))
def async_connection(event_loop): _connection = AsyncConnection(event_loop) yield _connection _connection._websession.close()
def fake_connection(event_loop): _connection = AsyncConnection(event_loop) _connection.api_call = AsyncMock(return_value='called') return _connection
def test_client(self): x = AsyncConnection() self.assertIs(x._httpclient, http.get_client()) client = Mock(name='client') y = AsyncConnection(http_client=client) self.assertIs(y._httpclient, client)
def test_when_boto_missing(self): with set_module_symbol('kombu.async.aws.connection', 'boto', None): with self.assertRaises(ImportError): AsyncConnection(Mock(name='client'))
def test_client(self): x = AsyncConnection() assert x._httpclient is http.get_client() client = Mock(name='client') y = AsyncConnection(http_client=client) assert y._httpclient is client
def test_when_boto_missing(self, patching): patching('kombu.async.aws.connection.boto', None) with pytest.raises(ImportError): AsyncConnection(Mock(name='client'))