mangotools 1.0.0__tar.gz

mangotools-1.0.0/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2024 maopeng
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

mangotools-1.0.0/MANIFEST.in

@@ -0,0 +1,3 @@
+exclude mangotools/mango/setup.py
+exclude mangotools/mango/mango.pyx
+exclude mangotools/mango/mango.c

mangotools-1.0.0/PKG-INFO

@@ -0,0 +1,34 @@
+Metadata-Version: 2.1
+Name: mangotools
+Version: 1.0.0
+Summary: 测试工具
+Home-page: https://gitee.com/mao-peng/testkit
+Author: 毛鹏
+Author-email: 729164035@qq.com
+Classifier: Programming Language :: Python :: 3.10
+License-File: LICENSE
+Requires-Dist: aiomysql==0.2.0
+Requires-Dist: PyMySQL==1.1.1
+Requires-Dist: jsonpath==0.82.2
+Requires-Dist: cachetools==5.3.1
+Requires-Dist: Faker==24.1.0
+Requires-Dist: diskcache==5.6.3
+Requires-Dist: pydantic==2.9.2
+Requires-Dist: colorlog==6.7.0
+Requires-Dist: pyarmor==9.0.6
+Requires-Dist: assertpy==1.1
+Requires-Dist: deepdiff==8.0.1
+
+# testkit
+
+#### 介绍
+
+测试工具
+
+#### 安装教程
+
+1. pip install mangokit
+
+#### 使用说明
+
+1.  

mangotools-1.0.0/README.md

@@ -0,0 +1,13 @@
+# testkit
+
+#### 介绍
+
+测试工具
+
+#### 安装教程
+
+1. pip install mangokit
+
+#### 使用说明
+
+1.  

mangotools-1.0.0/mangotools/__init__.py

@@ -0,0 +1,6 @@
+# -*- coding: utf-8 -*-
+# @Project: 芒果测试平台
+# @Description: 
+# @Time   : 2024-09-07 22:15
+# @Author : 毛鹏
+

mangotools-1.0.0/mangotools/assertion/__init__.py

@@ -0,0 +1,26 @@
+# -*- coding: utf-8 -*-
+# @Project: 芒果测试平台# @Description:
+# @Time   : 2023/4/6 13:36
+# @Author : 毛鹏
+
+from ..assertion._custom_assertion import CustomAssertion
+from ..assertion._public_assertion import WhatIsItAssertion, ContainAssertion, MatchingAssertion, \
+    WhatIsEqualToAssertion, PublicAssertion
+from ..assertion._sql_assertion import SqlAssertion
+import sys
+
+python_version = sys.version_info
+if f"{python_version.major}.{python_version.minor}" != "3.10":
+    raise Exception("必须使用>Python3.10.4")
+
+
+class Assertion(WhatIsItAssertion, ContainAssertion, MatchingAssertion, WhatIsEqualToAssertion):
+    pass
+
+
+__all__ = [
+    'Assertion',
+    'CustomAssertion',
+    'SqlAssertion',
+    'PublicAssertion',
+]

mangotools-1.0.0/mangotools/assertion/_custom_assertion.py

@@ -0,0 +1,27 @@
+# -*- coding: utf-8 -*-
+# @Project: 芒果测试平台
+# @Description: 
+# @Time   : 2024-01-05 18:05
+# @Author : 毛鹏
+from ..decorator import sync_method_callback
+from ..exceptions import ERROR_MSG_0061, MangoKitError
+from ..models import MethodModel
+
+
+class CustomAssertion:
+    """自定义断言"""
+
+    @staticmethod
+    @sync_method_callback('ass', '自定义断言', 7, [
+        MethodModel(f='func_str', p='请输入一个函数，在函数里面自己断言', d=True),
+        MethodModel(f='func_name', p='请输入这个函数的名称', d=True), ])
+    def ass_func(func_str, func_name='func'):
+        """输入断言代码"""
+        try:
+            global_namespace = {}
+            exec(func_str, global_namespace)
+            return global_namespace[func_name]
+        except (KeyError, SyntaxError, TypeError):
+            import traceback
+            traceback.print_exc()
+            raise MangoKitError(*ERROR_MSG_0061)

