异步等待功能的单元测试

时间:2018-10-11 02:41:57

标签: python-3.x unit-testing pymongo tornado-motor

我应该如何编写用于单元测试的异步等待(示例函数)? 这部分代码包括使用motor-tornado从mongodb数据库中删除集合!

import motor
import traceback
import pymongo

from contextlib import contextmanager
from tornado.ioloop import IOLoop 

async def drop(uri,tables):
        """Drop the collections of the database
        """
        with _connect_db(uri) as con:
            db = con['mongo_master']
            try:
                for tbl in tables:
                    cursor = await db.drop_collection(tbl)
                    if cursor['ok'] == 1.0:
                        res = "Collection %s is dropped!" %tbl
                        return res
                    else:
                        res = "Collection %s is not found!" %tbl
                        return res
            except:
                print(traceback.format_exc())

我已经尝试过 asynio模块 ,但是还有其他方法可以简单地使用 unittest 吗?

0 个答案:

没有答案