如何给tornado做一个mock类?

发布于 2022-09-02 14:07:39 字数 1905 浏览 26 评论 0

我所在的环境是,tornado4.3,python2.7
目前在给自己的服务写单元测试,
我想要让我的单元测试完全独立于第三方环境(比如一个三方的url接口)。
首先,我尝试封装了这样两个函数:
@tornado.gen.coroutine
def put_get_request_into_ioloop(context, url, origin=False):

"""
把一个异步get请求扔到tornado ioloop中
:param url: 请求url
:param handler_id: 处理id
:param origin: 是否日志请求和响应原串
:return:
True, 返回结果
False, None
"""
if origin:
    logger.info("Request:%s\t%s" % (context["handler_id"], url))
try:
    client = tornado.httpclient.AsyncHTTPClient()
    response = yield client.fetch(url)
    if origin:
        logger.info("Response:%s\t%s" % (context["handler_id"], response.body))
except:
    logger.error("ErrorRequest:%s\t%s" % (context["handler_id"], url), exc_info=True)
    raise tornado.gen.Return((False, None))
else:
    raise tornado.gen.Return((True, response.body))

@tornado.gen.coroutine
def put_post_request_into_ioloop(context, url, body, origin=False):

"""
把一个异步post请求扔到tornado ioloop中
:param url: 请求url
:param body: 请求体
:param handler_id: 处理id
:param origin: 是否日志请求和响应原串
:return:
True, 返回结果
False, None
"""
if origin:
    logger.info("Request:%s\t%s\t%s" % (context["handler_id"], url, body))
try:
    client = tornado.httpclient.AsyncHTTPClient()
    response = yield client.fetch(url, method="POST", body=body)
    if origin:
        logger.info("Response:%s\t%s" % (context["handler_id"], response.body))
except:
    logger.error("ErrorRequest:%s\t%s\t%s" % (context["handler_id"], url, body), exc_info=True)
    raise tornado.gen.Return((False, None))
else:
    raise tornado.gen.Return((True, response.body))

在我的系统中,我在调用第三方接口的时候,会调用这两个函数。

我想要用mock的方法,替换这种函数的调用返回结果,使我的系统独立,便于单元测试。但是有yield和coroutine的“干扰”,我不知道应该如何实现,替换这个结果,并能运行我的单元测试。
我尝试过写一个函数返回自己raise tornaod.gen.Return,然后用MethodType的方式,对其进行替换。但是没能跑起来。
我想知道:
1.我的这种思路是否对?有更好的解决方法?
2.是不是因为ioloop的启动方式有误,tornado的单测要做特殊处理?

如果你对这篇内容有疑问,欢迎到本站社区发帖提问 参与讨论,获取更多帮助,或者扫码二维码加入 Web 技术交流群。

扫码二维码加入Web技术交流群

发布评论

需要 登录 才能够评论, 你可以免费 注册 一个本站的账号。
列表为空,暂无数据
我们使用 Cookies 和其他技术来定制您的体验包括您的登录状态等。通过阅读我们的 隐私政策 了解更多相关信息。 单击 接受 或继续使用网站,即表示您同意使用 Cookies 和您的相关数据。
原文