python响应库将URL的一部分添加到请求参数中

时间:2020-06-18 14:18:39

标签: python api python-requests mocking python-responses

我正在尝试使用响应库模拟外部API。 我想检查自己是否在请求中正确传递了参数,因此我使用的是responses docs中的最低要求示例:

import responses
import requests

@responses.activate
def test_request_params():
    responses.add(
        method=responses.GET,
        url="http://example.com?hello=world",
        body="test",
        match_querystring=False,
    )

    resp = requests.get('http://example.com', params={"hello": "world"})
    assert responses.calls[0].request.params == {"hello": "world"}

问题是,一旦我将http://example.com替换为类似于API端点的URL,此操作就会中断:

@responses.activate
def test_request_params():
    responses.add(
        method=responses.GET,
        url="http://example.com/api/endpoint?hello=world",
        body="test",
        match_querystring=False,
    )

    resp = requests.get('http://example.com/api/endpoint', params={"hello": "world"})
    assert responses.calls[0].request.params == {"hello": "world"}

现在,响应已将部分URL添加到第一个查询参数:

>       assert responses.calls[0].request.params == {"hello": "world"}
E       AssertionError: assert {'/api/endpoint?hello': 'world'} == {'hello': 'world'}

我错过了什么吗?

1 个答案:

答案 0 :(得分:0)

您可以将正则表达式作为url参数传递,而忽略参数:

@responses.activate
def test_request_params():

    url = r"http://example.com/api/endpoint"
    params = {"hello": "world", "a": "b"}

    # regex that matches the url and ignores anything that comes after
    rx = re.compile(rf"{url}*")
    responses.add(method=responses.GET, url=rx)

    response = requests.get(url, params=params)
    assert responses.calls[-1].request.params == params

    url_path, *queries = responses.calls[-1].request.url.split("?")
    assert url_path == url