概要

这里我们测试一个接口有两个文件。一个是接口定义、一个测试用例集。

接口定义示例

每个接口定义一个类，url和请求参数为类方法，方便复用！

  1. # -*- coding: utf-8 -*-#
  2. # file:         coll_work.py
  3. # Author:       zaygee
  4. # Date:         2021/5/13
  6. from config import env
  9. class CollWork:
  10.     """设置该作品为协作作品"""
  12.     @classmethod
  13.     def url(cls, work_Id):
  14.         return env.SOCKET_COLL_HOST + '/coll/{}'.format(work_Id)
  16.     @classmethod
  17.     def data(cls):
  18.         pass

普通接口测试用例示例

每个接口测试文件定义setup_class，这里我们用作请求登陆接口并写入cookie.json文件

tips：简单说下这么处理的原因吧，原本鉴于requests库的session对象，是可以让我们跨请求保持某些参数，也会在同一个session实例发出的所有请求之间保持cookies，但是为了实现权限类接口的场景，我们需要两个不同用户的cookie，必要时候在headers传入不用用户的cookie 😊嘿嘿嘿。

  1. @allure.feature("设置该作品为协作作品接口")
  2. class TestCollKitten:
  4.     @classmethod
  5.     def headers(cls):
  6.         return {'cookie': base.read_cookie()['cookie']}
  8.     @classmethod
  9.     def setup_class(cls):
  10.         base.write_cookie()
  12.     @allure.title("传入普通作品work_id")
  13.     def test_01(self, work):
  14.         data = Work.data()
  15.         work_id = work(data)    # 业务场景需要生成请求参数
  16.         res = base.post_req(CollWork.url(work_id), headers=self.headers())
  17.         assert res.status_code == 200
  18.         assert base.extract(res.text, '$.msg'), '操作成功'

socket接口测试用例示例

  1. @allure.feature("协作websocket接口")
  2. class TestSocketCollaborate:
  4.     @classmethod
  5.     def setup_class(cls):
  6.         “”“
  7.         根据业务场景需要，
  8.         调用登陆接口并将cookie、cookie的value写入文件
  9.         ”“”
  10.         base.write_cookie_value()
  11.         base.write_cookie()
  13.     @classmethod
  14.     def headers(cls):
  15.         “”“headers添加cookie”“”
  16.         return {'cookie': base.read_cookie()['cookie']}
  18.     @classmethod
  19.     def cookie_value(cls):
  20.         “”“读取cookie value用于请求参数”“”
  21.         return base.read_cookie_value()['value']
  23.     def setup(self):
  24.         “”“当前环境下的全局变量，用于当前用例集数据共享”“”
  25.         self.g = globals()
  27.     @allure.title("传入协作作品的workId")
  28.     def test_01(self, work, coll_work):
  29.         data = Work().data()
  30.         work_id = coll_work(work(data))  # 业务场景需要生成请求参数
  31.         self.g['work_id'] = work_id    # 设置全局共享数据
  32.         coll_url = Req.url(work_id, self.cookie_value())
  33.         wss = SocketCon.con(coll_url)
  34.         wss.send('42["join",{}]'.format(work_id))    # 发送socket数据
  35.         assert SocketCon.recv_log(wss, 4).split(',')[2].split(':')[1].strip("\""), True