类编程02--测试框架代码具体实现

使用函数与类封装框架

面向对象的编程过程

1.创建类
    构造方法(可以省略,要写的话必须写__init__(self),一般用于定义成员变量)
    成员方法(可以省略,名字自己起，self参数必须写,而且写在第一个,用于处理数据)
2.创建对象
    对象名 = 类名()   -- 实例化
3.调用方法
    对象名.成员方法()     

import logging, time, configparser, pymysql, os, pandas, requests


def log():
    """
    日志函数
    :return:
    """
    # 创建日志对象
    logger = logging.getLogger()
    # 禁止日志重复输出 ，每个日志输出一次
    if not logger.handlers:
        # 指定日志输出级别
        logger.setLevel(logging.INFO)  # 高于INFO的信息都输出到日志
        # 指定日志中的输出格式
        # asctime 当前日志时间
        # 当前日期时间 - 类型 - 文件名[行号] - 消息
        formatter = logging.Formatter('%(asctime)s-%(levelname)s - %(filename)s[%(lineno)d]-%(message)s')
        # 创建日志文件
        log_file_name = '../log/' + time.strftime('%Y%m%d') + '.log'
        # 打开日志文件
        logfile = open(log_file_name, 'wb')  # w:写 ,b:字节
        # 创建处理器
        console = logging.StreamHandler()  # 流处理器  输出控制台
        filehand = logging.FileHandler(log_file_name, encoding='utf-8')  # 文件处理器 输出文件
        # 指定处理器的日志输出格式
        console.setFormatter(formatter)
        filehand.setFormatter(formatter)
        # 增加处理器到日志对象
        logger.addHandler(console)
        logger.addHandler(filehand)
        # 关闭处理器
        console.close()
        filehand.close()
    # 返回日志对象
    return logger


class Entry:  # 入口类 建议类名首字母大写
    def __init__(self):
        """
        构造方法
        :return: 返回值是一个元组(被测接口服务器名,数据库服务器名)
        """
        try:
            # 创建对象
            conf = configparser.ConfigParser()
            # 读取文件
            conf.read('../conf/entry.ini', encoding='utf-8')
            # 读取接口服务器名 -- 通过节点名 找到服务器名字 成员变量
            self.which_server = conf.get('entry', 'which_server')
            # 读取接口数据库服务器名 -- 通过节点名 找到数据库名字 成员变量
            self.which_db = conf.get('entry', 'which_db')
        except Exception as e:
            log().error('读取入口配置文件entry.ini失败' + e)

    def get_which_server(self):  # 成员方法 获得接口服务器方法
        log().info('本次测试的接口服务器是:' + self.which_server)
        return self.which_server

    def get_which_db(self):  # 成员方法 获得数据库名字
        log().info('本次测试的数据库服务器是:' + self.which_db)
        return self.which_db


# 调式检查代码
# e = Entry()  # 创建一个对象,自动执行__init__中的程序代码,括号不能省略
# print(e.which_server, e.which_db)  # 很少这样运行
# print(e.get_which_server(), e.get_which_db())  # 更多的使用方法
# 真正运用时,存储返回值,但是很少写print()
# 使用方法最多是调用成员方法
# server = e.get_which_server()
# db = e.get_which_db()
# print('接口服务器名是:' + server + ',数据库服务器名是:' + db)

class ServerConf:  # 获取接口服务器的地址
    def __init__(self):
        try:
            # 获得要测试的接口服务器的名字
            which_server = Entry().get_which_server()
            conf = configparser.ConfigParser()
            # conf对象中read方法，读取server.conf文件，字符编码设置为utf-8
            conf.read('../conf/server.conf', encoding='utf-8')
            # 根据exam节点获取键名IP所对应的值，赋值给ip变量
            ip = conf.get(which_server, 'IP')
            port = conf.get(which_server, 'port')
            self.host = 'http://' + ip + ':' + port
        except Exception as e:  # Exception是一个关键字,表示所有的异常,e是别名
            log().error('接口服务器地址[server.conf]获取失败' + e)

    def get_host(self):
        log().info('接口服务器地址==' + self.host)
        return self.host


