File: test_custom_ca.py

package info (click to toggle)
python-truststore 0.10.1-1
  • links: PTS, VCS
  • area: main
  • in suites: sid, trixie
  • size: 236 kB
  • sloc: python: 1,835; makefile: 13; sh: 6
file content (44 lines) | stat: -rw-r--r-- 1,288 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
import asyncio
import ssl

import pytest
import requests
import urllib3
from aiohttp import ClientSession

import truststore
from tests import SSLContextAdapter
from tests.conftest import Server


@pytest.mark.asyncio
async def test_urllib3_custom_ca(server: Server) -> None:
    def test_urllib3():
        ctx = truststore.SSLContext(ssl.PROTOCOL_TLS_CLIENT)
        with urllib3.PoolManager(ssl_context=ctx) as client:
            resp = client.request("GET", server.base_url)
        assert resp.status == 200

    thread = asyncio.to_thread(test_urllib3)
    await thread


@pytest.mark.asyncio
async def test_aiohttp_custom_ca(server: Server) -> None:
    ctx = truststore.SSLContext(ssl.PROTOCOL_TLS_CLIENT)
    async with ClientSession() as client:
        resp = await client.get(server.base_url, ssl=ctx)
        assert resp.status == 200


@pytest.mark.asyncio
async def test_requests_custom_ca(server: Server) -> None:
    def test_requests():
        ctx = truststore.SSLContext(ssl.PROTOCOL_TLS_CLIENT)
        with requests.Session() as http:
            http.mount("https://", SSLContextAdapter(ssl_context=ctx))
            resp = http.request("GET", server.base_url)
        assert resp.status_code == 200

    thread = asyncio.to_thread(test_requests)
    await thread