mangotools-1.0.0/mangotools/assertion/_public_assertion.py

@@ -0,0 +1,291 @@
+# -*- coding: utf-8 -*-
+# @Project: 芒果测试平台# @Description:
+# @Time   : 2023-09-09 23:17
+# @Author : 毛鹏
+from assertpy import assert_that
+from deepdiff import DeepDiff
+
+from mangokit.decorator import sync_method_callback
+from mangokit.models import MethodModel
+
+
+class WhatIsItAssertion:
+    """是什么"""
+
+    @staticmethod
+    @sync_method_callback('ass', '是什么', 0, [
+        MethodModel(f='actual', d=True)])
+    def p_is_not_none(actual):
+        """不是null"""
+        try:
+            assert_that(actual).is_not_none()
+        except AssertionError as e:
+            raise AssertionError(f'实际={actual}, 预期=不是null') from e
+
+    @staticmethod
+    @sync_method_callback('ass', '是什么', 1, [
+        MethodModel(f='actual', d=True)])
+    def p_is_none(actual):
+        """是null"""
+        try:
+            assert_that(actual).is_none()
+        except AssertionError as e:
+            raise AssertionError(f'实际={actual}, 预期=是null') from e
+
+    @staticmethod
+    @sync_method_callback('ass', '是什么', 2, [
+        MethodModel(f='actual', d=True)])
+    def p_is_empty(actual):
+        """是空字符串"""
+        try:
+            assert_that(actual).is_empty()
+        except AssertionError as e:
+            raise AssertionError(f'实际={actual}, 预期=是空字符串') from e
+
+    @staticmethod
+    @sync_method_callback('ass', '是什么', 3, [
+        MethodModel(f='actual', d=True)])
+    def p_is_not_empty(actual):
+        """不是空符串"""
+        try:
+            assert_that(actual).is_not_empty()
+        except AssertionError as e:
+            raise AssertionError(f'实际={actual}, 预期=不是空符串') from e
+
+    @staticmethod
+    @sync_method_callback('ass', '是什么', 4, [
+        MethodModel(f='actual', d=True)])
+    def p_is_false(actual):
+        """是false"""
+        try:
+            assert_that(actual).is_false()
+        except AssertionError as e:
+            raise AssertionError(f'实际={actual}, 预期=是false') from e
+
+    @staticmethod
+    @sync_method_callback('ass', '是什么', 5, [
+        MethodModel(f='actual', d=True)])
+    def p_is_true(actual):
+        """是true"""
+        try:
+            assert_that(actual).is_true()
+        except AssertionError as e:
+            raise AssertionError(f'实际={actual}, 预期=是true') from e
+
+    @staticmethod
+    @sync_method_callback('ass', '是什么', 6, [
+        MethodModel(f='actual', d=True)])
+    def p_is_alpha(actual):
+        """是字母"""
+        try:
+            assert_that(actual).is_alpha()
+        except AssertionError as e:
+            raise AssertionError(f'实际={actual}, 预期=是字母') from e
+
+    @staticmethod
+    @sync_method_callback('ass', '是什么', 7, [
+        MethodModel(f='actual', d=True)])
+    def p_is_digit(actual):
+        """是数字"""
+        try:
+            assert_that(actual).is_digit()
+        except AssertionError as e:
+            raise AssertionError(f'实际={actual}, 预期=是数字') from e
+
+
+class WhatIsEqualToAssertion:
+    """等于什么"""
+
+    @staticmethod
+    @sync_method_callback('ass', '等于什么', 0, [
+        MethodModel(f='actual', d=True), MethodModel(f='expect', p='请输入断言值', d=True)])
+    def p_is_equal_to(actual: str, expect: str):
+        """等于expect"""
+        try:
+            assert_that(actual).is_equal_to(expect),
+        except AssertionError as e:
+            raise AssertionError(f'实际={actual}, 预期={expect}') from e
+
+    @staticmethod
+    @sync_method_callback('ass', '等于什么', 1, [
+        MethodModel(f='actual', d=True), MethodModel(f='expect', p='请输入断言值', d=True)])
+    def p_is_not_equal_to(actual: str, expect: str):
+        """不等于expect"""
+        try:
+            assert_that(actual).is_not_equal_to(expect)
+        except AssertionError as e:
+            raise AssertionError(f'实际={actual}, 预期={expect}') from e
+
+    @staticmethod
+    @sync_method_callback('ass', '等于什么', 2, [
+        MethodModel(f='actual', d=True), MethodModel(f='expect', p='请输入断言值', d=True)])
+    def p_is_length(actual: str, expect: str):
+        """长度等于expect"""
+        try:
+            assert_that(actual).is_length(int(expect))
+        except AssertionError as e:
+            raise AssertionError(f'实际={actual}, 预期={expect}') from e
+
+    @staticmethod
+    @sync_method_callback('ass', '等于什么', 3, [
+        MethodModel(f='actual', d=True), MethodModel(f='expect', p='请输入断言值', d=True)])
+    def p_sum_equal_expect(actual: list, expect: str):
+        """长度等于expect"""
+        try:
+            assert_that(sum(actual)).is_equal_to(expect)
+        except AssertionError as e:
+            raise AssertionError(f'实际={actual}, 预期={expect}') from e
+
+
+class ContainAssertion:
+    """包含什么"""
+
+    @staticmethod
+    @sync_method_callback('ass', '包含什么', 0, [
+        MethodModel(f='actual', d=True), MethodModel(f='expect', p='请输入断言值', d=True)])
+    def p_contains(actual: str, expect: str):
+        """包含expect"""
+        try:
+            assert_that(actual).contains(expect)
+        except AssertionError as e:
+            raise AssertionError(f'实际={actual}, 预期={expect}') from e
+
+    @staticmethod
+    @sync_method_callback('ass', '包含什么', 1, [
+        MethodModel(f='actual', d=True), MethodModel(f='expect', p='请输入断言值', d=True)])
+    def p_is_equal_to_ignoring_case(actual: str, expect: str):
+        """忽略大小写等于expect"""
+        try:
+            assert_that(actual).is_equal_to_ignoring_case(expect)
+        except AssertionError as e:
+            raise AssertionError(f'实际={actual}, 预期={expect}') from e
+
+    @staticmethod
+    @sync_method_callback('ass', '包含什么', 2, [
+        MethodModel(f='actual', d=True), MethodModel(f='expect', p='请输入断言值', d=True)])
+    def p_contains_ignoring_case(actual: str, expect: str):
+        """包含忽略大小写expect"""
+        try:
+            assert_that(actual).contains_ignoring_case(expect)
+        except AssertionError as e:
+            raise AssertionError(f'实际={actual}, 预期={expect}') from e
+
+    @staticmethod
+    @sync_method_callback('ass', '包含什么', 3, [
+        MethodModel(f='actual', d=True), MethodModel(f='expect', p='请输入断言值', d=True)])
+    def p_contains_only(actual: str, expect: str):
+        """仅包含expect"""
+        try:
+            assert_that(actual).contains_only(expect)
+        except AssertionError as e:
+            raise AssertionError(f'实际={actual}, 预期={expect}') from e
+
+    @staticmethod
+    @sync_method_callback('ass', '包含什么', 4, [
+        MethodModel(f='actual', d=True), MethodModel(f='expect', p='请输入断言值', d=True)])
+    def p_does_not_contain(actual: str, expect: str):
+        """不包含expect"""
+        try:
+            assert_that(actual).does_not_contain(expect)
+        except AssertionError as e:
+            raise AssertionError(f'实际={actual}, 预期={expect}') from e
+
+
+class MatchingAssertion:
+    """匹配什么"""
+
+    @staticmethod
+    @sync_method_callback('ass', '匹配什么', 0, [
+        MethodModel(f='actual', d=True), MethodModel(f='expect', p='请输入断言值', d=True)])
+    def p_in_dict(actual: dict, expect: dict):
+        """JSON匹配"""
+        filtered_actual = filter_dict(dict(actual), dict(expect))
+        diff = DeepDiff(filtered_actual, expect, ignore_order=True)
+        assert not diff, f'实际={actual}, 预期={expect}'
+
+    @staticmethod
+    @sync_method_callback('ass', '匹配什么', 1, [
+        MethodModel(f='actual', d=True), MethodModel(f='expect', p='请输入断言值', d=True)])
+    def p_is_in(actual: str, expect: str):
+        """在expect里面"""
+        try:
+            assert_that(actual).is_in(expect)
+        except AssertionError as e:
+            raise AssertionError(f'实际={actual}, 预期={expect}') from e
+
+    @staticmethod
+    @sync_method_callback('ass', '匹配什么', 2, [
+        MethodModel(f='actual', d=True), MethodModel(f='expect', p='请输入断言值', d=True)])
+    def p_is_not_in(actual: str, expect: str):
+        """不在expect里面"""
+        try:
+            assert_that(actual).is_not_in(expect)
+        except AssertionError as e:
+            raise AssertionError(f'实际={actual}, 预期={expect}') from e
+
+    @staticmethod
+    @sync_method_callback('ass', '匹配什么', 3, [
+        MethodModel(f='actual', d=True), MethodModel(f='expect', p='请输入断言值', d=True)])
+    def p_starts_with(actual: str, expect: str):
+        """以expect开头"""
+        try:
+            assert_that(actual).starts_with(expect)
+        except AssertionError as e:
+            raise AssertionError(f'实际={actual}, 预期={expect}') from e
+
+    @staticmethod
+    @sync_method_callback('ass', '匹配什么', 4, [
+        MethodModel(f='actual', d=True), MethodModel(f='expect', p='请输入断言值', d=True)])
+    def p_ends_with(actual: str, expect: str):
+        """以expect结尾"""
+        try:
+            assert_that(actual).ends_with(expect)
+        except AssertionError as e:
+            raise AssertionError(f'实际={actual}, 预期={expect}') from e
+
+    @staticmethod
+    @sync_method_callback('ass', '匹配什么', 5, [
+        MethodModel(f='actual', d=True), MethodModel(f='expect', p='请输入断言值', d=True)])
+    def p_matches(actual: str, expect: str):
+        """正则匹配等于expect"""
+        try:
+            assert_that(actual).matches(expect)
+        except AssertionError as e:
+            raise AssertionError(f'实际={actual}, 预期={expect}') from e
+
+    @staticmethod
+    @sync_method_callback('ass', '匹配什么', 6, [
+        MethodModel(f='actual', d=True), MethodModel(f='expect', p='请输入断言值', d=True)])
+    def p_does_not_match(actual: str, expect: str):
+        """正则不匹配expect"""
+        try:
+            assert_that(actual).does_not_match(expect)
+        except AssertionError as e:
+            raise AssertionError(f'实际={actual}, 预期={expect}') from e
+
+
+def filter_dict(actual: dict, expect: dict) -> dict:
+    filtered = {}
+    for key in expect.keys():
+        if key in actual:
+            if isinstance(expect[key], dict):
+                filtered[key] = filter_dict(actual[key], expect[key])
+            elif isinstance(expect[key], list) and isinstance(actual[key], list):
+                filtered[key] = []
+                for item in actual[key]:
+                    if isinstance(item, dict):
+                        filtered_item = filter_dict(item, expect[key][0])
+                        filtered[key].append(filtered_item)
+                    else:
+                        filtered[key].append(item)
+            else:
+                filtered[key] = actual[key]
+    return filtered
+
+
+class PublicAssertion(WhatIsItAssertion, ContainAssertion, MatchingAssertion, WhatIsEqualToAssertion):
+    pass
+
+
+if __name__ == '__main__':
+    PublicAssertion.p_is_equal_to('1', '2')

