Ejemplo n.º 1
0
    async def test_cancel_raised_cancelled_error(self):
        app = web.Application()

        request_made = asyncio.Event()

        async def handle_get(_):
            request_made.set()
            await asyncio.Future()

        app.add_routes([web.get('/page', handle_get)])

        runner = web.AppRunner(app)
        await runner.setup()
        self.add_async_cleanup(runner.cleanup)
        site = web.TCPSite(runner, '0.0.0.0', 8080)
        await site.start()

        request, close = Pool()
        self.add_async_cleanup(close)

        request = asyncio.ensure_future(
            request(b'GET', 'http://localhost:8080/page'))
        await request_made.wait()
        request.cancel()
        with self.assertRaises(asyncio.CancelledError):
            await request
Ejemplo n.º 2
0
    async def test_http_chunked_responses(self):
        response_datas = []

        data = b'abcdefghijklmnopqrstuvwxyz'
        chunk_size = None

        async def handle_get(request):
            await request.content.read()
            response = web.StreamResponse()
            await response.prepare(request)

            for chars in [
                    data[i:i + chunk_size]
                    for i in range(0, len(data), chunk_size)
            ]:
                await response.write(chars)
                await asyncio.sleep(0)

            return response

        app = web.Application()
        app.add_routes([web.get('/page', handle_get)])
        runner = web.AppRunner(app)
        await runner.setup()
        self.add_async_cleanup(runner.cleanup)
        site = web.TCPSite(runner, '0.0.0.0', 8080)
        await site.start()

        request, close = Pool()
        self.add_async_cleanup(close)

        for recv_bufsize, keep_alive_timeout in itertools.product(
            (1, 26, 16384), (0, 15)):
            request, close = Pool(recv_bufsize=recv_bufsize,
                                  keep_alive_timeout=keep_alive_timeout)
            self.add_async_cleanup(close)
            for chunk_size in range(1, 27):
                _, headers, body = await request(
                    b'GET',
                    'http://localhost:8080/page',
                )
                self.assertEqual(
                    dict(headers)[b'transfer-encoding'], b'chunked')
                response_data = b''
                async for body_bytes in body:
                    response_data += body_bytes
                response_datas.append(response_data)

        self.assertEqual(response_datas, [data] * 26 * 3 * 2)
Ejemplo n.º 3
0
    async def test_http_retry_then_succeed(self):
        async def handle_post(_):
            return web.Response(text='the-response-data')

        app = web.Application()
        app.add_routes([web.post('/page', handle_post)])
        runner = web.AppRunner(app)

        request, close = Pool()
        self.add_async_cleanup(close)

        retriable_request = retry(
            request,
            exception_intervals=((HttpConnectionError, (0, 1, 2)), ),
        )

        async def delayed_start():
            await asyncio.sleep(0.5)
            await runner.setup()
            self.add_async_cleanup(runner.cleanup)
            site = web.TCPSite(runner, '0.0.0.0', 8080)
            await site.start()

        asyncio.ensure_future(delayed_start())
        _, _, body = await retriable_request(b'POST',
                                             'http://localhost:8080/page',
                                             body=streamed(b'some-data'),
                                             headers=((b'content-length',
                                                       b'9'), ))

        self.assertEqual(b'the-response-data', await buffered(body))
Ejemplo n.º 4
0
    async def test_http_timeout(self):
        loop = asyncio.get_event_loop()
        app = web.Application()

        async def handle_get(_):
            await asyncio.Future()

        app.add_routes([web.get('/page', handle_get)])

        runner = web.AppRunner(app)
        await runner.setup()
        self.add_async_cleanup(runner.cleanup)
        site = web.TCPSite(runner, '0.0.0.0', 8080)
        await site.start()

        with FastForward(loop) as forward:

            request, close = Pool()
            self.add_async_cleanup(close)

            request = asyncio.ensure_future(
                request(b'GET', 'http://localhost:8080/page'))
            with self.assertRaises(HttpDataError) as cm:
                at_10 = forward(10)
                await request

            await at_10
        self.assertIsInstance(cm.exception.__cause__, asyncio.TimeoutError)
