使用requests_mock做单元测试遇NoMockAddress异常求助

嘿，你遇到的这个问题其实很常见——你mock的地址和实际代码发送的请求地址完全不匹配。requests_mock默认是精确匹配完整URL的，包括后面的查询参数部分，而你现在只mock了基础的MOCK_ADDRESS，但实际请求的URL带上了query=username%3Atestsetset&page=1这些参数，自然就匹配失败啦。

下面给你几个实用的解决办法，你可以根据自己的测试需求选：

1. 精确匹配带参数的完整URL #

如果你的测试需要严格验证请求参数，那可以直接构造出实际请求的完整URL来mock：

def test_general_search(requests_mock):
    params = {'query': 'testsetset', 'exact': 0, 'max_pages': '1', 'category': 'username'}
    # 注意这里要和实际请求的URL完全一致，包括编码后的参数
    full_request_url = f"{MOCK_ADDRESS}?query=username%3Atestsetset&page=1"
    requests_mock.get(full_request_url, json=MOCK_RESPONSE)
    
    client = Client(MOCK_ADDRESS, auth=MOCK_CREDS, headers=MOCK_HEADERS)
    res = general_search(client, params)
    assert ...

不过这种方法有点死板，如果参数顺序变了或者多了其他参数，mock就会失效。

2. 用qs参数灵活匹配查询参数 #

requests_mock支持通过qs参数来匹配查询参数，不管参数在URL里的顺序，只要键值对对应上就行：

def test_general_search(requests_mock):
    params = {'query': 'testsetset', 'exact': 0, 'max_pages': '1', 'category': 'username'}
    # mock基础地址，同时指定要匹配的查询参数（注意实际请求里的query是"username:testsetset"）
    requests_mock.get(
        MOCK_ADDRESS,
        json=MOCK_RESPONSE,
        qs={'query': 'username:testsetset', 'page': '1'}
    )
    
    client = Client(MOCK_ADDRESS, auth=MOCK_CREDS, headers=MOCK_HEADERS)
    res = general_search(client, params)
    assert ...

这种方法更灵活，适合你需要验证参数但不关心参数顺序的场景。

3. 匹配路径前缀，忽略所有查询参数 #

如果你的测试不需要验证查询参数，只是想mock这个路径下的所有GET请求，可以用正则表达式或者any_path参数：

def test_general_search(requests_mock):
    params = {'query': 'testsetset', 'exact': 0, 'max_pages': '1', 'category': 'username'}
    # 用正则匹配以MOCK_ADDRESS开头的所有URL
    requests_mock.get(re.compile(f"^{MOCK_ADDRESS}"), json=MOCK_RESPONSE)
    # 或者用any_path=True，只要路径匹配就命中mock
    # requests_mock.get(MOCK_ADDRESS, json=MOCK_RESPONSE, any_path=True)
    
    client = Client(MOCK_ADDRESS, auth=MOCK_CREDS, headers=MOCK_HEADERS)
    res = general_search(client, params)
    assert ...

这种方法适合快速mock整个接口，不用纠结参数细节。

另外，调试的时候你可以打印出实际发送的请求URL，方便对比排查：

# 在assert之前打印实际请求的URL
print("实际请求URL:", requests_mock.last_request.url)

内容的提问来源于stack exchange，提问作者Lior Dahan