"""
send_method.py 文件说明：
1，封装接口请求方式
    根据项目接口文档提供的内容进行封装
    不同的项目，sendmethod也不太一样，如请求体格式等。
2.封装思路-结合接口三要素
    请求方式+请求地址
    请求参数
    返回值
3.以学生管理系统SMS为例:
    结合学生管理系统项目的接口文档，封装SendMethod类

"""
# 导入所需模块
import requests
import json


# 封装请求模块
class SendMethod:
    """
        结合学生管理系统SMS，请求方式包括如下:
            get ---> parmas标准请求参数
            post--->请求参数类型 json
            put --->请求参数类型 json
            delete ---> parmas标准请求参数
    """

    # 定义该方法为静态方法
    @staticmethod
    def send_method(method, url, parmas=None, json=None):
        """
        封装适用于学生管理系统项目的接口请求
        :param method: 请求方式
        :param url: 请求地址
        :param parmas: get和delete请求参数
        :param json: post和put请求参数
        :param headers: 请求头
        :return:
        """
        # 定义发送请求的方法
        if method == "get" or method == "delete":
            response = requests.request(method=method, url=url, params=parmas)
        elif method == "post" or method == "put":
            response = requests.request(method=method, url=url, json=json)
            # 如果有不同的请求头，还可以继续添加接收的参数
            # response = requests.request(method=method, url=url, json=json, data=data, files=data)
        else:
            # 这里是简单处理，完成封装需要加上异常处理。
            response = None
            print("请求方式不正确")

        # 如果请求方式是delete，只返回状态码
        # 这是根据项目接口文档中delete方法的返回规则定的。
        if method == "delete":
            return response.status_code
        else:
            # 项目中接口的返回值是json格式的，就可以用json()进行格式化返回结果。
            return response.json()

    @staticmethod
    def json_2_python(res):
        """
        格式化返回数据
        :param res:接口返回的数据
        :return:
        """
        return json.dumps(res, indent=2, ensure_ascii=False)


if __name__ == '__main__':
    method = "post"
    url = "http://127.0.0.1:8000/api/departments/"
    data = {
        "data": [
            {
                "dep_id": "T02",
                "dep_name": "接口测试学院",
                "master_name": "Test-Master",
                "slogan": "Here is Slogan"
            }
        ]
    }
    res = SendMethod.send_method(method=method, url=url, json=data)
    # print(res)
    print(SendMethod.json_2_python(res))

    # method = "get"
    # params = {"$dep_id_list": "1, 2, 3"}
    # res = SendMethod.send_method(method=method, url=url, json=data)
    # print(SendMethod.json_2_python(res))

3、封装获取接口返回结果指定内容

该文件是封装处理返回值结果的一些方法。

我们需要用到一个Python中的模块JsonPath ，下面就先来介绍一下JsonPath 模块。

（1）JsonPath 介绍

用来解析多层嵌套的Json数据。

JsonPath是一种信息抽取类库，是从JSON文档中抽取指定信息的工具，提供多种语言实现版本，包括：Javascript，Python， PHP 和 Java。

JsonPath 对于 JSON 来说，相当于 XPath 对于 XML。

（2）JsonPath 安装

安装方法：pip install jsonpath

使用方法如下：

# 导入jsonpath模块
import jsonpath模块

# 嵌套n层也能取到所有key_nane信息，
# 其中："$"表示最外层的{}，
# ".."表示模糊匹配，
# 当传入不存在的key_nane时，程序会返回false。

res = jsonpath.jsonpath(response, f"$..{keyword}")[0]

"""
jsonpath方法说明
jsonpath(obj, expr, result_type='VALUE', debug=0, use_eval=True):

# obj表是要处理的json对象。
# expr是jsonpath匹配表达式。$..{keyword} 这种方式比较通用
"""