Ejemplo n.º 5
0
    async def test_get_301(self):
        async def handle_get_a(_):
            return web.Response(
                status=301,
                headers={'location': '/b'},
            )

        async def handle_get_b(_):
            return web.Response(body=b'def')

        app = web.Application()
        app.add_routes([
            web.get('/a', handle_get_a),
            web.get('/b', handle_get_b),
        ])
        runner = web.AppRunner(app)
        await runner.setup()
        self.add_async_cleanup(runner.cleanup)
        site = web.TCPSite(runner, '0.0.0.0', 8080)
        await site.start()

        request, close = Pool()
        self.add_async_cleanup(close)

        redirectable_request = redirectable(request)
        response_status, _, response_body = await redirectable_request(
            b'GET',
            'http://localhost:8080/a',
        )
        response_body_buffered = await buffered(response_body)

        self.assertEqual(response_body_buffered, b'def')
        self.assertEqual(response_status, b'200')
Ejemplo n.º 6
0
    async def test_http_delete_200_no_content_length_no_body_keep_alive(self):
        async def handle_delete(request):
            await request.content.read()
            response = web.StreamResponse(status=200,
                                          headers={'connection': 'keep-alive'})
            await response.prepare(request)
            return web.Response(status=200)

        app = web.Application()
        app.add_routes([web.delete('/page', handle_delete)])
        runner = web.AppRunner(app)
        await runner.setup()
        self.add_async_cleanup(runner.cleanup)
        site = web.TCPSite(runner, '0.0.0.0', 8080)
        await site.start()

        request, close = Pool(http_version=b'HTTP/1.0')
        self.add_async_cleanup(close)
        code, headers, body = await request(b'DELETE',
                                            'http://localhost:8080/page')
        self.assertEqual(code, b'200')
        self.assertNotIn('content-length', dict(headers))
        self.assertNotIn('transfer-encoding', dict(headers))
        self.assertEqual(b'keep-alive', dict(headers)[b'connection'])
        body_bytes = await buffered(body)
        self.assertEqual(body_bytes, b'')
Ejemplo n.º 7
0
    async def test_http_get_chunked(self):
        request, close = Pool()
        self.add_async_cleanup(close)

        _, _, body = await request(
            b'GET',
            'http://postman-echo.com/stream/1000',
            params=(('some', 'arg'), ),
        )
        body_bytes = await buffered(body)

        # Slightly odd response: a concatanation of identical JSON objects
        part_length = int(len(body_bytes) / 1000)
        parts = [
            body_bytes[i:i + part_length]
            for i in range(0, len(body_bytes), part_length)
        ]

        self.assertEqual(len(parts), 1000)

        for part in parts:
            response_dict = json.loads(part)
            self.assertEqual(response_dict['headers']['host'],
                             'postman-echo.com')
            self.assertEqual(response_dict['args'], {
                'n': '1000',
                'some': 'arg'
            })
Ejemplo n.º 8
0
    async def test_https_post_small(self):
        request, close = Pool()
        self.add_async_cleanup(close)

        async def data():
            yield b'some-data=something'

        code, headers, body = await request(
            b'POST',
            'https://postman-echo.com/post',
            params=(
                ('some', 'value'),
                ('?=&', '/&'),
            ),
            headers=(
                (b'content-length', b'19'),
                (b'content-type', b'application/x-www-form-urlencoded'),
            ),
            body=data,
        )
        body_bytes = b''
        async for chunk in body:
            body_bytes += chunk

        headers_dict = dict(headers)
        response_dict = json.loads(body_bytes)

        self.assertEqual(code, b'200')
        self.assertEqual(headers_dict[b'content-type'],
                         b'application/json; charset=utf-8')
        self.assertEqual(response_dict['args'], {'some': 'value', '?=&': '/&'})
        self.assertEqual(response_dict['headers']['host'], 'postman-echo.com')
        self.assertEqual(response_dict['headers']['content-length'], '19')
        self.assertEqual(response_dict['form'], {'some-data': 'something'})
Ejemplo n.º 9
0
    async def test_http_post_small_buffered_streamed(self):
        request, close = Pool()
        self.add_async_cleanup(close)

        code, headers, body = await request(
            b'POST',
            'http://postman-echo.com/post',
            (),
            (
                (b'content-length', b'19'),
                (b'content-type', b'application/x-www-form-urlencoded'),
            ),
            streamed(b'some-data=something'),
        )
        body_bytes = await buffered(body)

        headers_dict = dict(headers)
        response_dict = json.loads(body_bytes)

        self.assertEqual(code, b'200')
        self.assertEqual(headers_dict[b'content-type'],
                         b'application/json; charset=utf-8')
        self.assertEqual(response_dict['headers']['host'], 'postman-echo.com')
        self.assertEqual(response_dict['headers']['content-length'], '19')
        self.assertEqual(response_dict['form'], {'some-data': 'something'})