mangotools-1.0.0/mangotools/assertion/_sql_assertion.py

@@ -0,0 +1,29 @@
+# -*- coding: utf-8 -*-
+# @Project: 芒果测试平台
+# @Description: 
+# @Time   : 2023-11-20 9:47
+# @Author : 毛鹏
+
+from mangokit.database import MysqlConnect
+from mangokit.decorator import sync_method_callback
+from mangokit.models import MethodModel
+
+
+class SqlAssertion:
+    """sql断言"""
+    mysql_connect: MysqlConnect = None
+
+    @staticmethod
+    @sync_method_callback('ass', 'sql断言', 0, [
+        MethodModel(f='sql', p='请输入sql语句，确保只会查出一个值', d=True),
+        MethodModel(f='expect', p='期望的json', d=True), ])
+    async def sql_is_equal(sql: str, expect: list[dict]):
+        """值相等"""
+        result = SqlAssertion.mysql_connect.condition_execute(sql)
+        assert all(dict2 in result for dict2 in expect), f'实际={result}, 预期=列表个数相等'
+
+
+if __name__ == '__main__':
+    _sql = "SELECT id,`name`,`status` FROM `project`;"
+    _expect = [{'id': 2, 'name': '1CDXP', 'status': 1}, {'id': 5, 'name': 'AIGC', 'status': 1},
+               {'id': 10, 'name': 'DESK', 'status': 1}, {'id': 11, 'name': 'AIGC-SaaS', 'status': 1}]