"""
getKeyword_forResult.py文件说明：
1.作用
    在接口返回值中，通过关键获取获取对应字段内容
2，前提：需要安装一个库：jsonpath库
    安装jsonpath ： pip install jsonpath
    使用jsonpath模块进行处理更加方便

"""
# 导入jsonpath模块
import jsonpath


# 封装获取接口返回值方法
class GetKeyword:
    # 定义成一个静态方法
    @staticmethod
    def get_keyword(response: dict, keyword):
        """
        通过关键字获取对应返回值,如果有多个值,只返回第一个,
        如果关键字不存在,返回False。
        :param:response 数据源  字典格式
        :param:keyword 要获取的字段
        :return:
        """
        try:
            return jsonpath.jsonpath(response, f"$..{keyword}")[0]
        except:
            print("关键字不存在")

    @staticmethod
    def get_keywords(response: dict, keyword):
        """
        通过关键字获取一组数据
        :param response: 数据源 dict格式
        :param keyword:  如果关键字不存在,返回False
        :return:
        """
        try:
            return jsonpath.jsonpath(response, f"$..{keyword}")
        except:
            print("关键字不存在")


if __name__ == '__main__':
    response = {
        "count": 2,
        "next": "下一页",
        "previous": None,
        "results": [
            {
                "dep_id": "10",
                "dep_name": "tester_10",
                "master_name": "master_10",
                "slogan": "随便"
            },
            {
                "dep_id": "11",
                "dep_name": "tester_11",
                "master_name": "master_11",
                "slogan": "随便"
            }
        ]
    }
    keyword = "dep_id"
    # print(GetKeyword.get_keyword(response, keyword))
    print(GetKeyword.get_keywords(response, keyword))

"""
新增学院接口
1.单接口测试方法
2.关联接口测试方法
    获取返回值中的字段
"""
# 导入自定义的公共方法
from common.send_method import SendMethod
from common.getKeyword_forResult import GetKeyword


# 封装新增学院接口测试
class Add_department:

    # url和请求方式对于一个接口来说是固定的，
    # 所以这两个参数可以写在初始化方法中。
    def __init__(self, url, method="post"):
        self.method = method
        self.url = url

    def add_dep(self, data):
        """
        定义新增学院接口：针对单接口测试
        :param data: 新增学院的请求参数
        :return:
        """
        return SendMethod.send_method(self.method, url=self.url, json=data)

    def get_keyword(self, data, keyword):
        """
        获取新增成功后的关键字值：为关联接口测试准备
        :param data:
        :param keyword:
        :return:
        """
        res = self.add_dep(data)

        # 获取新增学院接口返回值中的学院的具体某一属性
        return GetKeyword.get_keyword(res, keyword)


if __name__ == '__main__':
    url = "http://127.0.0.1:8000/api/departments/"
    data = {
        "data": [
            {
                "dep_id": "T03",
                "dep_name": "Test学院",
                "master_name": "Test-Master",
                "slogan": "Here is Slogan"
            }
        ]
    }
    add = Add_department(url)
    res = add.add_dep(data)  # 新增学院接口方法
    print(res)
    keyword = "dep_id"
    dep_id = add.get_keyword(data, keyword)  # 获取新增成功后depid
    print(dep_id)

"""
get_dep.py文件说明：
1.查询接口测试
2.获取查询接口返回值
"""
from common.send_method import SendMethod


class Get_Departments:
    def __init__(self, url, method="get"):
        self.url = url
        self.method = method

    def get_departments(self):
        """
        查询所有学院
        :return:
        """
        return SendMethod.send_method(method=self.method, url=self.url)

    def get_department(self, dep_id):
        """
        根据id查询单个学院
        :return:
        """
        url = self.url + f"{dep_id}/"
        return SendMethod.send_method(method=self.method, url=url)

    def get_department_for_multpart(self, data):
        """
        根据参数查询学院
        :return:
        """
        return SendMethod.send_method(method=self.method, url=self.url, parmas=data)