# 调试
# s = ServerConf() # s可以使用host,但是不能使用which_server,conf,ip,port
# print(s.get_host())# 调用成员方法,获得被测接口服务器的url地址

class DB:
    def __init__(self):
        """
        读取db.conf文件,获得数据库服务器的信息
        返回内容是 数据库服务器信息
        """
        try:  # 异常处理
            # 获得测试要用的数据库服务器的节点名
            which_db = Entry().get_which_db()
            conf = configparser.ConfigParser()
            conf.read('../conf/db.conf', encoding='utf-8')
            host = conf.get(which_db, 'host')
            port = conf.get(which_db, 'port')
            user = conf.get(which_db, 'user')
            password = conf.get(which_db, 'password')
            db = conf.get(which_db, 'db')
            self.dbinfo = {'host': host, 'port': int(port), "user": user, 'password': password, 'db': db}
            log().info('数据库信息==' + str(self.dbinfo))
        except Exception as e:
            log().error('数据库配置文件[db.conf]读取出错' + e)

    def conn_db(self):  # 连接数据库的函数
        try:
            # conn = pymysql.connect(host='192.168.175.128',port=3306,user='root',password='123456',db='exam')
            conn = pymysql.connect(**self.dbinfo)  # 连接数据库
            log().info('连接数据库成功')
            return conn
        except Exception as e:
            log().error('数据库连接出错' + e)

    # 调试
    # 创建对象,自动调用DB中的init的函数或方法
    # DB()
    # 调用DB()对象的方法conn_db连接数据库
    # DB().conn_db()
    # 返回为True表示成功
    # print(DB().conn_db().open)

    def get_sqlfiles(sqlf, sqlfiles=[]):  # sqlfiles默认参数,要给参数只能给列表,要么不给
        try:
            if len(sqlfiles) == 0:  # 表示没有给实参,则读取所有sql文件
                sqlfiles = [file for file in os.listdir('../initsqls/') if
                            file.endswith('.sql')]  # 所有.sql文件名存入sqlfiles列表
            sqlfiles = ['../initsqls/' + i for i in sqlfiles]
            log().info('获得sql文件名' + str(sqlfiles) + '列表成功')
            return sqlfiles
        except Exception as e:
            log().error('获得sql文件名列表失败')


# 调试
# 只写文件名 不用写路径 程序自动加路径
# print(DB().get_sqlfiles(['login.sql']))

# 读取带.sql文件扩展名文件中的sql语句
    def read_sqls(self, sqlfiles=[]):
        sqlfiles = self.get_sqlfiles(sqlfiles)  # 获得带路径的文件名列表
        sqls = []  # 存sql语句的列表
        try:
            for sqlfile in sqlfiles:  # sqlfile表示每一个sql语句文件
                sfile = open(sqlfile, 'r', encoding='utf-8')  # 打开文件
                for sql in sfile:  # sql是从sfile文件中获得每一行
                    # 如果这行字符长度大于0 并且 不是以--开头的
                    if len(sql.strip()) > 0 and not sql.startswith('--'):
                        sqls.append(sql.strip())  # sql语句中的回车不存入列表
            log().info('读取初始化sql语句成功==' + str(sqlfiles))
            return sqls
        except Exception as e:
            log().error('读取初始化sql语句出错' + e)


# 调试
# print(DB().read_sqls(['login.sql']))
# print(DB().read_sqls())

    def init_db(self,sqlfiles=[]):  # #执行sqlfiles文件中的sql语句,初始化数据库
        """
        目的:数据初始化
        :param sqlfiles: 指定sql文件
        :return:
        """
        sqls = self.read_sqls(sqlfiles)  # 读取所有需要执行的sql命令
        conn = self.conn_db()  # 连接数据库
        cursor = conn.cursor()  # 创建游标
        try:
            for sql in sqls:
                cursor.execute(sql)  # 只能一条一条的执行,结果暂存到cursor中
            conn.commit()  # 提交数据到数据库
            conn.close()
            log().info('初始化数据库成功')
        except Exception as e:
            log().error('初始化数据库出错')