mangotools-1.0.0/mangotools/data_processor/__init__.py

@@ -0,0 +1,123 @@
+# -*- coding: utf-8 -*-
+# @Project: 芒果测试平台
+# @Description:
+# @Time   : 2023-03-07 8:24
+# @Author : 毛鹏
+import json
+import re
+
+import sys
+
+from ..data_processor._cache_tool import CacheTool
+from ..data_processor._coding_tool import CodingTool
+from ..data_processor._encryption_tool import EncryptionTool
+from ..data_processor._json_tool import JsonTool
+from ..data_processor._random_character_info_data import RandomCharacterInfoData
+from ..data_processor._random_number_data import RandomNumberData
+from ..data_processor._random_string_data import RandomStringData
+from ..data_processor._random_time_data import RandomTimeData
+from ..data_processor._sql_cache import SqlCache
+from ..exceptions import ERROR_MSG_0002, ERROR_MSG_0047, MangoKitError
+
+"""
+ObtainRandomData类的函数注释必须是： “”“中间写值”“”
+"""
+
+python_version = sys.version_info
+if f"{python_version.major}.{python_version.minor}" != "3.10":
+    raise Exception("必须使用>Python3.10.4")
+
+
+class ObtainRandomData(RandomNumberData, RandomCharacterInfoData, RandomTimeData, RandomStringData):
+    """ 获取随机数据 """
+
+    def regular(self, func: str):
+        """
+        反射并执行函数
+        :param func: 函数
+        :return:
+        """
+        match = re.search(r'\((.*?)\)', func)
+        if match:
+            try:
+                content = json.loads(match.group(1))
+                if not isinstance(content, dict):
+                    content = {'data': match.group(1)}
+            except json.decoder.JSONDecodeError:
+                content = {'data': match.group(1)}
+
+            func = re.sub(r'\(' + match.group(1) + r'\)', '', func)
+            try:
+                if content['data'] != '':
+                    return getattr(self, func)(**content)
+                return getattr(self, func)()
+            except AttributeError:
+                raise MangoKitError(*ERROR_MSG_0047)
+
+
+class DataClean(JsonTool, CacheTool, EncryptionTool, CodingTool):
+    """存储或处理随机数据"""
+    pass
+
+
+class DataProcessor(ObtainRandomData, DataClean):
+
+    def __init__(self):
+        ObtainRandomData.__init__(self)
+        DataClean.__init__(self)
+
+    def replace(self, data: list | dict | str | None) -> list | dict | str | None:
+        if not data:
+            return data
+        if isinstance(data, list):
+            return [self.replace(item) for item in data]
+        elif isinstance(data, dict):
+            return {key: self.replace(value) for key, value in data.items()}
+        else:
+            return self.replace_str(data)
+
+    def replace_str(self, data: str) -> str:
+        replace_list = re.findall(r"\${{.*?}}", str(data))
+        for replace_value in replace_list:
+            key_text = self.remove_parentheses(replace_value)
+            args = key_text.split("，")
+            key_text, key = (args[0], args[1]) if len(args) == 2 else (args[0], None)
+            if key and (key_value := self.get_cache(key)):
+                return key_value
+
+            value = self.regular(key_text) if self.identify_parentheses(key_text) else self.get_cache(key_text)
+            if value is None:
+                raise MangoKitError(*ERROR_MSG_0002, value=(key_text,))
+
+            if key:
+                self.set_cache(key, value)
+            data = data.replace(replace_value, str(value))
+        return data
+
+    @classmethod
+    def remove_parentheses(cls, data: str) -> str:
+        return data.replace("${{", "").replace("}}", "").strip()
+
+    @classmethod
+    def identify_parentheses(cls, value: str):
+        return re.search(r'\((.*?)\)', str(value))
+
+    @classmethod
+    def is_extract(cls, string: str) -> bool:
+        return bool(re.search(r'\$\{\{.*\}\}', string))
+
+
+__all__ = [
+    'CacheTool',
+    'CodingTool',
+    'EncryptionTool',
+    'JsonTool',
+    'RandomCharacterInfoData',
+    'RandomNumberData',
+    'RandomStringData',
+    'RandomTimeData',
+    'ObtainRandomData',
+    'DataClean',
+    'DataProcessor',
+    'SqlCache'
+]