Ejemplo n.º 10
0
async def async_main(logger, target_file, url, username, password):
    request, _ = Pool()

    while True:
        await asyncio.sleep(10)
        try:
            logger.debug("Fetching from %s", url)
            headers = ((
                b"Authorization",
                b"Basic " +
                base64.b64encode(f"{username}:{password}".encode("ascii")),
            ), )
            code, _, body = await request(b"GET", url, headers=headers)
            logger.debug("Code %s", code)
            raw_json = await buffered(body)
            logger.debug("Received %s", raw_json)
            applications = json.loads(raw_json)["applications"]
            logger.debug("Found %s", applications)
            file_sd_config = [{
                "labels": {
                    "job": "tools",
                    "tool_name": application["name"],
                    "user": application["user"],
                },
                "targets": [
                    urllib.parse.urlsplit(application["proxy_url"]).hostname +
                    ":8889"
                ],
            } for application in applications
                              if application["proxy_url"] is not None]
            logger.debug("Saving %s to %s", file_sd_config, target_file)
            with open(target_file, "w") as file:
                file.write(json.dumps(file_sd_config))
        except Exception:  # pylint: disable=broad-except
            logger.exception("Exception fetching targets")
Ejemplo n.º 11
0
    async def test_get_301_too_many_redirects(self):
        async def handle_get_a(_):
            return web.Response(
                status=301,
                headers={'location': '/b'},
            )

        async def handle_get_b(_):
            return web.Response(
                status=301,
                headers={'location': '/a'},
            )

        app = web.Application()
        app.add_routes([
            web.get('/a', handle_get_a),
            web.get('/b', handle_get_b),
        ])
        runner = web.AppRunner(app)
        await runner.setup()
        self.add_async_cleanup(runner.cleanup)
        site = web.TCPSite(runner, '0.0.0.0', 8080)
        await site.start()

        request, close = Pool()
        self.add_async_cleanup(close)

        redirectable_request = redirectable(request)
        with self.assertRaises(HttpTooManyRedirects):
            await redirectable_request(
                b'GET',
                'http://localhost:8080/a',
            )
Ejemplo n.º 12
0
    async def test_http_post(self):
        posted_data_received = b''

        async def handle_get(request):
            nonlocal posted_data_received
            posted_data_received = await request.content.read()
            return web.Response(status=200)

        app = web.Application()
        app.add_routes([web.post('/page', handle_get)])
        runner = web.AppRunner(app)
        await runner.setup()
        self.add_async_cleanup(runner.cleanup)
        site = web.TCPSite(runner, '0.0.0.0', 8080)
        await site.start()

        async def data():
            for _ in range(10):
                yield b'abc' * 1000000

        content_length = str(1000000 * 10 * 3).encode()

        request, close = Pool()
        self.add_async_cleanup(close)
        _, _, body = await request(
            b'POST',
            'http://localhost:8080/page',
            (),
            ((b'content-length', content_length), ),
            data,
        )
        async for _ in body:
            pass
        self.assertEqual(posted_data_received, b'abc' * 1000000 * 10)
Ejemplo n.º 13
0
    async def test_get_301_same_domain_auth_preserved(self):
        auth_b = None

        async def handle_get_a(_):
            return web.Response(
                status=301,
                headers={'location': '/b'},
            )

        async def handle_get_b(request):
            nonlocal auth_b
            auth_b = request.headers['authorization']
            return web.Response()

        app = web.Application()
        app.add_routes([
            web.get('/a', handle_get_a),
            web.get('/b', handle_get_b),
        ])
        runner = web.AppRunner(app)
        await runner.setup()
        self.add_async_cleanup(runner.cleanup)
        site = web.TCPSite(runner, '0.0.0.0', 8080)
        await site.start()

        request, close = Pool()
        self.add_async_cleanup(close)

        redirectable_request = redirectable(request)
        _, _, body = await redirectable_request(b'GET',
                                                'http://localhost:8080/a',
                                                headers=((b'Authorization',
                                                          b'the-key'), ))
        await buffered(body)
        self.assertEqual(auth_b, 'the-key')
