测试平台系列(70) 丰富断言类型

这是我参与11月更文挑战的第2天，活动详情查看：2021最后一次更文挑战

大家好，我是米洛，正在打造一个能用版测试平台。求点赞!求关注!求分享！🙋🙋🙋

项目地址: pity.readthedocs.io/

关注公众号: 米洛的测开日记，让我们进一步交流。

回顾

上一节我们支持了SQL类型的构造方法。

接下来我们要讲的是如何丰富我们的断言类型。

基础断言

我们之前的断言部分还是比较残缺的，就好像辟邪剑谱一样。今天我们就来尽量完善下断言的内容。

先看看之前的断言类型:

等于
不等于
属于

但这些往往还不够，我们还需要新增一些。还记得我们之前写的如何比对2个JSON对象吗？我们今天也要把它运用到实战里面。

编写utils/json_compare.py

由于之前已经讲过json对比的思路，所以这边贴一下源码，并把JsonCompare类进行单例包装。

import json

from app.utils.decorator import SingletonDecorator


@SingletonDecorator
class JsonCompare(object):

    def compare(self, exp, act):
        ans = []
        self._compare(exp, act, ans, '')
        return ans

    def _compare(self, a, b, ans, path):
        a = self._to_json(a)
        b = self._to_json(b)
        if type(a) != type(b):
            ans.append(f"{path} 类型不一致, 分别为{type(a)} {type(b)}")
            return
        if isinstance(a, dict):
            keys = []
            for key in a.keys():
                pt = path + "/" + key
                if key in b.keys():
                    self._compare(a[key], b[key], ans, pt)
                    keys.append(key)
                else:
                    ans.append(f"{pt} 在实际结果中不存在")
            for key in b.keys():
                if key not in keys:
                    pt = path + "/" + key
                    ans.append(f"{pt} 在实际结果中多出")
        elif isinstance(a, list):
            i = j = 0
            while i < len(a):
                pt = path + "/" + str(i)
                if j >= len(b):
                    ans.append(f"{pt} 在实际结果中不存在")
                    i += 1
                    j += 1
                    continue
                self._compare(a[i], b[j], ans, pt)
                i += 1
                j += 1
            while j < len(b):
                pt = path + "/" + str(j)
                ans.append(f"{pt} 在预期结果中不存在")
                j += 1
        else:
            if a != b:
                ans.append(
                    f"{path} 数据不一致: {a} "
                    f"!= {b}" if path != "" else
                    f"数据不一致: {a} != {b}")

    def _color(self, text, _type=0):
        if _type == 0:
            # 说明是绿色
            return """<span style="color: #13CE66">{}</span>""".format(text)
        return """<span style="color: #FF4949">{}</span>""".format(text)

    def _weight(self, text):
        return """<span style="font-weight: 700">{}</span>""".format(text)

    def _to_json(self, string):
        try:
            float(string)
            return string
        except:
            try:
                if isinstance(string, str):
                    return json.loads(string)
                return string
            except:
                return string