Python-如何测试异步数据库代码

问题产生

用pytest写好pymysql的测试代码之后,完成了pymysql代码,测试通过。然后决定将数据库改成异步,即pymysql->aiomyqsql,然而返回结果不再是元组,而是coroutine对象,这时产生困惑,如何测试异步数据库?

解决思路

看到了静觅的a异步协程使用方法,获得了启发,可以在测试代码里,使用asyncio,并通过task的result()方法,来assert结果是否正确。

# 摘取部分代码,以展示思路
import asyncio

loop = asyncio.get_event_loop()

def test_get_sentence_by_text():
    tasks = [asyncio.ensure_future(get_sentence_by_text('君')),
             asyncio.ensure_future(get_sentence_by_text('君子'))]
    loop.run_until_complete(asyncio.wait(tasks))         
    
    assert tasks[0].result() == (6, 4, '关关雎鸠,在河之洲。窈窕淑女,君子好逑。')
    assert tasks[1].result() == (6, 4, '关关雎鸠,在河之洲。窈窕淑女,君子好逑。')
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值