python - 如何模拟使用 getattr 动态调用的请求方法

Question

我有一个使用 getattr 调用请求方法的类:

import requests

class CustomRequests(object):
     def __init__(self):
         pass

     def _do_requests(self, method='GET', url='', expected_status=200):
         make_request = getattr(requests, method.lower())
         url = url if url else 'http://example.com'

         try:
            response = make_request(method, url=url)
            except response.exceptions.RequestException as exception:
                raise exception
         if response.status_code != expected_status:
            raise ValueError

    def get(self, *args, **kwargs):
        self._do_requests(method='GET', *args, **kwargs)

我正在尝试像这样使用模拟和响应库来测试 api:

import responses

@responses.activate
def test_get_method(self):
    responses.add('GET', url='http://test_this_api.com', status=200)
    custom_request = CustomRequest()
    response_data = custom_request.get(method='GET')
    AssertIsNotNone(response_data)

是否有更好或正确的方法来测试此方法。收到此错误:

    message = message.format(**values)
KeyError: 'method'

Answer 1

无需使用 getattr。 requests.get、requests.post 等只是 requests.request 的便捷方法。 ，它允许您将 HTTP 方法作为参数传递:

requests.request('GET', url) # equivalent to requests.get(url)

还有:

• 您的try/except 毫无意义，因为您所做的只是重新引发异常。
• 当响应状态与您的预期不匹配时，引发 ValueError 是没有意义的。 ValueError 适用于 "when a built-in operation or function receives an argument that has the right type but an inappropriate value."创建您自己的异常类，例如意外的 HTTPStatusError。
• 由于您的 CustomRequests 类的全部要点似乎是在响应的状态代码与用户预期的不匹配时引发异常，因此您的测试应该断言异常实际上是用 assertRaises() 提出.