Ejemplo n.º 14
0
    async def test_http_post_204_with_content(self):
        # Against the spec, but that's the server's fault: we still need to deal with it

        async def handle_post(request):
            await request.content.read()
            response = web.StreamResponse(status=204,
                                          headers={'content-length': '12'})
            await response.prepare(request)
            await response.write(b'some-content')
            return response

        app = web.Application()
        app.add_routes([web.post('/page', handle_post)])
        runner = web.AppRunner(app)
        await runner.setup()
        self.add_async_cleanup(runner.cleanup)
        site = web.TCPSite(runner, '0.0.0.0', 8080)
        await site.start()

        request, close = Pool()
        self.add_async_cleanup(close)
        code, _, body = await request(b'POST', 'http://localhost:8080/page')
        self.assertEqual(code, b'204')
        body_bytes = await buffered(body)
        self.assertEqual(body_bytes, b'some-content')
Ejemplo n.º 15
0
async def async_main(logger):
    # Suspect that S3 has a very small keep-alive value
    request_non_redirectable, close_pool = Pool(keep_alive_timeout=4)
    request = redirectable(request_non_redirectable)

    credentials = get_ecs_role_credentials(
        'http://169.254.170.2' +
        os.environ['AWS_CONTAINER_CREDENTIALS_RELATIVE_URI'])
    bucket = S3Bucket(
        region=os.environ['MIRRORS_BUCKET_REGION'],
        host=os.environ['MIRRORS_BUCKET_HOST'],
        name=os.environ['MIRRORS_BUCKET_NAME'],
    )
    s3_context = S3Context(request=request,
                           credentials=credentials,
                           bucket=bucket)

    if os.environ['MIRROR_ANACONDA_R'] == 'True':
        await conda_mirror(logger, request, s3_context,
                           'https://conda.anaconda.org/r/', 'r/')

    if os.environ['MIRROR_ANACONDA_CONDA_FORGE'] == 'True':
        await conda_mirror(
            logger,
            request,
            s3_context,
            'https://conda.anaconda.org/conda-forge/',
            'conda-forge/',
        )

    if os.environ['MIRROR_ANACONDA_CONDA_ANACONDA'] == 'True':
        await conda_mirror(
            logger,
            request,
            s3_context,
            'https://conda.anaconda.org/anaconda/',
            'anaconda/',
        )

    if os.environ['MIRROR_CRAN'] == 'True':
        await cran_mirror(logger, request, s3_context)

    if os.environ['MIRROR_PYPI'] == 'True':
        await pypi_mirror(logger, request, s3_context)

    if os.environ['MIRROR_DEBIAN'] == 'True':
        await debian_mirror(
            logger,
            request,
            s3_context,
            'https://cloudfront.debian.net/debian/',
            'debian/',
        )

    if os.environ['MIRROR_NLTK'] == 'True':
        await nltk_mirror(logger, request, s3_context, 'nltk/')

    await close_pool()
    await asyncio.sleep(0)
Ejemplo n.º 16
0
    async def test_http_get_small_via_ip_address(self):
        request, close = Pool()
        self.add_async_cleanup(close)

        _, _, body = await request(b'GET', 'http://212.183.159.230/5MB.zip')
        m = hashlib.md5()
        async for chunk in body:
            m.update(chunk)

        self.assertEqual(m.hexdigest(), 'b3215c06647bc550406a9c8ccc378756')
Ejemplo n.º 17
0
    async def test_http_get_small_via_dns(self):
        request, close = Pool()
        self.add_async_cleanup(close)

        _, _, body = await request(b'GET', 'http://www.ovh.net/files/1Mio.dat')
        m = hashlib.md5()
        async for chunk in body:
            m.update(chunk)

        self.assertEqual(m.hexdigest(), '6cb91af4ed4c60c11613b75cd1fc6116')