# 调试
# DB().init_db(['login.sql'])
# DB().init_db()
    def check_db(self,case_name,expectsql,argument,expect_db_rows):
        """
        :param case_name: 用例名称
        :param expectsql: 预期sql语句
        :param argument: 接口参数
        :param expect_db_rows: 预期行数
        :return:
        """
        conn = self.conn_db()  # 连接数据库
        try:
            cursor = conn.cursor()  # 创建游标
            cursor.execute(expectsql)  # 执行sql语句
            dbactual = cursor.fetchone()[0]  # 取第一行第一列数据
            if dbactual == expect_db_rows:
                log().info(case_name + '==落库检查==通过')
            else:
                log().warning(case_name + '==落库检查==失败==预期行数:' + str(expect_db_rows) + ',实际行数:' + str(dbactual))
        except Exception as e:
            log().error('落库检查出错' + e)

# 调试
DB().check_db('测试','select count(*) from user',{'username':'test01'},4)

def read_cases(excel, columns=[]):  # 读取excel用例中的指定列
    excel = '../excelcase/' + excel
    try:
        if len(columns) == 0:
            file = pandas.read_excel(excel)  # 读取所有列
        else:
            file = pandas.read_excel(excel, usecols=columns)  # 读取指定列
        data = file.values.tolist()  # 文件数据转为列表
        log().info('读取用例文件' + excel + '成功')
        return data
    except Exception as e:
        log().error('读取测试用例文件出错')


def post(address, argument, case_name, expect):
    """
    :param address: 接口地址
    :param argument: 请求参数
    :param case_name: 用例名称
    :param expect: 预期结果
    :return:
    """
    try:
        # 发送请求 判断响应结果的正确性
        res = requests.post(url=address, data=argument)
        if 'text/html' in res.headers['Content-Type']:  # 表示响应内容的类型是字符串
            actual = res.text
            # 结果比对
            if expect in actual:
                log().info('比对登录接口返回==' + case_name + '==通过')
            else:
                log().warning('比对登录接口返回==' + case_name + '==失败==预期结果:' + expect + ',实际结果:' + actual)
        elif 'application/json' in res.headers['Content-Type']:
            actual = res.json()  # 实际结果(字典)
            # 比对接口返回结果
            if expect == actual:
                log().info("比对注册接口返回==" + case_name + '==通过')
            else:
                log().warning('比对注册接口返回==' + case_name + '==失败==预期结果:' + str(expect) + ',实际结果:' + str(actual))
        else:  # xml
            pass  # 暂时不写代码
    except Exception as e:
        log().error('发送数据，比对响应结果出错')


def test_login():
    """
    测试登录接口
    :return:
    """
    # 数据库初始化
    DB().init_db(['login.sql'])
    # 读测试用例数据
    cases = read_cases('login.xlsx')  # 用例:用例名0、参数数据1、预期结果2
    # 读接口服务器地址
    host = ServerConf().get_host()
    address = host + '/exam/login/'
    try:
        # 发送请求
        for case in cases:  # case代表一行用例
            case_name = case[0]  # 第0列,是用例名称
            argument = eval(case[1])  # 要发送的参数
            expect = case[2]  # 预期接口的返回结果
            post(address, argument, case_name, expect)
    except Exception as e:
        log().error('登录接口测试出错' + e)


# 调试
# test_login()


def test_signup():
    """
    测试注册接口
    :return:
    """
    # 数据库初始化
    DB().init_db(['signup.sql'])
    # 读测试用例
    cases = read_cases('signup.xlsx')
    # 接口地址
    host = ServerConf().get_host()
    address = host + '/exam/signup/'
    try:
        # 发送请求
        for case in cases:
            case_name = case[0]  # 用例名称
            argument = eval(case[1])  # 参数
            expect = eval(case[2])  # 预期接口返回结果
            sql = case[3]  # 预期sql
            dbexpect = case[4]  # 预期数据库行数
            # 发送请求
            post(address, argument, case_name, expect)
            # 落库检查
            DB().check_db(case_name,sql,argument,dbexpect)
    except Exception as e:
        log().error('注册接口测试出错' + e)


# 调试
test_signup()