if __name__ == '__main__':
    url_1 = "http://127.0.0.1:8000/api/departments/"
    data = {"$dep_id_list": "12,13"}
    get_dep = Get_Departments(url=url_1)
    # 查询所有学院
    # print(get_dep.get_departments())
    # 查询指定学院
    dep_id = 16
    # print(get_dep.get_department(dep_id))
    # 根据条件查询学院
    print(get_dep.get_department_for_multpart(data))

"""
测试新增学院接口
"""

# 测试用例是在unittest框架下编写
import unittest
from interface.add_departments import Add_department  # 导入新增学院接口
from common.getKeyword_forResult import GetKeyword  # 返回值处理接口


# 测试添加和查询学院的关联型接口
class Test_Add_Dep(unittest.TestCase):
    def setUp(self) -> None:
        self.url = "http://127.0.0.1:8000/api/departments/"
        # 实例化Add_department
        self.add_dep = Add_department(self.url)

    # 开始编写测试用例
    def test_add_dep_success(self):
        """
        测试添加学院成功接口
        :return:
        """

        # 封装请求参数
        data = {
            "data": [
                {
                    "dep_id": "T100",
                    "dep_name": "Test学院",
                    "master_name": "Test-Master",
                    "slogan": "Here is Slogan"
                }
            ]
        }

        # 新增学院
        response = self.add_dep.add_dep(data)
        # 获取添加成功后的dep.id
        """
        # 因为直接使用该方法相当于又执行了一次添加学院接口
        # 所以不能够这样调用
        self.add_dep.get_depid(data)
        """
        res_dep_id = GetKeyword.get_keyword(response["create_success"], "dep_id")
        expect = "T100"
        self.assertEqual(res_dep_id, expect)

    # 测试添加学院完整性实现
    def test_add_dep(self):
        """
        测试添加学院接口
        :return:
        """

        # 封装请求参数
        data = {
            "data": [
                {
                    "dep_id": "T101",
                    "dep_name": "Test学院",
                    "master_name": "Test-Master",
                    "slogan": "Here is Slogan"
                }
            ]
        }
        # 新增学院
        response = self.add_dep.add_dep(data)

        """
        并返回值的验证有3种情况
            #1.添加成功
            #2.添川id已存在的学院
            #3.参敖错误(自己实现)
        根据对接口档的分析
            可以通过判断返回值是否包含“status_code”区分1,2和3，然后区分1,2
            根据返回值中already_exist.count是否为0，判断是否添加成功
        """

        # 这里只判断上面的1,2情况，工作中根据实际业务自己在完成
        if GetKeyword.get_keyword(response["already_exist"], "count") == 0:
            # 获取添加成功后的dep.id
            res_dep_id = GetKeyword.get_keyword(response["create_success"], "dep_id")
        else:
            res_dep_id = GetKeyword.get_keyword(response["already_exist"], "dep_id")

        expect = "T101"
        self.assertEqual(res_dep_id, expect)


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

"""
测试新增和查询接口(组合接口业务)
    先新增--->再查询
"""
# 测试用例是在unittest框架下编写
import unittest
from interface.add_departments import Add_department  # 导入新增学院接口
from interface.get_departments import Get_Departments  # 查询学院接口
from common.getKeyword_forResult import GetKeyword  # 返回值处理接口


# 测试添加和查询学院的关联型接口
class Test_Add_Get_Dep(unittest.TestCase):
    def setUp(self) -> None:
        self.url = "http://127.0.0.1:8000/api/departments/"
        # 实例化Add_department添加学院
        self.add_dep = Add_department(self.url)
        # 实例化Get_Departments查询学院
        self.get_dep = Get_Departments(self.url)

    # 开始编写测试用例
    def test_add_get(self):
        # 封装请求参数
        add_data = {
            "data": [
                {
                    "dep_id": "T03",
                    "dep_name": "Test学院",
                    "master_name": "Test-Master",
                    "slogan": "Here is Slogan"
                }
            ]
        }

        # 一下逻辑待查证，知道组合的形式即可。
        # 执行添加学院接口。目的：获取添加成功后的学院id
        # 获取新增学院后的id
        dep_id = self.add_dep.get_keyword(add_data, "dep_id")
        # 查询新增学院信息
        result = self.get_dep.get_department(dep_id)
        # 通过获取查询后的学院id作为实际结果
        res_dep_id = GetKeyword.get_keyword(result, "dep_id")
        # 获取预期结果id
        expect = GetKeyword.get_keyword(add_data, "dep_id")
        # 断言结果
        self.assertEqual(expect, res_dep_id)


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

