接口测试模块完整版

先上代码

#data_test.py
from openpyxl import load_workbook
class Date_test():
    @classmethod
    def Date_test_1(cls):
        """
        配置文件读取模块
        :return:
        """
        wb = load_workbook("data_test.xlsx")
        ws = wb["Sheet1"]
        url = 'http://localhost/api'
        list1 = []
        for i in range(1, ws.max_row + 1):
            list = []
            for j in range(1, ws.max_column + 1):
                if j == 3:
                    str = url + ws.cell(i, j).value
                else:
                    str = ws.cell(i, j).value
                list.append(str)
            list1.append(list)
        wb.close()
        Dict = {}
        for i in list1:
            str = ",".join(i)
            list = []
            list = str.split(",",1)
            Dict[list[0]] = list[1]
        return Dict
    def Date_test_2(self,test_function,test_text,x):
        """
        运行结果写入模块
        :param test_function:
        :param test_text:
        :param x:
        :return:
        """
        wb = load_workbook("data_test.xlsx")
        ws = wb["Sheet2"]
        ws.cell(x,1).value = test_function
        ws.cell(x,2).value = test_text
        wb.save("data_test.xlsx")
        wb.close()


    @staticmethod
    def Email_Date():
        username = "username"
        password = "password"
        To_Email = "from_email_1,from_email_2"
        return username,password,To_Email

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

        return result

#test_case.py
from data_test import *
from Http_pg import *
import json
import unittest
class test_case(unittest.TestCase):
    @classmethod
    def setUp(cls):
        cls.Cookie = test_case().test_login()
    @classmethod
    def tearDown(cls):
        print("测试用例执行完毕")

    def test_register(self):
        str = Date_test.Date_test_1()["register"]
        list = str.split(",")
        Method = list[0]
        url = list[1]
        Dict_login = {}
        Dict_login["mobilephone"] = eval(list[2])
        Dict_login["pwd"] = eval(list[3])
        result = HTTP_CONSOLE().http_console(Method=Method, url=url, parameter=Dict_login)
        Date_test().Date_test_2(test_function="register",test_text=json.loads(result.text)["msg"],x=1)
        # self.assertEqual(json.loads(result.text)["msg"],"注册成功")
        self.assertEqual(json.loads(result.text)["msg"],"手机号码已被注册")


    def test_login(self):
        str = Date_test.Date_test_1()["login"]
        list = str.split(",")
        Method = list[0]
        url = list[1]
        Dict_login= {}
        Dict_login["mobilephone"] = eval(list[2])
        Dict_login["pwd"] = eval(list[3])
        result = HTTP_CONSOLE().http_console(Method=Method,url=url,parameter=Dict_login)
        self.Cookie = result.cookies
        Date_test().Date_test_2(test_function="login", test_text=json.loads(result.text)["msg"],x=2)
        self.assertEqual(json.loads(result.text)["msg"],"登录成功")
        return self.Cookie

    def test_recharge(self):
        str = Date_test.Date_test_1()["recharge"]
        list = str.split(",")
        Method = list[0]
        url = list[1]
        Dict_login = {}
        Dict_login["mobilephone"] = eval(list[2])
        Dict_login["amount"] = eval(list[3])
        result = HTTP_CONSOLE().http_console(Method=Method, url=url, parameter=Dict_login,Cookies = self.Cookie)
        Date_test().Date_test_2(test_function="recharge", test_text=json.loads(result.text)["msg"], x=3)
        self.assertEqual(json.loads(result.text)["msg"], "充值成功")

    def test_withdraw(self):
        str = Date_test.Date_test_1()["withdraw"]
        list = str.split(",")
        Method = list[0]
        url = list[1]
        Dict_login = {}
        Dict_login["mobilephone"] = eval(list[2])
        Dict_login["amount"] = eval(list[3])
        result = HTTP_CONSOLE().http_console(Method=Method, url=url, parameter=Dict_login,Cookies = self.Cookie)
        Date_test().Date_test_2(test_function="withdraw", test_text=json.loads(result.text)["msg"], x=4)
        self.assertEqual(json.loads(result.text)["msg"], "取现成功")

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

#Email_test.py
import HTMLTestRunnerNew
from email.mime.text import MIMEText
from email.header import Header
import smtplib
import unittest
import os
from 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'))

        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)
        with open("data_test.xlsx","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="data_test.xlsx"'
            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月9日前程贷接口测试报告', 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.xlsx文件中Sheet1

　　　　　　　　　Sheet2

 

注意点：

　　　　1.setUp中获取返回cookies值算不上用得好，只能算面前能用，其他方法还有修改TestCase模块中默认值，添加self.Cookie这个值，然后直接用，但我不怎么想使用这个方法，所以放弃了

后记：

　　　　后续会陆续更新test_case文件中的测试用例，多写自然就熟悉了

 

转载于:https://www.cnblogs.com/keima/p/10694239.html