mangotools-1.0.0/mangotools/data_processor/_cache_tool.py

@@ -0,0 +1,38 @@
+# -*- coding: utf-8 -*-
+# @Project: 芒果测试平台# @Description:
+# @Time   : 2023-08-29 10:23
+# @Author : 毛鹏
+
+from cachetools import LRUCache
+
+
+class CacheTool:
+    """ 内存缓存 """
+
+    def __init__(self):
+        self._cache: LRUCache = LRUCache(maxsize=500)
+
+    def get_cache(self, key: str) -> any:
+        """得到缓存key的value"""
+        return self._cache.get(key)
+
+    def set_cache(self, key: str, value: any) -> None:
+        """设置一个内容到缓存"""
+        self._cache[key] = value
+
+    def delete_cache(self, key: str) -> None:
+        """删除一个缓存"""
+        if key in self._cache:
+            del self._cache[key]
+
+    def clear_cache(self) -> None:
+        """清理所有缓存"""
+        self._cache.clear()
+
+    def has_cache(self, key: str) -> bool:
+        """判断缓存是否存在"""
+        return key in self._cache
+
+    def get_all(self, ) -> dict:
+        """获取全部的缓存数据"""
+        return {k: v for k, v in self._cache.items()}

mangotools-1.0.0/mangotools/data_processor/_coding_tool.py

@@ -0,0 +1,21 @@
+# -*- coding: utf-8 -*-
+# @Project: 芒果测试平台
+# @Description:
+# @Time   : 2023-08-30 14:14
+# @Author : 毛鹏
+import base64
+import json
+
+
+class CodingTool:
+    """编码"""
+
+    @classmethod
+    def base64_encode(cls, data: str) -> str:
+        """编码字符串"""
+        return base64.b64encode(json.dumps(data).encode('utf-8')).decode('utf-8')
+
+    @classmethod
+    def response_decoding(cls, data) -> str:
+        """解码字符串"""
+        return data.encode('latin-1').decode('unicode_escape').encode('utf-8', 'ignore').decode('utf-8')