Ejemplo n.º 18
0
    async def test_post_307_chain(self):
        body_c = None

        async def handle_post_a(request):
            await request.content.read()
            return web.Response(
                status=307,
                headers={'location': '/b'},
            )

        async def handle_post_b(request):
            await request.content.read()
            return web.Response(
                status=307,
                headers={'location': '/c'},
            )

        async def handle_post_c(request):
            nonlocal body_c
            body_c = await request.content.read()
            return web.Response(body=b'def')

        app = web.Application()
        app.add_routes([
            web.post('/a', handle_post_a),
            web.post('/b', handle_post_b),
            web.post('/c', handle_post_c),
        ])
        runner = web.AppRunner(app)
        await runner.setup()
        self.add_async_cleanup(runner.cleanup)
        site = web.TCPSite(runner, '0.0.0.0', 8080)
        await site.start()

        request, close = Pool()
        self.add_async_cleanup(close)

        async def data():
            yield b'a'
            yield b'b'
            yield b'c'

        redirectable_request = redirectable(request)
        response_status, _, response_body = await redirectable_request(
            b'POST',
            'http://localhost:8080/a',
            headers=((b'content-length', b'3'), ),
            body=data,
        )
        response_body_buffered = await buffered(response_body)

        self.assertEqual(body_c, b'abc')
        self.assertEqual(response_body_buffered, b'def')
        self.assertEqual(response_status, b'200')
Ejemplo n.º 19
0
    async def test_http_retry_fail_immediately_b(self):

        request, close = Pool()
        self.add_async_cleanup(close)

        retriable_request = retry(
            request,
            exception_intervals=(),
        )
        with self.assertRaises(HttpConnectionError):
            await retriable_request(b'GET', 'http://localhost:8080/page')
Ejemplo n.º 20
0
    async def test_http_identity_close_no_content_length_responses(self):
        response_datas = []

        data = b'abcdefghijklmnopqrstuvwxyz'
        chunk_size = None

        async def handle_get(request):
            await request.content.read()
            response = web.StreamResponse()
            response.force_close()
            await response.prepare(request)

            for chars in [
                    data[i:i + chunk_size]
                    for i in range(0, len(data), chunk_size)
            ]:
                await response.write(chars)
                await asyncio.sleep(0)

            return response

        app = web.Application()
        app.add_routes([web.get('/page', handle_get)])
        runner = web.AppRunner(app)
        await runner.setup()
        self.add_async_cleanup(runner.cleanup)
        site = web.TCPSite(runner, '0.0.0.0', 8080)
        await site.start()

        combinations = itertools.product((1, 26, 16384),
                                         (b'close', b'keep-alive'), (0, 15))
        for recv_bufsize, connection, keep_alive_timeout in combinations:
            request, close = Pool(recv_bufsize=recv_bufsize,
                                  keep_alive_timeout=keep_alive_timeout,
                                  http_version=b'HTTP/1.0')
            self.add_async_cleanup(close)

            for chunk_size in range(1, 27):
                _, _, body = await request(
                    b'GET',
                    'http://localhost:8080/page',
                    headers=((b'connection', connection), ),
                )
                response_data = b''
                async for body_bytes in body:
                    response_data += body_bytes
                response_datas.append(response_data)

        self.assertEqual(response_datas, [data] * 26 * 3 * 2 * 2)
Ejemplo n.º 21
0
    async def test_http_head_200(self):
        async def handle_head(request):
            await request.content.read()
            return web.Response(status=200, headers={'content-length': '20'})

        app = web.Application()
        app.add_routes([web.head('/page', handle_head)])
        runner = web.AppRunner(app)
        await runner.setup()
        self.add_async_cleanup(runner.cleanup)
        site = web.TCPSite(runner, '0.0.0.0', 8080)
        await site.start()

        request, close = Pool()
        self.add_async_cleanup(close)
        code, _, body = await request(b'HEAD', 'http://localhost:8080/page')
        await buffered(body)
        self.assertEqual(code, b'200')