import xlrd
from xlrd import xldate_as_tuple
from datetime import datetime

class OperationExcel:
    def __init__(self, filepath):
        book = xlrd.open_workbook(filename=filepath)
        self.sheet = book.sheet_by_index(0)

    def read_excel(self):
        rows = self.sheet.nrows
        cols = self.sheet.ncols
        all_data_list = []
        for row in range(1, rows):
            data_list = []
            for col in range(cols):
                ctype = self.sheet.cell(row, col).ctype
                cell = self.sheet.cell_value(row, col)
                if ctype == 2 and cell % 1 == 0:
                    cell = int(cell)
                elif ctype == 3:
                    date = datetime(*xldate_as_tuple(cell, 0))
                    cell = date.strftime("%Y-%m-d %H-%M-%S")
                elif ctype == 4:
                    cell = True if cell == 1 else False  # 三目云算法
                data_list.append(cell)
            all_data_list.append(data_list)
        return all_data_list

    def get_data_by_dict(self):
        keys = self.sheet.row_values(0)
        values = self.read_excel()
        data_list = []
        for value in values:
            tmp = zip(keys, value)
            data_list.append(dict(tmp))
        return data_list


if __name__ == '__main__':
    oper = OperationExcel('testdata.xlsx')
    # data = oper.read_excel()
    data = oper.get_data_by_dict()
    print(data)

"""
新增学院接口测试--批量新增
"""

# 测试用例是在unittest框架下编写
import unittest
from interface.add_departments import Add_department  # 导入新增学院接口
from common.getKeyword_forResult import GetKeyword  # 返回值处理接口
# 步骤1：导入OperationExcel数据读取脚本和ddt模块
from common.opreationexcel import OperationExcel
import ddt

# 步骤2：对OperationExcel进行实例化
# 获得文件对象
oper = OperationExcel("../data/add_dep.xls")
# 获取数据
test_data = oper.get_data_by_dict()


# 测试添加和查询学院的关联型接口
# 步骤3
@ddt.ddt()
class Test_Add_Dep(unittest.TestCase):
    def setUp(self) -> None:
        self.url = "http://127.0.0.1:8000/api/departments/"
        # 实例化Add_department
        self.add_dep = Add_department(self.url)

    # 开始编写测试用例
    # 步骤4
    @ddt.data(*test_data)
    def test_add_dep_success(self, data):  # 步骤5：出入data参数
        """
        测试添加学院成功接口
        :return:
        """
        # 步骤6：准备数据
        req_data = {
            "data": [
                {
                    "dep_id": data["dep_id"],
                    "dep_name": data["dep_name"],
                    "master_name": data["master_name"],
                    "slogan": data["slogan"]
                }
            ]
        }

        # 新增学院
        response = self.add_dep.add_dep(req_data)
        # 获取添加成功后的dep.id

        # 步骤7：完成测试逻辑
        # 如果添加学院参数请求错误，会出现status_code属性
        # 且status_code属性返回400
        if "status_code" in response.keys():
            res = GetKeyword.get_keyword(response, "status_code")
        else:
            # 添加学院成功，则获取添加后学院的id
            """
            # 因为直接使用该方法相当于又执行了一次添加学院接口
            # 所以不能够这样调用
            self.add_dep.get_depid(data)
            """
            res = GetKeyword.get_keyword(response["create_success"], "dep_id")

        # 断言
        self.assertEqual(res, data["expect"])


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