5. 接口自动化框架---封装测试用例基类(base_test_case)

"""
抽象出测试用例基类
    1. 提取Excel文件中的用例数据
    2. 自动处理鉴权请求头
    3. 生成一个没有被使用的手机号码
"""
import unittest
import settings

from common.request_handler import send_request
from common.log_handler import logger
from common.db_handler import db
from common.test_data_handler import get_data_from_excel, generate_phone_num


class BaseTestCase(unittest.TestCase):

    name = None   # 功能名称
    logger = logger   # 日志处理器
    db = db   # 数据库处理对象
    auth_key = 'v1'   # 鉴权请求头
    settings = settings   # 配置模块

    @classmethod
    def setUpClass(cls) -> None:
        # 类前置
        cls.logger.info('========={}接口测试开始========='.format(cls.name))

    @classmethod
    def tearDownClass(cls) -> None:
        # 类后置
        cls.logger.info('========={}接口测试结束========='.format(cls.name))

    @classmethod
    def load_case(cls, sheet_name):
        """
        提取Excel文件中的用例数据
        :param sheet_name: 表名
        :return:
        """
        return get_data_from_excel(cls.settings.TEST_DATA_FILE, sheet_name)

    def send_request(self, url, method='GET', **kwargs):
        # 自动处理鉴权请求头
        kwargs.setdefault('headers', {})
        kwargs['headers'].update(self.settings.CUSTOM_HEADERS[self.auth_key])

        if self.auth_key == 'v2':
            pass
        if self.auth_key == 'v3':
            pass

        return send_request(url, method, **kwargs)

    @classmethod
    def get_unused_phone_num(cls, sql_template="select id from table where mobile_phone='{}'"):
        """
        生成一个没有被使用的手机号码
        :param sql_template: SQL模板，用来查询数据库中是否存在指定的电话号码，在下方的方法中使用format风格
        :return: 手机号码
        """
        while True:
            phone_num = generate_phone_num()
            if not cls.db.exist(sql_template.format(phone_num)):
                return phone_num