Howto¶
This documentation is a collection of the most common use cases, and their solutions. If you have not used this library before, it may be better to read the Tutorial first.
Matching query parameters¶
To match query parameters, you must not included them to the URI, as this will not work:
def test_query_params(httpserver):
httpserver.expect_request("/foo?user=bar") # never do this
There’s an explicit place where the query string should go:
def test_query_params(httpserver):
httpserver.expect_request("/foo", query_string="user=bar")
The query_string
is the parameter which does not contain the leading
question mark ?
.
Note
The reason behind this is the underlying http server library werkzeug,
which provides the Request
object which is used for the matching the
request with the handlers. This object has the query_string
attribute
which contains the query.
As the order of the parameters in the query string usually does not matter, you
can specify a dict for the query_string
parameter (the naming may look a bit
strange but we wanted to keep API compatibility and this dict matching feature
was added later).
import requests
def test_query_params(httpserver):
httpserver.expect_request("/foo", query_string={"user": "user1", "group": "group1"}).respond_with_data("OK")
assert requests.get(httpserver.url_for("/foo?user=user1&group=group1")).status_code == 200
assert requests.get(httpserver.url_for("/foo?group=group1&user=user1")).status_code == 200
In the example above, both requests pass the test as we specified the expected query string as a dictionary.
Behind the scenes an additional step is done by the library: it parses up the query_string into the dict and then compares it with the dict provided.
URI matching¶
The simplest form of URI matching is providing as a string. This is a equality match, if the URI of the request is not equal with the specified one, the request will not be handled.
If this is not desired, you can specify a regexp object (returned by the
re.compile()
call).
import re
import requests
def test_httpserver_with_regexp(httpserver):
httpserver.expect_request(re.compile("^/foo"), method="GET")
requests.get(httpserver.url_for("/foobar"))
The above will match every URI starting with “/foo”.
There’s an additional way to extend this functionality. You can specify your own
method which will receive the URI. All you need is to subclass from the
URIPattern
class and define the match()
method which will get the uri as
string and should return a boolean value.
from pytest_httpserver import HTTPServer
from pytest_httpserver import URIPattern
class PrefixMatch(URIPattern):
def __init__(self, prefix: str):
self.prefix = prefix
def match(self, uri):
return uri.startswith(self.prefix)
def test_uripattern_object(httpserver: HTTPServer):
httpserver.expect_request(PrefixMatch("/foo")).respond_with_json({"foo": "bar"})
Authentication¶
When doing http digest authentication, the client may send a request like this:
GET /dir/index.html HTTP/1.0
Host: localhost
Authorization: Digest username="Mufasa",
realm="testrealm@host.com",
nonce="dcd98b7102dd2f0e8b11d0f600bfb0c093",
uri="/dir/index.html",
qop=auth,
nc=00000001,
cnonce="0a4f113b",
response="6629fae49393a05397450978507c4ef1",
opaque="5ccc069c403ebaf9f0171e9517f40e41"
Implementing a matcher is difficult for this request as the order of the
parameters in the Authorization
header value is arbitrary.
By default, pytest-httpserver includes an Authorization header parser so the
order of the parameters in the Authorization
header does not matter.
import requests
from pytest_httpserver import HTTPServer
def test_authorization_headers(httpserver: HTTPServer):
headers_with_values_in_direct_order = {
"Authorization": (
'Digest username="Mufasa",'
'realm="testrealm@host.com",'
'nonce="dcd98b7102dd2f0e8b11d0f600bfb0c093",'
'uri="/dir/index.html",'
"qop=auth,"
"nc=00000001,"
'cnonce="0a4f113b",'
'response="6629fae49393a05397450978507c4ef1",'
'opaque="5ccc069c403ebaf9f0171e9517f40e41"'
)
}
httpserver.expect_request(uri="/", headers=headers_with_values_in_direct_order).respond_with_data("OK")
response = requests.get(httpserver.url_for("/"), headers=headers_with_values_in_direct_order)
assert response.status_code == 200
assert response.text == "OK"
headers_with_values_in_modified_order = {
"Authorization": (
"Digest qop=auth,"
'username="Mufasa",'
'nonce="dcd98b7102dd2f0e8b11d0f600bfb0c093",'
'uri="/dir/index.html",'
"nc=00000001,"
'realm="testrealm@host.com",'
'response="6629fae49393a05397450978507c4ef1",'
'cnonce="0a4f113b",'
'opaque="5ccc069c403ebaf9f0171e9517f40e41"'
)
}
response = requests.get(httpserver.url_for("/"), headers=headers_with_values_in_modified_order)
assert response.status_code == 200
assert response.text == "OK"
JSON matching¶
Matching the request data can be done in two different ways. One way is to provide a python string (or bytes object) whose value will be compared to the request body.
When the request contains a json, matching to will be error prone as an object can be represented as json in different ways, for example when different length of indentation is used.
To match the body as json, you need to add the python data structure (which could be dict, list or anything which can be the result of json.loads() call). The request’s body will be loaded as json and the result will be compared to the provided object. If the request’s body cannot be loaded as json, the matcher will fail and pytest-httpserver will proceed with the next registered matcher.
Example:
import requests
from pytest_httpserver import HTTPServer
def test_json_matcher(httpserver: HTTPServer):
httpserver.expect_request("/foo", json={"foo": "bar"}).respond_with_data("Hello world!")
resp = requests.get(httpserver.url_for("/foo"), json={"foo": "bar"})
assert resp.status_code == 200
assert resp.text == "Hello world!"
Note
JSON requests usually come with Content-Type: application/json
header.
pytest-httpserver provides the headers parameter to match the headers of
the request, however matching json body does not imply matching the
Content-Type header. If matching the header is intended, specify the expected
Content-Type header and its value to the headers parameter.
Note
json and data parameters are mutually exclusive so both of then cannot be specified as in such case the behavior is ambiguous.
Note
The request body is decoded by using the data_encoding parameter, which is default to utf-8. If the request comes in a different encoding, and the decoding fails, the request won’t match with the expected json.
Advanced header matching¶
For each http header, you can specify a callable object (eg. a python function) which will be called with the header name, header actual value and the expected value, and will be able to determine the matching.
You need to implement such a function and then use it:
from typing import Optional
import requests
from pytest_httpserver import HTTPServer
def case_insensitive_matcher(header_name: str, actual: Optional[str], expected: str) -> bool:
if actual is None:
return False
if header_name == "X-Foo":
return actual.lower() == expected.lower()
else:
return actual == expected
def test_case_insensitive_matching(httpserver: HTTPServer):
httpserver.expect_request(
"/", header_value_matcher=case_insensitive_matcher, headers={"X-Foo": "bar"}
).respond_with_data("OK")
assert requests.get(httpserver.url_for("/"), headers={"X-Foo": "bar"}).status_code == 200
assert requests.get(httpserver.url_for("/"), headers={"X-Foo": "BAR"}).status_code == 200
Note
Header value matcher is the basis of the Authorization
header parsing.
If you want to change the matching of only one header, you may want to use the
HeaderValueMatcher
class.
In case you want to do it globally, you can add the header name and the callable
to the HeaderValueMatcher.DEFAULT_MATCHERS
dict.
from pytest_httpserver import HeaderValueMatcher
def case_insensitive_compare(actual: str, expected: str) -> bool:
return actual.lower() == expected.lower()
HeaderValueMatcher.DEFAULT_MATCHERS["X-Foo"] = case_insensitive_compare
def test_case_insensitive_matching(httpserver: HTTPServer):
httpserver.expect_request("/", headers={"X-Foo": "bar"}).respond_with_data("OK")
assert (
requests.get(httpserver.url_for("/"), headers={"X-Foo": "bar"}).status_code
== 200
)
assert (
requests.get(httpserver.url_for("/"), headers={"X-Foo": "BAR"}).status_code
== 200
)
In case you don’t want to change the defaults, you can provide the
HeaderValueMatcher
object itself.
from typing import Optional
import requests
from pytest_httpserver import HeaderValueMatcher
from pytest_httpserver import HTTPServer
def case_insensitive_compare(actual: Optional[str], expected: str) -> bool:
# actual is `None` if it is not specified
if actual is None:
return False
return actual.lower() == expected.lower()
def test_own_matcher_object(httpserver: HTTPServer):
matcher = HeaderValueMatcher({"X-Bar": case_insensitive_compare})
httpserver.expect_request("/", headers={"X-Bar": "bar"}, header_value_matcher=matcher).respond_with_data("OK")
assert requests.get(httpserver.url_for("/"), headers={"X-Bar": "bar"}).status_code == 200
assert requests.get(httpserver.url_for("/"), headers={"X-Bar": "BAR"}).status_code == 200
Using custom request handler¶
In the case the response is not static, for example it depends on the request,
you can pass a function to the respond_with_handler
function. This function
will be called with a request object and it should return a Response object.
from random import randint
from werkzeug.wrappers import Request
from werkzeug.wrappers import Response
from pytest_httpserver import HTTPServer
def test_expected_request_handler(httpserver: HTTPServer):
def handler(request: Request):
return Response(str(randint(1, 10)))
httpserver.expect_request("/foobar").respond_with_handler(handler)
The above code implements a handler which returns a random number between 1 and 10. Not particularly useful but shows that the handler can return any computed or derived value.
In the response handler you can also use the assert
statement, similar to
the tests, but there’s a big difference. As the server is running in its own
thread, this will cause a HTTP 500 error returned, and the exception registered
into a list. To get that error, you need to call check_assertions()
method
of the httpserver.
In case you want to ensure that there was no other exception raised which was
unhandled, you can call the check_handler_errors()
method of the httpserver.
Two notable examples for this:
import pytest
import requests
from pytest_httpserver import HTTPServer
def test_check_assertions_raises_handler_assertions(httpserver: HTTPServer):
def handler(_):
assert 1 == 2
httpserver.expect_request("/foobar").respond_with_handler(handler)
requests.get(httpserver.url_for("/foobar"))
# if you leave this "with" statement out, check_assertions() will break
# the test by re-raising the assertion error caused by the handler
# pytest will pick this exception as it was happened in the main thread
with pytest.raises(AssertionError):
httpserver.check_assertions()
httpserver.check_handler_errors()
def test_check_handler_errors_raises_handler_error(httpserver: HTTPServer):
def handler(_):
raise ValueError("should be propagated")
httpserver.expect_request("/foobar").respond_with_handler(handler)
requests.get(httpserver.url_for("/foobar"))
httpserver.check_assertions()
# if you leave this "with" statement out, check_handler_errors() will
# break the test with the original exception
with pytest.raises(ValueError):
httpserver.check_handler_errors()
If you want to call both methods (check_handler_errors()
and
check_assertions()
) you can call the check()
method, which will call
these.
import pytest
import requests
from pytest_httpserver import HTTPServer
@pytest.mark.xfail
def test_check_assertions(httpserver: HTTPServer):
def handler(_):
assert 1 == 2
httpserver.expect_request("/foobar").respond_with_handler(handler)
requests.get(httpserver.url_for("/foobar"))
# this will raise AssertionError:
httpserver.check()
Note
The scope of the errors checked by the check()
method may
change in the future - it is added to check all possible errors happened in
the server.
Customizing host and port¶
By default, the server run by pytest-httpserver will listen on localhost on a random available port. In most cases it works well as you want to test your app in the local environment.
If you need to change this behavior, there are a plenty of options. It is very important to make these changes before starting the server, eg. before running any test using the httpserver fixture.
Use IP address 0.0.0.0 to listen globally.
Warning
You should be careful when listening on a non-local ip (such as 0.0.0.0). In this case anyone knowing your machine’s IP address and the port can connect to the server.
Environment variables¶
Set PYTEST_HTTPSERVER_HOST
and/or PYTEST_HTTPSERVER_PORT
environment
variables to the desired values.
Class attributes¶
Changing HTTPServer.DEFAULT_LISTEN_HOST
and
HTTPServer.DEFAULT_LISTEN_PORT
attributes. Make sure that you do this before
running any test requiring the httpserver
fixture. One ideal place for this
is putting it into conftest.py
.
Fixture¶
Overriding the httpserver_listen_address
fixture. Similar to the solutions
above, this needs to be done before starting the server (eg. before referencing
the httpserver
fixture).
import pytest
@pytest.fixture(scope="session")
def httpserver_listen_address():
return ("127.0.0.1", 8000)
Multi-threading support¶
When your client runs in a thread, everything completes without waiting for the first response. To overcome this problem, you can wait until all the handlers have been served or there’s some error happened.
This is available only for oneshot and ordered handlers, as permanent handlers last forever.
To have this feature enabled, use the context object returned by the wait()
method of the httpserver
object.
This method accepts the following parameters:
- raise_assertions: whether raise assertions on unexpected request or timeout or not
- stop_on_nohandler: whether stop on unexpected request or not
- timeout: time (in seconds) until time is out
Behind the scenes it synchronizes the state of the server with the main thread.
Last, you need to assert on the result
attribute of the context object.
import requests
from pytest_httpserver import HTTPServer
def test_wait_success(httpserver: HTTPServer):
waiting_timeout = 0.1
with httpserver.wait(stop_on_nohandler=False, timeout=waiting_timeout) as waiting:
requests.get(httpserver.url_for("/foobar"))
httpserver.expect_oneshot_request("/foobar").respond_with_data("OK foobar")
requests.get(httpserver.url_for("/foobar"))
assert waiting.result
httpserver.expect_oneshot_request("/foobar").respond_with_data("OK foobar")
httpserver.expect_oneshot_request("/foobaz").respond_with_data("OK foobaz")
with httpserver.wait(timeout=waiting_timeout) as waiting:
requests.get(httpserver.url_for("/foobar"))
requests.get(httpserver.url_for("/foobaz"))
assert waiting.result
In the above code, all the request.get() calls could be in a different thread, eg. running in parallel, but the exit condition of the context object is to wait for the specified conditions.
Emulating connection refused error¶
If by any chance, you want to emulate network errors such as Connection reset by peer or Connection refused, you can simply do it by connecting to a random port number where no service is listening:
import pytest
import requests
def test_connection_refused():
# assumes that there's no server listening at localhost:1234
with pytest.raises(requests.exceptions.ConnectionError):
requests.get("http://localhost:1234")
However, connecting to the port where the httpserver had been started will still succeed as the server is running continuously. This is working by design as starting/stopping the server is costly.
import pytest
import requests
# setting a fixed port for httpserver
@pytest.fixture(scope="session")
def httpserver_listen_address():
return ("127.0.0.1", 8000)
# this test will pass
def test_normal_connection(httpserver):
httpserver.expect_request("/foo").respond_with_data("foo")
assert requests.get("http://localhost:8000/foo").text == "foo"
# this tess will FAIL, as httpserver started in test_normal_connection is
# still running
def test_connection_refused():
with pytest.raises(requests.exceptions.ConnectionError):
# this won't get Connection refused error as the server is still
# running.
# it will get HTTP status 500 as the handlers registered in
# test_normal_connection have been removed
requests.get("http://localhost:8000/foo")
To solve the issue, the httpserver can be stopped explicitly. It will start
implicitly when the first test starts to use it. So the
test_connection_refused
test can be re-written to this:
def test_connection_refused(httpserver):
httpserver.stop() # stop the server explicitly
with pytest.raises(requests.exceptions.ConnectionError):
requests.get("http://localhost:8000/foo")
Emulating timeout¶
To emulate timeout, there’s one way to register a handler function which will sleep for a given amount of time.
import time
from pytest_httpserver import HTTPServer
import pytest
import requests
def sleeping(request):
time.sleep(2) # this should be greater than the client's timeout parameter
def test_timeout(httpserver: HTTPServer):
httpserver.expect_request("/baz").respond_with_handler(sleeping)
with pytest.raises(requests.exceptions.ReadTimeout):
assert requests.get(httpserver.url_for("/baz"), timeout=1)
There’s one drawback though: the test takes 2 seconds to run as it waits the handler thread to be completed.
Running an HTTPS server¶
To run an https server, trustme can be used to do the heavy lifting:
@pytest.fixture(scope="session")
def ca():
return trustme.CA()
@pytest.fixture(scope="session")
def httpserver_ssl_context(ca):
context = ssl.SSLContext(ssl.PROTOCOL_TLS_SERVER)
localhost_cert = ca.issue_cert("localhost")
localhost_cert.configure_cert(context)
return context
@pytest.fixture(scope="session")
def httpclient_ssl_context(ca):
with ca.cert_pem.tempfile() as ca_temp_path:
return ssl.create_default_context(cafile=ca_temp_path)
@pytest.mark.asyncio
async def test_aiohttp(httpserver, httpclient_ssl_context):
import aiohttp
httpserver.expect_request("/").respond_with_data("hello world!")
connector = aiohttp.TCPConnector(ssl=httpclient_ssl_context)
async with aiohttp.ClientSession(connector=connector) as session:
async with session.get(httpserver.url_for("/")) as result:
assert (await result.text()) == "hello world!"
def test_requests(httpserver, ca):
import requests
httpserver.expect_request("/").respond_with_data("hello world!")
with ca.cert_pem.tempfile() as ca_temp_path:
result = requests.get(httpserver.url_for("/"), verify=ca_temp_path)
assert result.text == "hello world!"
def test_httpx(httpserver, httpclient_ssl_context):
import httpx
httpserver.expect_request("/").respond_with_data("hello world!")
result = httpx.get(httpserver.url_for("/"), verify=httpclient_ssl_context)
assert result.text == "hello world!"
Using httpserver on a dual-stack (IPv4 and IPv6) system¶
pytest-httpserver can only listen on one address and it also means that address family is determined by that. As it relies on Werkzeug, it passes the provided host parameter to it and then it is up to Werkzeug how the port binding is done.
Werkzeug determines the address family by examining the string provided. If it
contains a colon (:
) then it will be an IPv6 (AF_INET6
) socket, otherwise, it
will be an IPv4 (AF_INET
) socket. The default string in pytest-httpserver is
localhost
so by default, the httpserver listens on IPv4. If you want it to
listen on IPv6 address, provide an IPv6 address (::1
for example) to it.
It should be noted that dual-stack systems are still working with
pytest-httpserver because the clients obtain the possible addresses for the a
given name by calling getaddrinfo()
or similar function which returns the
addresses together with address families, and the client iterates over this
list. In the case when pytest-httpserver is listening on 127.0.0.1
, and
the client uses localhost
name in the url, it will try ::1
first, and
then it will move on to 127.0.0.1
, which will succeed, or vica-versa, where
127.0.0.1
will be successful first.
If you want to test a connection error case in your test (such as TLS error),
the client can fail in a strange way as we seen in this issue. In such case,
client tries with 127.0.0.1
first, then reaches a TLS error (which is normal
as the test case is about testing for the TLS issue), then it moves on to
::1
, then it fails with Connection reset
. In such case fixing the bind
address to 127.0.0.1
(and thereby fixing the host part of the URL returned
by the url_for call) solves the issue as the client will receive the address
(127.0.0.1
) instead of the name (localhost
) so it won’t move on to the
IPv6 address.
Running httpserver in blocking mode¶
In this mode, the code which is being tested (the client) is executed in a background thread, while the server events are synchronized to the main thread, so it looks like it is running in the main thread. This allows to catch the assertions occured on the server side synchronously, and assertions are raised to the main thread. You need to call check_assertions at the end for only the unexpected requests.
This is an experimental feature so pytest-httpserver has no fixture for it yet. If you find this feature useful any you have ideas or suggestions related to this, feel free to open an issue.
Example:
import threading
from queue import Queue
import pytest
import requests
from pytest_httpserver import BlockingHTTPServer
# override httpserver fixture
@pytest.fixture
def httpserver():
server = BlockingHTTPServer(timeout=1)
server.start()
yield server
server.clear()
if server.is_running():
server.stop()
# this is to check if the client has made any request where no
# `assert_request` was called on it from the test
server.check_assertions()
server.clear()
def test_simplified(httpserver: BlockingHTTPServer):
def client(response_queue: Queue):
response = requests.get(httpserver.url_for("/foobar"), timeout=10)
response_queue.put(response)
# start the client, server is not yet configured
# it will block until we add a request handler to the server
# (see the timeout parameter of the http server)
response_queue: Queue[requests.models.Response] = Queue(maxsize=1)
thread = threading.Thread(target=client, args=(response_queue,))
thread.start()
try:
# check that the request is for /foobar and it is a GET method
# if this does not match, it will raise AssertionError and test will fail
client_connection = httpserver.assert_request(uri="/foobar", method="GET")
# with the received client_connection, we now need to send back the response
# this makes the request.get() call in client() to return
client_connection.respond_with_json({"foo": "bar"})
finally:
# wait for the client thread to complete
thread.join(timeout=1)
assert not thread.is_alive() # check if join() has not timed out
# check the response the client received
response = response_queue.get(timeout=1)
assert response.status_code == 200
assert response.json() == {"foo": "bar"}