Ejemplo n.º 22
0
    async def test_after_keep_alive_timeout(self):
        loop = asyncio.get_event_loop()
        app = web.Application()

        bodies = iter([b'abcd', b'efgh', b'ijkl'])
        remote_ports = []

        async def handle_get(request):
            remote_ports.append(request.transport._sock.getpeername()[1])
            return web.Response(status=200, body=next(bodies))

        app.add_routes([web.get('/page', handle_get)])

        runner = web.AppRunner(app)
        await runner.setup()
        self.add_async_cleanup(runner.cleanup)
        site = web.TCPSite(runner, '0.0.0.0', 8080)
        await site.start()

        with FastForward(loop) as forward:

            request, close = Pool()
            self.add_async_cleanup(close)

            _, _, body_1_stream = await request(b'GET',
                                                'http://localhost:8080/page')
            body_1 = await buffered(body_1_stream)

            await forward(15)

            _, _, body_2_stream = await request(b'GET',
                                                'http://localhost:8080/page')
            body_2 = await buffered(body_2_stream)

            _, _, body_3_stream = await request(b'GET',
                                                'http://localhost:8080/page')
            body_3 = await buffered(body_3_stream)

        self.assertEqual(body_1, b'abcd')
        self.assertEqual(body_2, b'efgh')
        self.assertEqual(body_3, b'ijkl')

        self.assertNotEqual(remote_ports[0], remote_ports[1])
        self.assertEqual(remote_ports[1], remote_ports[2])
Ejemplo n.º 23
0
    async def test_http_post_204_without_content(self):
        async def handle_post(request):
            await request.content.read()
            return web.Response(status=204)

        app = web.Application()
        app.add_routes([web.post('/page', handle_post)])
        runner = web.AppRunner(app)
        await runner.setup()
        self.add_async_cleanup(runner.cleanup)
        site = web.TCPSite(runner, '0.0.0.0', 8080)
        await site.start()

        request, close = Pool()
        self.add_async_cleanup(close)
        code, _, body = await request(b'POST', 'http://localhost:8080/page')
        self.assertEqual(code, b'204')
        body_bytes = await buffered(body)
        self.assertEqual(body_bytes, b'')
Ejemplo n.º 24
0
    async def test_http_delete_200_with_body(self):
        async def handle_delete(request):
            await request.content.read()
            return web.Response(status=200, body=b'some-data')

        app = web.Application()
        app.add_routes([web.delete('/page', handle_delete)])
        runner = web.AppRunner(app)
        await runner.setup()
        self.add_async_cleanup(runner.cleanup)
        site = web.TCPSite(runner, '0.0.0.0', 8080)
        await site.start()

        request, close = Pool()
        self.add_async_cleanup(close)
        code, _, body = await request(b'DELETE', 'http://localhost:8080/page')
        self.assertEqual(code, b'200')
        body_bytes = await buffered(body)
        self.assertEqual(body_bytes, b'some-data')
Ejemplo n.º 25
0
    async def test_get_301_different_domain_auth_lost(self):
        auth_b = None

        async def handle_get_a(_):
            return web.Response(
                status=301,
                headers={'location': 'http://anotherhost.com:8080/b'},
            )

        async def handle_get_b(request):
            nonlocal auth_b
            auth_b = request.headers.get('authorization', None)
            return web.Response(body=b'def')

        app = web.Application()
        app.add_routes([
            web.get('/a', handle_get_a),
            web.get('/b', handle_get_b),
        ])
        runner = web.AppRunner(app)
        await runner.setup()
        self.add_async_cleanup(runner.cleanup)
        site = web.TCPSite(runner, '0.0.0.0', 8080)
        await site.start()

        def get_dns_resolver():
            async def get_host(_, __, ___):
                return IPv4AddressExpiresAt('127.0.0.1', expires_at=0)

            return Resolver(get_host=get_host, )

        request, close = Pool(get_dns_resolver=get_dns_resolver)
        self.add_async_cleanup(close)

        redirectable_request = redirectable(request)
        _, _, body = await redirectable_request(b'GET',
                                                'http://localhost:8080/a',
                                                headers=((b'authorization',
                                                          b'the-key'), ))
        response = await buffered(body)
        self.assertEqual(auth_b, None)
        self.assertEqual(response, b'def')
Ejemplo n.º 26
0
    async def test_http_large_header(self):
        posted_data_received = b''

        async def handle_get(request):
            nonlocal posted_data_received
            posted_data_received = await request.content.read()
            return web.Response(status=200, headers={'example': 'b' * 32000})

        app = web.Application()
        app.add_routes([web.get('/page', handle_get)])
        runner = web.AppRunner(app)
        await runner.setup()
        self.add_async_cleanup(runner.cleanup)
        site = web.TCPSite(runner, '0.0.0.0', 8080)
        await site.start()

        request, close = Pool()
        self.add_async_cleanup(close)

        with self.assertRaises(HttpHeaderTooLong):
            await request(b'GET', 'http://localhost:8080/page')
