Update test framework: fix run_tests.py to support all test files, add auto-import-check for test files

base_framework/platform_tools/IDF/common/apiResponse.py

@@ -0,0 +1,54 @@
+# encoding: UTF-8
+
+from flask import make_response
+import json
+from ..const import RES_CODE
+
+
+class ApiResponse(object):
+    def __init__(self):
+        self.success = False
+        self.code = ''
+        self.message = ''
+        self.data = {}
+
+    @staticmethod
+    def build_success(code=20000, message='', data=None):
+        if data is None:
+            data = {}
+        response = ApiResponse()
+        response.success = True
+        response.code = code
+        response.message = message
+        response.data = data
+        return response.cors_response(make_response(json.dumps(response, default=obj_2_json)))
+
+    @staticmethod
+    def build_failure(code, msg='', data=None):
+        response = ApiResponse()
+        if data is None:
+            data = {}
+        if not msg:
+            response.message = RES_CODE[code]
+        else:
+            response.message = msg
+        response.success = False
+        response.code = code
+        response.data = data
+        return response.cors_response(make_response(json.dumps(response, default=obj_2_json)))
+
+    @staticmethod
+    def cors_response(res):
+        res.headers['Access-Control-Allow-Origin'] = '*'
+        res.headers['Access-Control-Allow-Methods'] = 'POST'
+        res.headers['Access-Control-Allow-Headers'] = 'x-requested-with,content-type'
+        return res
+
+
+def obj_2_json(obj):
+    return {
+        'success': obj.success,
+        'code': obj.code,
+        'message': obj.message,
+        'data': obj.data
+    }

base_framework/platform_tools/IDF/common/cronRequest.py

@@ -0,0 +1,35 @@
+from ..const import CREATE_URI
+
+from ..common.getRequest import Request
+
+
+class CronRequest(object):
+    def __init__(self, token):
+        self.stress_api = CREATE_URI
+        self.headers = {'accesstoken': token, 'Accept': '*/*', 'content-type': 'application/json;charset=UTF-8'}
+
+    def create(self, params):
+        url = self.stress_api + '/back-end/stress/schedule/save'
+        ret = Request.go('post', url, params, self.headers)
+        if not ret:
+            return
+        return ret.get('id')
+
+    def pause(self, jid):
+        url = self.stress_api + '/back-end/stress/schedule/pause'
+        params = [jid]
+        Request.go('post', url, params, self.headers)
+
+    def resume(self, jid):
+        url = self.stress_api + '/back-end/stress/schedule/resume'
+        params = [jid]
+        Request.go('post', url, params, self.headers)
+
+    def remove(self, jid):
+        url = self.stress_api + '/back-end/stress/schedule/delete'
+        params = [jid]
+        Request.go('post', url, params, self.headers)
+
+    def update(self, req_params):
+        url = self.stress_api + '/back-end/stress/schedule/update'
+        Request.go('post', url, req_params, self.headers)

base_framework/platform_tools/IDF/common/feishuMessage.py

@@ -0,0 +1,16 @@
+import requests
+
+
+class FeiShuMessage:
+
+    def __init__(self):
+        self.headers = {'Content-Type': 'application/json; charset=utf-8'}
+        self.webhook = "https://open.feishu.cn/open-apis/bot/v2/hook/180fa48e-1474-448e-a3d5-1a530f6ca689"
+
+    def send_message(self, msg, url=None):
+        url = url if url else self.webhook
+        res = requests.post(url, headers=self.headers, json=msg, verify=False)
+        if res.status_code == 200:
+            return True
+        else:
+            return False

base_framework/platform_tools/IDF/common/getRequest.py

