Arthur-Wu commit

commons/EngineX.py

@@ -0,0 +1,107 @@
+# !/usr/bin/python
+# -*- coding: utf-8 -*-
+'''@Author:: Arthur Wu
+@Date:: 2024/10/8
+'''
+import os,pytest,subprocess,json,shutil,time,requests
+from configs.globalObj import *
+from configs.globalParams import *
+
+class Engine():
+
+    def __init__(self):
+        self.CasesSuite = []
+        self.rootpath = os.path.dirname(os.path.dirname(__file__))
+        self.FolderPath = ReturnFolder
+        self.projectCasesPath = str(os.path.join(GlobalPath['YMCaseScriptsPath']).replace("\\", "/"))
+        self.ReportName = "Reports"
+
+    def __execute(self):
+        '''@Author:: Arthur Wu
+        @Date:: 2024/5/9
+        :return:
+        '''
+        for dirpath, dirnames, filenames in os.walk(self.projectCasesPath):
+            for fn in filenames:
+                if fn.startswith("test_"):
+                    caseScriptPath = os.path.join(dirpath, fn).replace("\\", "/")
+                    self.CasesSuite.append(caseScriptPath)
+        if 0 == len(self.CasesSuite):
+            raise ValueError("---[ERROR] the test suite was empty !")
+        else:
+            pytest.main(args=[*self.CasesSuite, '-vsq', '--alluredir', self.FolderPath])
+            subprocess.Popen(
+                "allure generate " + self.FolderPath + ' -o ' + self.FolderPath + "html"
+                + ' --workers=2 '+'--tests-per-worker=2',
+                shell=True,
+                stdout=subprocess.PIPE).stdout.read()
+            LOGGER.info("---[INFO] Allure html report generate complete !")
+
+    def __move_file(self):
+        rootpath = (os.getcwd()).replace("\\", "/")
+        pathli = rootpath.split("/")
+        LastReportPath = ''
+        PathList = self.FolderPath.split("/")
+        try:
+            LastReportPath = "D:/Program Files/AutoTestReports/"
+            shutil.move(self.FolderPath, LastReportPath)
+        except Exception as e:
+            LOGGER.error(f"---Move the html report file error: {e}")
+        return os.path.join(LastReportPath, PathList[-2]).replace("\\", "/")
+
+    def __delete_pycache_dirs(self, target_dir):
+        for root, dirs, files in os.walk(target_dir):
+            for dir_name in dirs:
+                if dir_name == '__pycache__':
+                    dir_path = os.path.join(root, dir_name)
+                    shutil.rmtree(dir_path)
+                    print(f"已删除: {dir_path}")
+
+    def __delete_txt_files(self, target_dir):
+        for root, dirs, files in os.walk(target_dir):
+            for file_name in files:
+                if file_name.endswith('.txt'):
+                    file_path = os.path.join(root, file_name)
+                    os.remove(file_path)
+                    print(f"已删除: {file_path}")
+
+    def run_test_suite(self):
+        txtPath = os.path.join(self.rootpath, "YiMao/ProcessData")
+        self.__delete_pycache_dirs(self.rootpath)
+        self.__delete_txt_files(txtPath)
+        self.__execute()
+        returnList = []
+        lastReportPath = self.__move_file()
+        returnList.append(lastReportPath)
+        self.__delete_txt_files(txtPath)
+        self.__delete_pycache_dirs(self.rootpath)
+        return returnList
+
+class NotificationModule():
+    def __init__(self):
+        self.__headers = {'Content-Type': 'application/json;charset=utf-8'}
+        datainfo = {
+            "Debug": "7aedbee7239870e3e653748a2889d8bf063c61efa9213c7099bd57476066dc86",
+            "Formal": "80b026022a28166cfc9eebaf8f6a880cc06f56a14b8803e8d67e7fb3cb05844e"
+        }
+        self.urlInfo = f'https://oapi.dingtalk.com/robot/send?access_token={datainfo["Debug"]}'
+        # self.urlInfo = f'https://oapi.dingtalk.com/robot/send?access_token={datainfo["Formal"]}'
+        self.ReportUrl = ProCfgData["ReportsURL"]
+        self.ExecutionEnvironment = ProCfgData["ExecutionEnv"]
+
+    def send_msg(self, ResultPath):
+        ReportPath = os.path.join(ResultPath[0], 'html', 'index.html').replace('\\', '/')
+        reportUrlpath = self.ReportUrl + ReportPath.split("AutoTestReports")[1]
+        my_data = {
+            "msgtype": "markdown",
+            "markdown": {
+                "title": "亿猫管理后台接口自动化测试",
+                "text": f"#### 亿猫管理后台({self.ExecutionEnvironment})环境自动化测试执行结束: \n"
+                        f">#### [自动化测试报告链接(请点击查看)]({reportUrlpath})\n>\n"
+            }
+        }
+        sess = requests.session()
+        sess.keep_alive = False
+        requests.post(url=self.urlInfo, data=json.dumps(my_data), headers=self.__headers, verify=False)
+        sess.close()
+