如何给tornado做一个mock类?

我所在的环境是,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 的单测要做特殊处理?
阅读 2.7k
撰写回答
你尚未登录,登录后可以
  • 和开发者交流问题的细节
  • 关注并接收问题和回答的更新提醒
  • 参与内容的编辑和改进,让解决方法与时俱进
推荐问题