测试平台系列(96) 如何停止测试任务执行.md

2022/6/13 测试平台接口测试FastApiPythonReact

大家好~我是米洛
我正在从0到1打造一个开源的接口测试平台, 也在编写一套与之对应的教程,希望大家多多支持。
欢迎关注我的公众号米洛的测开日记,获取最新文章教程!

# 回顾

上一节我们的前置条件支持了python脚本类型,让我们能够更灵活处理数据了。

今天我们就玩点好玩的,和大家一起探讨:怎么停止一段python代码。之后我们将会运用到pity之中。

# 知识要点

本文需要大家对asyncio相关知识有一些了解,至于原理方面,大家可以自行查阅,因为我也没细看。

# 为什么要做这个?

针对测试任务执行非常久的时候,看起来会是阻塞的情况,举个例子:

某个python脚本里面写了time.sleep(1000),导致你的case一直好像没有执行完成,那我们想结束它,该咋整呢?

今天博主要聊的就是这方面的内容。

# 先看看"难题"

我们来写一个简单的demo。

  • 我们先定义一个异步无限循环的方法,让它每隔一秒就打印一行内容
import asyncio


async def run():
    while True:
        print("still alive")
        await asyncio.sleep(1)
1
2
3
4
5
6
7

试想一下,如果这个方法开启了,不结束它,它是不是会一直打印下去?没错,我们先试试直接运行之:

很显然,它是绝对不会停止的,除非你关闭这个py程序。

我们把它想象成同步方法,是不是也会遇到这个困难:

import time


def run():
    time.sleep(10000)
    
if __name__ == "__main__":
    run()
    print("done")
1
2
3
4
5
6
7
8
9

可以发现run一旦开始,它就阻塞了整个程序,下面的done必须要等run结束了才能打印出来。

# 怎么停止run

我们都知道,在python3.4以后新增了异步编程相关的概念,最初是由@coroutine这样的装饰器放到方法上,把方法标注为异步方法,后面直接从语言层面支持了异步方法定义(async),那其实里面还有很多我们不太常用的部分,比如今天要说的create_task。

  • 我们改写下方法
import asyncio


async def run():
    while True:
        print("still alive")
        await asyncio.sleep(1)
        
        
async def main():
    await asyncio.create_task(run())


if __name__ == "__main__":
    asyncio.run(main())
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15

这次我们包了一层方法,利用asyncio.create_task来创建异步任务,create_task接收一个coroutine并执行。

执行之后,可以发现这个和上面的情况,任务也会一直进行下去。

我们继续下一步改造:

我们去掉await,可以看到run方法确实执行了,但是still alive只打印了一次就结束了。

我对它的理解是,虽然create_task创建了一个异步任务,但没说要await,也就是说没有说要等它结束。

想象一下这些异步任务都由一个事件循环控制,当你执行main(main本身也是一个异步任务)的时候,asyncio.run默认是要执行到main方法执行完毕的,也就是说,现在事件循环等待main方法执行完毕,main方法里面又创建了一个异步任务,但没有强调需要该任务完成,创建完毕后,由于main任务已经完成了,就导致整个事件结束了。

梳理一下:

  • 有await

    线程开启 -> 执行main -> main里面创建异步任务run -> 等待异步任务run(一直等一直等)

    由于主线程没有结束,所以整个python程序一直在等待异步任务执行完毕,毕竟它是死循环,所以会一直等下去。

  • 无await

    线程开启 -> 执行main -> main里面创建异步任务run -> 不等待异步任务run -> main方法结束 -> 线程结束 -> 程序退出

    以上都是个人结合go的goroutine给出的理解。肯定会有一些差别的地方。

  • 再次改造

    其实create_task会返回一个task对象,里面有done和cancel方法,也就是说咱们可以取消他也可以完成他。

import asyncio


async def run():
    while True:
        print("still alive")
        await asyncio.sleep(1)


async def main():
    task = asyncio.create_task(run())
    await asyncio.sleep(2)
    # 2秒后,就停掉这个任务
    task.cancel()
    print("task任务结束了,main也即将完成,整个程序即将退出")


if __name__ == "__main__":
    asyncio.run(main())
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19

一旦调用了create_task,那么任务就已经开始了,接着我们用await让main等待2秒,再调用task.cancel方法就可以取消这个task,这样所有事件都结束,程序也会退出了。

看看gif:

这时候有的同学可能会问了,await2秒以后,不管你是否调用cancel,因为没有await task,所以程序照样会退出啊,没法证明task真的被cancel了

仔细想想,确实说的有道理。那我们再来改造下:

可以看到,这个task死之前还做了垂死挣扎!!!

  • 加上await

可以看到加上await之后,它还是不会停止

那我们加上cancel试试:

很遗憾,报错了~~~不过没关系,我们继续改下。只需要加上异常处理,就可以完美实现2秒后自动取消任务了。

但这个例子依然不是很帅,如果只是想控制异步任务的执行时间,那我们可以用wait_for:

import asyncio


async def run():
    while True:
        print("still alive")
        await asyncio.sleep(1)


async def main():
    try:
        await asyncio.wait_for(run(), 2)
    except asyncio.TimeoutError:
        print("run方法超过2秒仍未执行完成")


if __name__ == "__main__":
    asyncio.run(main())
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18

这样2秒后,任务还没结束也不会继续执行了。

# 去掉死循环以后,任务会在2秒内完成(因为只等待1秒),这时候就可以看到timeout异常不会被触发

emmmm, 今天的内容就介绍到这了,异步还有挺多玩法的,我也不是很清楚,大家可以互相交流交流~~~