# 回顾
上一节,我们说到了关于整个用例执行的生命周期。我后面细想了一下,如果一开始就要完全兼容代码生成的话,写起来会很费时间。为了给大家一些好的体验,还是得一口一口地吃,不能一下吃胖。
所以我们就先完成无码模式吧~
# 粗略地设计几个核心表
我们的用例后续如果要方便管理,肯定是需要做一些分类和权限的控制,所以我们先设计几个核心的表:
项目表
用例分步在各个项目,类似于目录的概念,不同的用例位于不同的项目之中,方便用例进行分类。
项目角色表
每个项目都是需要有自己的成员,对于成员们来说,他们因为身份的不同,身上担负的职责也不一样,我们给出几个简单的身份如:
组长和组员,其中组长可以对组员进行相关权限的变化。
测试用例表
存放了各个用例的信息,是测试平台的一个最小的执行单元。
# 项目表 models/project.py
from app.models import db
from datetime import datetime
class Project(db.Model):
id = db.Column(db.INT, primary_key=True)
name = db.Column(db.String(16), unique=True, index=True)
owner = db.Column(db.INT)
created_at = db.Column(db.DATETIME, nullable=False)
updated_at = db.Column(db.DATETIME, nullable=False)
deleted_at = db.Column(db.DATETIME)
create_user = db.Column(db.INT, nullable=True)
update_user = db.Column(db.INT, nullable=True)
private = db.Column(db.BOOLEAN, default=False)
def __init__(self, name, owner, create_user, private=False):
self.name = name
self.owner = owner
self.private = private
self.created_at = datetime.now()
self.updated_at = datetime.now()
self.create_user = create_user
self.update_user = create_user
self.deleted_at = None
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
大概分为以下几个字段,除了创建/更新时间/用户以外,还有以下字段:
owner(组长) name(项目名) private(是否私有)
# 项目角色表 models/project_role.py
from app.models import db
from datetime import datetime
class ProjectRole(db.Model):
id = db.Column(db.INT, primary_key=True)
project_id = db.Column(db.INT, index=True)
project_role = db.Column(db.INT, index=True)
created_at = db.Column(db.DATETIME, nullable=False)
updated_at = db.Column(db.DATETIME, nullable=False)
deleted_at = db.Column(db.DATETIME)
create_user = db.Column(db.INT, nullable=True)
update_user = db.Column(db.INT, nullable=True)
def __init__(self, project_id, project_role, create_user):
self.project_id = project_id
self.project_role = project_role
self.created_at = datetime.now()
self.updated_at = datetime.now()
self.create_user = create_user
self.update_user = create_user
self.deleted_at = None
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
与上个表单类似,project_id是项目id,project_role是项目角色。
咱们规定:
1: 组长 2: 组员
# 测试用例表 models/test_case.py
from app.models import db
from datetime import datetime
class TestCase(db.Model):
id = db.Column(db.INT, primary_key=True)
name = db.Column(db.String(32), unique=True, index=True)
request_type = db.Column(db.INT, default=1, comment="请求类型 1: http 2: grpc 3: dubbo")
url = db.Column(db.TEXT, nullable=False, comment="请求url")
request_method = db.Column(db.String(12), nullable=True, comment="请求方式, 如果非http可为空")
request_header = db.Column(db.TEXT, comment="请求头,可为空")
project_id = db.Column(db.INT, comment="所属项目")
tag = db.Column(db.String(64), comment="用例标签")
status = db.Column(db.INT, comment="用例状态: 1: 待完成 2: 暂时关闭 3: 正常运作")
expected = db.Column(db.TEXT, comment="预期结果, 支持el表达式", nullable=False)
created_at = db.Column(db.DATETIME, nullable=False)
updated_at = db.Column(db.DATETIME, nullable=False)
deleted_at = db.Column(db.DATETIME)
create_user = db.Column(db.INT, nullable=False)
update_user = db.Column(db.INT, nullable=False)
def __init__(self, name, request_type, url, project_id, tag, status, expected, create_user, request_header=None,
request_method=None):
self.name = name
self.request_type = request_type
self.url = url
self.project_id = project_id
self.tag = tag
self.status = status
self.expected = expected
self.create_user = create_user
self.update_user = create_user
self.request_header = request_header
self.request_method = request_method
self.created_at = datetime.now()
self.updated_at = datetime.now()
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
基本上内容都在comment注释里面了,大家看着就行,也不多讲解了。肯定现在还有一些特性没有考虑到的,后序可以加字段解决。
# 编写检查用户权限的装饰器
我们在上面三张表里面定义好了create_user
和update_user
,其实这2个字段接受的都是咱们的用户id,而用户id又是可以通过token
进行解析的。
还记得之前编写过的解析token的方法吗?我们现在就来编写一个装饰器: 自动判断用户token是否有效以及获取用户信息。
原理很简单,就是通过headers中的token字段获取token,经过解析后拿到用户信息,这里更加复杂一点,对于那种需要有权限控制的地方,我们做了一层判断,如果权限在配置范围以下,那么普通成员就不能调用这个接口。
完整代码:
from functools import wraps
from flask import request, jsonify
from app import pity
from app.middleware.Jwt import UserToken
FORBIDDEN = "对不起, 你没有足够的权限"
class SingletonDecorator:
def __init__(self, cls):
self.cls = cls
self.instance = None
def __call__(self, *args, **kwargs):
if self.instance is None:
self.instance = self.cls(*args, **kwargs)
return self.instance
def permission(role=pity.config.get("GUEST")):
def login_required(func):
@wraps(func)
def wrapper(*args, **kwargs):
try:
headers = request.headers
token = headers.get('token')
if token is None:
return jsonify(dict(code=401, msg="用户信息认证失败, 请检查"))
user_info = UserToken.parse_token(token)
# 这里把user信息写入kwargs
kwargs["user_info"] = user_info
except Exception as e:
return jsonify(dict(code=401, msg=str(e)))
# 判断用户权限是否足够, 如果不足够则直接返回,不继续
if user_info.get("role", 0) < role:
return jsonify(dict(code=400, msg=FORBIDDEN))
return func(*args, **kwargs)
return wrapper
return login_required
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
kwargs["user_info"] = user_info
重点注意一下这句话。
# 为之前咱们编写的http/request接口加入权限
由于我们这个接口暂时支持登录用户使用,所以权限默认就行了,也就是说用户role为0(普通用户)即可!
这里需要注意的是,我们的http_request必须带有user_info
参数,否则会报错,因为我们后续可能会用到用户信息。
# 改造前端代码
因为我们之前前端在请求接口的时候,是不会带上用户token的,所以我们需要改造前端请求方法:
- 在发送http请求的时候带上token数据
- 校验返回code,如果为401则自动注销用户,提示用户未登录!
- 创建src/utils/auth.js
import { message } from 'antd';
export default {
headers: (json = true) => {
const token = localStorage.getItem('pityToken');
const headers = { token };
if (json) {
headers['Content-Type'] = 'application/json';
}
return headers;
},
response: (res, info = false) => {
if (res.code === 0) {
if (info) {
message.info(res.msg);
}
return true;
}
if (res.code === 401) {
// 说明用户未认证
message.info(res.msg);
localStorage.setItem('pityToken', null);
localStorage.setItem('pityUser', null);
window.location.href = '/user/login';
}
message.error(res.msg);
return false;
},
};
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
这里编写了获取token并组成headers的方法,也编写了针对token过期,权限不够的code码进行不同的处理。
- 修改src/services/request.js
- 修改请求结果:
开始测试
发现我们的错误信息并没有完全展示,是因为我们需要没有针对请求失败的情况进行校验,所以我们还需要调整后端接口:
判断里面的status是否为True,不为True的话返回110状态码
来个动态图测试一下吧:
# 接着测试一下权限
通过开发者工具的application可以看到:
接着我们调整一下权限为: 只有管理员可以请求接口
来个动图看看:
# ![](https://static.pity.fun/picture/2021-4-3/1617380169596-ff.gif)
打完收工,最近清明可以稍微更新多一点
😳
晚安,我的12个(包括我自己)观众们!
# 广告时间
预览地址: http://47.112.32.195/ (opens new window)
后端代码地址: https://github.com/wuranxu/pity (opens new window)
前端代码地址: https://github.com/wuranxu/pityWeb (opens new window)
欢迎关注公众号测试开发入坑,还有加群一起讨论相关问题呀!如果群二维码过期了,可以加我个人微信: wuranxu
我拉你进群~