3.4 KiB
Request Matching
Whilst it is preferable to provide the whole URI to :pyrequests_mock.Adapter.register_uri
it is possible to
just specify components.
Basic
You can specify a protocol-less path:
>>> adapter.register_uri('GET', '//test.com/5', text='resp')
>>> session.get('mock://test.com/5').text
'resp'
or you can specify just a path:
>>> adapter.register_uri('GET', '/6', text='resp')
>>> session.get('mock://test.com/6').text
'resp'
>>> session.get('mock://another.com/6').text
'resp'
Query Strings
Query strings provided to a register will match so long as at least those provided form part of the request.
>>> adapter.register_uri('GET', '/7?a=1', text='resp')
>>> session.get('mock://test.com/7?a=1&b=2').text
'resp'
>>> session.get('mock://test.com/7?a=3')
Traceback (most recent call last):
...//test.com/7?a=3 requests_mock.exceptions.NoMockAddress: No mock address: GET mock:
This can be a problem in certain situations, so if you wish to match only the complete query string there is a flag `complete_qs`:
>>> adapter.register_uri('GET', '/8?a=1', complete_qs=True, text='resp')
>>> session.get('mock://test.com/8?a=1&b=2')
Traceback (most recent call last):
...//test.com/8?a=1&b=2 requests_mock.exceptions.NoMockAddress: No mock address: GET mock:
Matching ANY
There is a special symbol at requests_mock.ANY which acts as the wildcard to match anything. It can be used as a replace for the method and/or the URL.
>>> adapter.register_uri(requests_mock.ANY, 'mock://test.com/8', text='resp')
>>> session.get('mock://test.com/8').text
'resp'
>>> session.post('mock://test.com/8').text
'resp'
>>> adapter.register_uri(requests_mock.ANY, requests_mock.ANY, text='resp')
>>> session.get('mock://whatever/you/like')
'resp'
>>> session.post('mock://whatever/you/like')
'resp'
Regular Expressions
URLs can be specified with a regular expression using the python
:pyre
module. To use
this you should pass an object created by :pyre.compile
.
The URL is then matched using :pyre.regex.search
which means that it will match any
component of the url, so if you want to match the start of a URL you
will have to anchor it.
>>> import re
>>> matcher = re.compile('tester.com/a')
>>> adapter.register_uri('GET', matcher, text='resp')
>>> session.get('mock://www.tester.com/a/b').text
'resp'
If you use regular expression matching then requests-mock can't do it's normal query string or path only matching, that will need to be part of the expression.
Request Headers
A dictionary of headers can be supplied such that the request will only match if the available headers also match. Only the headers that are provided need match, any additional headers will be ignored.
>>> adapter.register_uri('POST', 'mock://test.com', request_headers={'key', 'val'}, text='resp')
>>> session.post('mock://test.com', headers={'key': 'val', 'another': 'header'}).text
'resp'
>>> session.post('mock://test.com')
Traceback (most recent call last):
...//test.com/ requests_mock.exceptions.NoMockAddress: No mock address: POST mock: