ydb/contrib/python/pytest-localserver/py3/tests/test_https.py

import pytest
import requests

from pytest_localserver import https
from pytest_localserver import plugin


# define test fixture here again in order to run tests without having to
# install the plugin anew every single time
httpsserver = plugin.httpsserver


def test_httpsserver_funcarg(httpsserver):
    assert isinstance(httpsserver, https.SecureContentServer)
    assert httpsserver.is_alive()
    assert httpsserver.server_address


def test_server_does_not_serve_file_at_startup(httpsserver):
    assert httpsserver.code == 204
    assert httpsserver.content == ""


def test_some_content_retrieval(httpsserver):
    httpsserver.serve_content("TEST!")
    resp = requests.get(httpsserver.url, verify=False)
    assert resp.text == "TEST!"
    assert resp.status_code == 200


def test_GET_request(httpsserver):
    httpsserver.serve_content("TEST!", headers={"Content-type": "text/plain"})
    resp = requests.get(httpsserver.url, headers={"User-Agent": "Test method"}, verify=False)
    assert resp.text == "TEST!"
    assert resp.status_code == 200
    assert "text/plain" in resp.headers["Content-type"]


def test_HEAD_request(httpsserver):
    httpsserver.serve_content("TEST!", headers={"Content-type": "text/plain"})
    print(httpsserver.url)
    resp = requests.head(httpsserver.url, verify=False)
    assert resp.status_code == 200
    assert resp.headers["Content-type"] == "text/plain"


def test_client_does_not_trust_self_signed_certificate(httpsserver):
    httpsserver.serve_content("TEST!", headers={"Content-type": "text/plain"})
    with pytest.raises(requests.exceptions.SSLError, match="CERTIFICATE_VERIFY_FAILED"):
        requests.get(httpsserver.url, verify=True)


def test_add_server_certificate_to_client_trust_chain(httpsserver):
    httpsserver.serve_content("TEST!", headers={"Content-type": "text/plain"})
    resp = requests.get(httpsserver.url, verify=httpsserver.certificate)
    assert resp.status_code == 200
    assert resp.headers["Content-type"] == "text/plain"