大家好~我是
米洛！
我正在从0到1打造一个开源的接口测试平台, 也在编写一套与之对应的完整教程，希望大家多多支持。
欢迎关注我的公众号米洛的测开日记，获取最新文章教程!
回顾
前文我们支持了Redis这种前置条件，但其实有个特别不友好的地方:
如果前置条件比较接近，我甚至于不能复制它们。
如果能像前置case一样，选择一个去快速复制，所以说这样是十分不友好的。
那这一节我们就来实现它。
效果图
老规矩，先放上效果图:
思路
其实要支持这个操作，对于后端来说还是比较简单的。
好在我们的后端的Constructor有个type字段，用于区分前置条件的类型。所以我们只需要安排一个查询接口:
根据类型查找出前置条件的明细，然后选择后，前端用表单的方式带入进来，即可复制整个前置条件。
后端先行
编写查询接口，展示它属于哪一个case，并且给出前置条件的细节。
前端呢，则是已经确定好了TreeSelect组件，所以后端采用2层的数据格式:
[
{
"title": "用例",
"key": "caseid",
"children": [
{
"title": "数据构造器A",
"key":
"constructor_id"
}
]}
]
那我们就造起来吧~
-
编写思路
我不打算用join(从一开始我就贯彻了自己这个思想，我更喜欢用2次查询，自己拼接数据的方式，其实是join用的不好)我们优先查出对应类型的
前置条件，接着就能拿到所有case_id了，再用case_id去查出case的信息，岂不美哉？只是难点在于怎么组成这个树，但其实这个树只有2层，我们要做的就是先组装children，最后拼parent。
这里我们就要用到
defaultdict了。- 查出符合条件的前置条件，并根据case_id为key，前置条件为value的方式，存放到defaultdict
- 根据defaultdict的keys查出case_id对应的case名称，遍历之，拼凑parent，并把defaultdict里面的value放到children字段。
来看看代码:
@staticmethod
async def get_case_and_constructor(constructor_type: int):
# 最终返回结果树
ans = list()
async with async_session() as session:
# 此处存放case_id => 前置条件的映射
constructors = defaultdict(list)
# 根据传入的前置条件类型，找出所有前置条件, 类型一致，共享开关打开，并未被删除
query = await session.execute(
select(Constructor).where(
Constructor.type == constructor_type,
Constructor.public == True,
Constructor.deleted_at == None))
# 并把这些前置条件放到constructors里面
for q in query.scalars().all():
constructors[q.case_id].append({
"title": q.name,
"key": f"{q.id}",
"isLeaf": True,
# 这里是为了拿到具体的代码，因为树一般只有name和id，我们这还需要其他数据
"constructor_json": q.constructor_json,
})
# 二次查询，查出有前置条件的case
query = await session.execute(
select(TestCase).where(TestCase.id.in_(constructors.keys()), TestCase.deleted_at == None))
# 构造树，要知道children已经构建好了，就在constructors里面
for q in query.scalars().all():
# 把用例id放入cs_list，这里就不用原生join了
ans.append({
"title": q.name,
"key": f"case_{q.id}",
"disabled": True,
"children": constructors[q.id]
})
return ans
注释写的非常详细，每一步都交代的很清楚。
接着是接口部分，就不需要多说了:
# 获取所有数据构造器
@router.get("/constructor/list")
async def list_case_and_constructor(constructor_type: int):
try:
ans = await ConstructorDao.get_case_and_constructor(constructor_type)
return PityResponse.success(ans)
except Exception as e:
return PityResponse.failed(str(e))
前端适配
其实前端适配很简单，加一个组件，并且在组件选中值的时候更新表单的值就可以了。
oh shit~~
写到这里我才发现我居然写过类似的代码，难受= =
果然人会在同一个地方倒下，不过我觉得之前写的，没有对前置条件进行分类，所以我得改造改造，或者说这次写的当做v2接口使用。
不得不说，前端的改动比想象中还大一点。仔细比对了这个接口，发现它的逻辑是:
- 找到对应的前置条件id
- 查询前置条件的数据，进行替换
但之前的接口只针对case类型，所以很鸡肋。
我们目前把它做了一个船新版本来适配我们所有的类型:
import {Col, Row, TreeSelect} from "antd";
import {connect} from 'umi';
import {useEffect} from "react";
const CopyTreeSelect = ({construct, dispatch}) => {
const {constructorData, searchConstructor, constructorType} = construct;
const save = (data) => {
dispatch({
type: 'construct/save',
payload: data,
})
}
const getConstructorData = () => {
dispatch({
type: 'construct/getConstructorTree',
payload: {
constructor_type: constructorType
}
})
}
useEffect(() => {
getConstructorData();
}, [constructorType])
return (
<Row style={{marginTop: 24, marginBottom: 24}}>
<Col span={3}/>
<Col span={18}>
<Row>
<Col span={4}/>
<Col span={20}>
<TreeSelect
allowClear
showSearch
style={{ '100%'}}
value={searchConstructor}
filterTreeNode={(inputValue, treeNode) => {
return treeNode.title.toLowerCase().indexOf(inputValue.toLowerCase()) > -1
}}
dropdownStyle={{maxHeight: 600, overflow: 'auto'}}
treeData={constructorData}
placeholder="通过搜索构造条件，可以快速复制参数哦！"
treeDefaultExpandAll
onChange={(e) => {
save({searchConstructor: e})
if (e !== undefined) {
dispatch({
type: 'construct/getConstructorData',
payload: {id: e.split("_")[1]}
})
} else {
dispatch({
type: 'construct/save',
payload: {testCaseConstructorData: {type: constructorType, public: true, enable: true}},
})
}
}}
/>
</Col>
</Row>
</Col>
<Col span={3}/>
</Row>
)
}
export default connect(({loading, construct}) => ({loading, construct}))(CopyTreeSelect);
把这个组件单独抽出，成为一个copy组件。并在前置条件类型变化的时候，自动更换数据源。
页面上再进行一些微调，就ok了。说实话这块有点臃肿，由于我自己都看不下去，所以我先放过自己，也放过大家。
下一节我们小试牛刀，编写一个简单的测试报告邮件通知功能。（其实说简单也不简单，如果要设计到样式，还挺麻烦）