@@ -0,0 +1,41 @@
+# -*- coding: utf-8 -*-
+import requests
+import json
+from requests.exceptions import ConnectionError
+
+from logger import logger
+
+
+class Request(object):
+
+    @classmethod
+    def go(cls, method, url, params, headers=None, noFormat=False):
+        # logger.info(f'发送{method}请求到: {url}, 参数： {params}')
+        try:
+            if method == 'get':
+                response = requests.get(url=url, params=params, headers=headers, timeout=20)
+            elif method == 'post':
+                response = requests.post(url=url, data=json.dumps(params), headers=headers, timeout=20)
+            else:
+                logger.error(f'暂不支持{method}方法')
+                return
+        except ConnectionRefusedError:
+            logger.error(f'服务请求失败：{url}')
+            return
+        except ConnectionError:
+            logger.error(f'服务无法链接： {url}')
+            return
+        if response.status_code != 200:
+            logger.error(f'返回码不等于200，请检查服务！{response.status_code}, {response.text}')
+        else:
+            resp_json = response.json()
+            # logger.info(f'返回内容：{resp_json}')
+            # noFormat: 不需要对返回内容进行校验，直接返回整个response
+            if noFormat:
+                return resp_json
+            # 对response做校验，返回体为qe平台的通用格式
+            if resp_json.get('success') or resp_json.get('code') == 20000:
+                return resp_json.get('data')
+            else:
+                logger.error(resp_json)
+                return

base_framework/platform_tools/IDF/common/getUserInfo.py

@@ -0,0 +1,24 @@
+from ..common.getRequest import Request
+
+from ..const import STRESS_URI
+
+
+class UserInfo(object):
+
+    @staticmethod
+    def get_user_info(access_token, url_prefix=None, info='userId'):
+        stress_uri = STRESS_URI
+        url = "/back-end/stress/user/info"
+        result = Request.go(method="get",
+                            url=stress_uri + url if not url_prefix else url_prefix + url,
+                            params=None,
+                            headers={"accessToken": access_token})
+        return None if not result else result.get(info)
+
+    @staticmethod
+    def get_user_info_by_user_id(access_token, user_id, info):
+        stress_uri = STRESS_URI
+        url = "/back-end/stress/user/infoFromId"
+        result = Request.go(method="get", url=stress_uri+url, params={'userId': user_id},
+                            headers={"accessToken": access_token})
+        return None if not result else result.get(info)

base_framework/platform_tools/IDF/common/sqlSession.py

@@ -0,0 +1,83 @@
+# 创建连接相关
+from sqlalchemy import create_engine, text
+from sqlalchemy.orm import sessionmaker
+
+from base_framework.platform_tools.IDF.const import create_sql_url
+
+from logger import logger
+
+"""
+sql操作：
+排序：order_by(ChartsName.column.desc()/asc())
+limit: .offset(n)过滤前面n条数据 .limit(n)
+count: .count()计数
+是否存在：is_exist = session.query(exists().where(Book.id > 10)).scalar()
+or: .filter(or_(Chart.column == x, Chart.column > y)).all()
+one: .one()只获取一条，如不存在或存在多条都会报错
+first: 通过主键获取记录  filter(**).first()
+"""
+
+
+class SqlSession:
+    def __init__(self, sql_uri=create_sql_url):
+        self.sql_uri = sql_uri
+        self._session = self.get_session()
+
+    # 创建连接对象，使用 pymsql 引擎
+    def get_session(self):
+        engine = create_engine(self.sql_uri, max_overflow=5, connect_args={'connect_timeout': 10})
+        Session = sessionmaker(bind=engine)
+        session = Session()
+        return session
+
+    def query(self, obj):
+        return self._session.query(obj)
+
+    def add(self, added):
+        self._session.add(added)
+
+    def add_all(self, added_list):
+        if isinstance(added_list, list):
+            self._session.add_all(added_list)
+        else:
+            logger.warning('只能传递list')
+
+    def flush(self):
+        self._session.flush()
+
+    def commit(self):
+        self._session.commit()
+
+    def close(self):
+        self._session.close()
+
+    def rollback(self):
+        self._session.rollback()
+
+    def object_session(self, instance):
+        self._session.object_session(instance)
+
+    def execute(self, sql):
+        return self._session.execute(text(sql))
+
+    def done(self, close=True):
+        """
+        执行完插入、删除、修改等操作后执行done，如报错回滚本次事务的sql操作
+        :return:
+        """
+        try:
+            self.commit()
+            if close:
+                self.close()
+        except Exception as e:
+            logger.warning(e)
+            self._session.rollback()
+            return e
+
+    @property
+    def session(self):
+        return self._session
+
+
+def to_dict(self):
+    return {c.name: getattr(self, c.name, None) for c in self.__table__.columns}