目的：pandas读取csv文件中的数据，做接口自动化测试
要求：每行数据是一个测试用例，运行脚本后，自动按顺序执行测试用例
1.请求方式不同，需要的参数不同。所以有缺失值需要处理
2.需要将返回的状态码写入原文件

import pandas as pd
#使用pandas库读取csv文件

class option_csv(object):
    def __init__(self):
        self.file = pd.read_csv(path,index_col=0,header=0)
        self.file = self.file.where(self.file.notnull(),None)#将nan替换为none

    def get_line(self):#获取所有行数
        lens = len(self.file)
        return lens

    def get_row(self,row):#获取某列的所有数据
        return self.file.iloc[row,:]

    def write_result(self,result,row):#修改第row行result的内容
        self.file.loc[row,'result'] = result
        self.file.to_csv(path)

'''
将数据从csv中取出
'''
from DataBase import read_excel

class read_data(object):
    def __init__(self):
        self.files = read_excel.option_csv()
  
    def get_value(self,row):
        self.case = self.files.get_row(row)

        self.name = self.case["case_name"]
        self.method = self.case['method']
        self.params = self.case['parame']
        self.url = self.case['url']
        self.data = self.case['data']
        self.body = self.case['body']
        self.espect_code = str(self.case["espect_code"])

        return self

'''
读取测试用例运行
'''
import unittest
import requests
from DataBase import read_data, read_excel

class test_run(unittest.TestCase):
    def setUp(self) -> None:
        print("====begin the test====")

    def test_run(self):
        nrows = read_excel.option_csv().get_line()
        get = read_data.read_data()

        for i in range(1, nrows):
            case = read_data.read_data().get_value(i)
            try:
                req = requests.request(method=case.method, url=case.url, params=case.params, json=case.body,
                                       data=case.data)
                res = req.status_code
                self.assertEqual(case.espect_code, res, msg=f"{case.name}is failed" + "\n" + f"{req.text}")
                read_excel.option_csv().write_result(res, i)

            except (requests.ConnectionError, requests.HTTPError, requests.URLRequired, requests.Timeout) as e:
                print(e)

    def tearDown(self) -> None:
        print("====end the test====")

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