接口自动化2.0 python+unittest+ddt

本次在原有的基础上，封装了http请求，加入了ddt做数据驱动

tools新增http_requests方法

代码如下：

import requests
import json


class HttpRequests:

    def http_requests(self,method, url, data=None, expect=None, title=None, headers=None):
        if method.upper() == ‘POST‘:
            res = requests.post(url=url, json=data, headers=headers)
            try:
                return res.json()
            except Exception as e:
                return json.loads(res.text)
        elif method.upper() == ‘GET‘:
            res = requests.get(url=url, headers=headers)
            try:
                return res.json()
            except Exception as e:
                return json.loads(res.text)


if __name__ == ‘__main__‘:
    from gongdan.tools.get_data import Do_Excel
    from gongdan.tools import login

    headers = {"Content-Type": "application/json",
               ‘Accept‘: "application/json, text/plain, */*",
               "passport": login.login()
               }
    test_data = Do_Excel(r‘../test_data/test_data.xlsx‘, ‘form‘).get_data()

    print(test_data[1][‘url‘])
    print(test_data[1][‘data‘])
    print(test_data[1])
    res = HttpRequests().http_requests(method=test_data[1][‘method‘], url=test_data[1][‘url‘],
                                     data=json.loads(test_data[1][‘data‘]),
                                     headers=headers)
    # res1 = requests.post(url=test_data[2][‘url‘],
    #                                  json=json.loads(test_data[2][‘data‘]),
    #                                  headers=headers)
    print(res)

写用例引入了ddt来处理数据，代码如下：

import unittest
import json
from gongdan.tools.get_data import Do_Excel
from gongdan.tools import login
from gongdan.tools.http_requests import HttpRequests
from ddt import ddt, data

test_data = Do_Excel(r‘../test_data/test_data.xlsx‘, ‘form‘).get_data()


@ddt
class TestForm(unittest.TestCase):
    def setUp(self):
        self.headers = {"Content-Type": "application/json",
                        ‘Accept‘: "application/json, text/plain, */*",
                        "passport": login.login()
                        }

    @data(*test_data)
    def test_form_api(self,item):
        res = HttpRequests().http_requests(method=item[‘method‘], url=item[‘url‘],
                                     data=json.loads(item[‘data‘]),
                                     headers=self.headers)
        self.assertEqual(item[‘expect‘], res[‘success‘])

 测试数据：

测试结果：

接口自动化2.0 python+unittest+ddt

原文：https://www.cnblogs.com/thcly/p/11996752.html