Ejemplo n.º 27
0
async def async_main(logger, target_file, url, username, password):
    request, _ = Pool()

    while True:
        await asyncio.sleep(10)
        try:
            logger.debug('Fetching from %s', url)
            headers = (
                (
                    b'Authorization',
                    b'Basic '
                    + base64.b64encode(f'{username}:{password}'.encode('ascii')),
                ),
            )
            code, _, body = await request(b'GET', url, headers=headers)
            logger.debug('Code %s', code)
            raw_json = await buffered(body)
            logger.debug('Received %s', raw_json)
            applications = json.loads(raw_json)['applications']
            logger.debug('Found %s', applications)
            file_sd_config = [
                {
                    'labels': {
                        'job': 'tools',
                        'tool_name': application['name'],
                        'user': application['user'],
                    },
                    'targets': [
                        urllib.parse.urlsplit(application['proxy_url']).hostname
                        + ':8889'
                    ],
                }
                for application in applications
                if application['proxy_url'] is not None
            ]
            logger.debug('Saving %s to %s', file_sd_config, target_file)
            with open(target_file, 'w') as file:
                file.write(json.dumps(file_sd_config))
        except Exception:
            logger.exception('Exception fetching targets')
Ejemplo n.º 28
0
    async def test_ssl_self_signed_fails_by_default(self):
        loop = asyncio.get_event_loop()

        def handler(_, __):
            pass

        loop.set_exception_handler(handler)

        ssl_context = ssl.SSLContext(ssl.PROTOCOL_TLSv1_2)
        ssl_context.load_cert_chain('public.crt', keyfile='private.key')
        app = web.Application()

        runner = web.AppRunner(app)
        await runner.setup()
        self.add_async_cleanup(runner.cleanup)
        site = web.TCPSite(runner, '0.0.0.0', 8080, ssl_context=ssl_context)
        await site.start()

        request, close = Pool()
        self.add_async_cleanup(close)
        with self.assertRaises(HttpTlsError) as cm:
            await request(b'GET', 'https://localhost:8080/page')

        self.assertIsInstance(cm.exception.__cause__, ssl.SSLError)
Ejemplo n.º 29
0
async def async_main(logger):
    # Suspect that S3 has a very small keep-alive value
    request_non_redirectable, close_pool = Pool(keep_alive_timeout=4,
                                                socket_timeout=30)
    request = redirectable(request_non_redirectable)

    credentials = get_ecs_role_credentials(
        "http://169.254.170.2" +
        os.environ["AWS_CONTAINER_CREDENTIALS_RELATIVE_URI"])
    bucket = S3Bucket(
        region=os.environ["MIRRORS_BUCKET_REGION"],
        host=os.environ["MIRRORS_BUCKET_HOST"],
        name=os.environ["MIRRORS_BUCKET_NAME"],
    )
    s3_context = S3Context(request=request,
                           credentials=credentials,
                           bucket=bucket)

    if os.environ["MIRROR_ANACONDA_R"] == "True":
        await conda_mirror(logger, request, s3_context,
                           "https://conda.anaconda.org/r/", "r/")

    if os.environ["MIRROR_ANACONDA_CONDA_FORGE"] == "True":
        await conda_mirror(
            logger,
            request,
            s3_context,
            "https://conda.anaconda.org/conda-forge/",
            "conda-forge/",
        )

    if os.environ["MIRROR_ANACONDA_CONDA_ANACONDA"] == "True":
        await conda_mirror(
            logger,
            request,
            s3_context,
            "https://conda.anaconda.org/anaconda/",
            "anaconda/",
        )

    if os.environ["MIRROR_CRAN"] == "True":
        await cran_mirror(logger, request, s3_context)

    if os.environ["MIRROR_PYPI"] == "True":
        await pypi_mirror(logger, request, s3_context)

    if os.environ["MIRROR_DEBIAN"] == "True":
        await debian_mirror(
            logger,
            request,
            s3_context,
            "https://cloudfront.debian.net/debian-security/",
            "debian-security/",
        )
        await debian_mirror(
            logger,
            request,
            s3_context,
            "https://cloudfront.debian.net/debian/",
            "debian/",
        )

    if os.environ["MIRROR_NLTK"] == "True":
        await nltk_mirror(logger, request, s3_context, "nltk/")

    await close_pool()
    await asyncio.sleep(0)