大家好~我是米洛

我正在从0到1打造一个开源的接口测试平台, 也在编写一套与之对应的完整教程,希望大家多多支持。

欢迎关注我的公众号米洛的测开日记,获取最新文章教程!

回顾

上一节我们写好了一个测试计划的初级版本,并且能够添加计划,也能定时执行。那今天我们就来完善测试计划删改查与APScheduler相结合的部分。

定时任务和测试计划的耦合

目前来看,测试计划和定时任务有着高度的耦合,考虑到我们并不打算做一款自由度高的定时job系统,所以暂时就先按这个思路设计。毕竟我们是一个做接口自动化的平台,而不是类似jenkins一样,有各类任务的平台。如果后续有相关的需求,我们再考虑。

所以现在的情况是,定时任务列表存放了一批job数据,测试计划表又存放了与之对应的测试计划数据。

查询定时任务

我们之前不是添加过一个定时任务吗?我们先来看看用APScheduler的api,获取到的数据内容。这时候就需要debug了~

Scheduler对象是有个get_job的方法,我们之前给job的id定义为测试计划的id。

所以我们现在可以用测试计划的id去查询定时任务的一些信息。

在utils/scheduler.py新增list_test_plan方法:

  1. @staticmethod
  2. def list_test_plan(data: List):
  3. for d in data:
  4. job = Scheduler.scheduler.get_job(str(d.get('id')))
  5. print(job)
  6. d["next_run"] = job.next_run_time

传入的data是查询测试计划接口拿到的测试计划数据

测试平台系列(75) 完善测试计划的删改查功能 - 图1

过程如下:

  1. 获取到测试计划数据
  2. 根据测试计划数据去获取到定时任务的数据
  3. 一并返回

测试平台系列(75) 完善测试计划的删改查功能 - 图2

其实我这里弱化了定时任务的特点,反而把cron这样的字段剥离到测试计划之中了。所以这个定时任务给我们的作用就只有2个:

  • 它是不是存在
  • 它是否正常工作,可以通过next_run_at来判断
    所以我们只需要在查询接口,加上next_run_at就可以了。

测试平台系列(75) 完善测试计划的删改查功能 - 图3

这点很重要,因为我们可能创建测试计划成功了,但创建定时任务失败了。所以这个数据不可忽略。

当我们没获取到job的时候,把job的state改为2(并非在数据库改),只是在页面上显示这个测试计划任务未添加成功!!!

测试平台系列(75) 完善测试计划的删改查功能 - 图4

编写编辑/删除定时任务方法

由于APS本身就提供了良好的api,所以我们简易封装一下即可。

  1. @staticmethod
  2. def edit_test_plan(plan_id, plan_name, cron):
  3. """
  4. 通过测试计划id,更新测试计划任务的cron,name等数据
  5. :param plan_id:
  6. :param plan_name:
  7. :param cron:
  8. :return:
  9. """
  10. Scheduler.scheduler.modify_job(job_id=str(plan_id), trigger=CronTrigger.from_crontab(cron), name=plan_name)
  11. @staticmethod
  12. def pause_resume_test_plan(plan_id, status):
  13. """
  14. 暂停或恢复测试计划,会影响到next_run_at
  15. :param plan_id:
  16. :param status:
  17. :return:
  18. """
  19. if status:
  20. Scheduler.scheduler.resume_job(job_id=str(plan_id))
  21. else:
  22. Scheduler.scheduler.pause_job(job_id=str(plan_id))
  23. @staticmethod
  24. def remove(plan_id):
  25. """
  26. 删除job,当删除测试计划时,调用此方法
  27. :param plan_id:
  28. :return:
  29. """
  30. Scheduler.scheduler.remove_job(str(plan_id))

这边已经都写好了注释,至于为什么我们需要pause和resume方法呢,是因为有时候我们可能不太想跑这个任务,我们想先暂停一下,所以有这个情况发生。

修改对应的接口

测试平台系列(75) 完善测试计划的删改查功能 - 图5

我们来测试下暂停功能:

  • 暂停之前获取测试计划列表

测试平台系列(75) 完善测试计划的删改查功能 - 图6

  • 暂停之后

测试平台系列(75) 完善测试计划的删改查功能 - 图7

可以看到,由于next_run_at为None了,所以我们之前的strftime失效了。

所以我们要做下调整:

测试平台系列(75) 完善测试计划的删改查功能 - 图8

所以我们testplan有了3个状态:

  • 0: 等待中
  • 1: 执行中
  • 2: 定时任务未添加
  • 3: 定时任务已暂停
    有了这些数据,我们前端就知道要怎么展示测试计划的定时任务状态了。
    下一节我们就要开始编写测试计划页面的前端部分了。