接口测试（二） 优化项目分层及cookies值带入

#test_case.py
from project_data.data_test import *
from project_driver.http_pg import *
import unittest
import json
from ddt import data,unpack,ddt

@ddt
class test_case(unittest.TestCase):
    @classmethod
    def setUpClass(cls):
        cls.http_test = HTTP_CONSOLE()
    @classmethod
    def tearDownClass(cls):
        print("用例执行完毕")

    @data(*Date_test.Date_test_1())
    @unpack
    def test_case_1(self,case_id,url,data,Method,expected):
        result = self.http_test.http_console(Method=Method, url=url, parameter=eval(data))
        Date_test().Date_test_2(test_function=result.text, x=case_id)
        self.assertEqual(json.loads(result.text)["status"], eval(expected)["status"])
        self.assertEqual(json.loads(result.text)["code"], eval(expected)["code"])
        self.assertEqual(json.loads(result.text)["msg"], eval(expected)["msg"])


if __name__ == ‘__main__‘:
    unittest.main()

#data_test.py
from openpyxl import load_workbook
import json
import os

class Date_test():
    filepath = os.path.dirname(os.path.dirname(__file__))
    @classmethod
    def Date_test_1(cls):
        """
        配置文件读取模块
        :return:
        """
        wb = load_workbook(Date_test().filepath + "\project_data\cases.xlsx")
        ws = wb["test"]
        url = ‘http://test.lemonban.com/futureloan/mvc/api‘
        list = []
        for i in range(2,ws.max_row + 1):
            result = {}
            result["case_id"] = ws.cell(i, 1).value
            result["url"] = url + ws.cell(i,3).value
            result["data"] = ws.cell(i,4).value
            result["Method"] = ws.cell(i,5).value
            result["expected"] = ws.cell(i,6).value
            list.append(result)
        wb.close()
        return list

    def Date_test_2(self, test_function, x):
        """
        运行结果写入模块
        :param test_function:
        :param test_text:
        :param x:
        :return:
        """
        Dict = {}
        wb = load_workbook(Date_test().filepath+"\project_data\cases.xlsx")
        ws = wb["test"]
        test_function = json.loads(test_function)
        Dict["status"] = test_function["status"]
        Dict["code"] = test_function["code"]
        Dict["msg"] = test_function["msg"]
        ws.cell(x+1, 7).value = str(Dict)
        if ws.cell(x+1,6).value == ws.cell(x+1, 7).value:
            ws.cell(x+1, 8).value = "PASS"
        else:
            ws.cell(x+1, 8).value = "FAIL"
        ws.cell(x+1, 9).value = str(test_function["data"])
        wb.save(Date_test().filepath+"\project_data\cases.xlsx")
        wb.close()

    @staticmethod
    def Email_Date():
        username = "username"
        password = "password"
        To_Email = "To_Email_1,To_Email_2"
        return username, password, To_Email

#http_pg.py
import requests
class HTTP_CONSOLE():
    def __init__(self):
        self.session = requests.session()
    def http_console(self, Method, url, parameter = None,json = None ,Cookies=None):
        if parameter == None:
            result = self.session.post(url=url, json=eval(json), cookies=Cookies)
        else:
            Method = Method.upper()
            if Method == "GET":
                result = self.session.get(url=url, params=parameter, cookies=Cookies)
            elif Method == "POST":
                result = self.session.post(url=url, data=parameter, cookies=Cookies)
            else:
                print("请求方式输入错误，目前仅支持POST/GET两种请求方式！")

        return result

#Email_console.py
import HTMLTestRunnerNew
from email.mime.text import MIMEText
from email.header import Header
import smtplib
import unittest
from project_data.data_test import *
from email.mime.multipart import MIMEMultipart


class Email_Console():
    def __init__(self):
        self.username, self.password, self.To_Email = Date_test.Email_Date()

    def send_main(self, file_new):
        """
        发送邮件方法
        :param file_new:
        :return:
        """
        msg = MIMEMultipart("Emaik_TestText")
        msg[‘Subject‘] = Header(‘自动化测试报告‘, ‘utf-8‘)
        msg[‘From‘] = self.username
        msg[‘To‘] = self.To_Email
        with open(file_new, ‘rb‘) as f:
            mail_body = f.read()
            msg.attach(MIMEText(mail_body, ‘html‘, ‘utf-8‘))
        # 添加附件result.html
        with open("result.html", "rb") as f:
            mail_attach = f.read()
            att1 = MIMEText(mail_attach, ‘base64‘, ‘utf-8‘)
            att1["Content-Type"] = ‘application/octet-stream‘
            att1["Content-Disposition"] = ‘attachment; filename="report_test.html"‘
            msg.attach(att1)
        # 获取路径
        filepath = Date_test().filepath + "\project_data\cases.xlsx"
        # 添加附件cases.xlsx
        with open(filepath, "rb") as f:
            mail_attach = f.read()
            att2 = MIMEText(mail_attach, ‘base64‘, ‘utf-8‘)
            att2["Content-Type"] = ‘application/octet-stream‘
            att2["Content-Disposition"] = ‘attachment; filename=filepath‘
            msg.attach(att2)
        try:
            smtp = smtplib.SMTP()
            smtp.connect("smtp.163.com", 25)
            smtp.login(self.username, self.password)
            smtp.sendmail(self.username, self.To_Email.split(","), msg.as_string())
            smtp.quit()
        except Exception as e:
            print("Send Email Failed!!!")
            raise e

    def new_report(self, testreport):
        """
        生成并查找查找最新测试报告方法
        :param testreport:
        :return:
        """
        # 生成测试用例
        fp = open("result.html", ‘wb‘)
        runner = HTMLTestRunnerNew.HTMLTestRunner(stream=fp, title=‘2019年4月11日前程贷接口测试报告‘, description=‘所有测试情况‘,
                                                  tester="桂马")
        discove = unittest.defaultTestLoader.discover(".", pattern="test_*.py")
        runner.run(discove)
        fp.close()
        # 查找测试用例
        lists = os.listdir(testreport)
        lists.sort(key=lambda fn: os.path.getmtime(testreport + "\\" + fn))
        file_new = os.path.join(testreport, lists[-1])
        print(file_new)
        return file_new


if __name__ == "__main__":
    email_test = Email_Console()
    file_path = email_test.new_report(os.getcwd())  # 查找新生的报告
    email_test.send_main(file_new=file_path)  # 调用发邮件模块(调用前需要修改data_test.py配置文件中email配置模块数据 )