python - 接口自动化测试 - TestRegister - 注册接口测试用例

# -*- coding:utf-8 -*-

'''
@project: ApiAutoTest
@author: Jimmy
@file: test_register.py
@ide: PyCharm Community Edition
@time: 2018-12-28 15:50
@blog: https://www.cnblogs.com/gotesting/

'''

import unittest
import os
from ddt import ddt,data
from Common.http_request import HttpRequest
from Common.read_excel import ReadExcel
from Common.read_config import ReadConfig
from Common.mysql_util import MysqlUtil
from Common.get_logger import GetLog
from Common.basic_data import DoRegex,Context
from Common.contants import *
import json
import re

# 读取配置文件，获取当前URL前缀，用于灵活更换测试服务器地址
read_config = ReadConfig()
url_pre = read_config.get_config_str('api','url_pre')

# 读取excel，获取login测试数据
data_dir = os.path.join(data_dir,'test_data.xlsx')
read_excel = ReadExcel(data_dir)
register_cases = read_excel.get_cases('register')

get_log = GetLog()

@ddt
class TestRegister(unittest.TestCase):

    @classmethod
    def setUpClass(cls):
        global max_mobilephone
        mysql = MysqlUtil()
        sql = 'SELECT MobilePhone FROM future.member WHERE MobilePhone != "" ORDER BY MobilePhone DESC LIMIT 1;'
        sql_resp = mysql.fetch_one(sql)
        if sql_resp is not None:
            max_mobilephone = sql_resp['MobilePhone']
            get_log.log_info(' the max_mobilephone from mysql is {0}'.format(max_mobilephone))
        else:
            max_mobilephone = 15964506666
            get_log.log_info(' the max_mobilephone from mysql is None , we use {0}'.format(max_mobilephone))

    @classmethod
    def tearDownClass(cls):
        pass

    @data(*register_cases)
    def test_resister(self,case):
        url = url_pre + case.url

        # 使用正则表达式 匹配 读取的case.data中的mobilephone，匹配成功后，将查询数据库获取的最大mobilephone+1 赋值给 case.data中的mobilephone，以防注册手机号重复
        data = json.loads(case.data)
        register_user = int(max_mobilephone) + 1
        if re.findall(pattern='\$\{(.*?)\}',string=data['mobilephone']):
            data['mobilephone'] = register_user

        # 记录当前测试case信息
        get_log.log_info('''Test Case Info：
        case_id : {0}
        title   : {1}
        method  : {2}
        url     : {3}
        data    : {4}
        expected: {5}
        '''.format(case.case_id,case.title,case.method,url,data,case.expected))

        # 注册接口请求，获取响应
        response = HttpRequest(method=case.method,url=url,data=data)
        actual = response.get_json()['msg']

        # 记录当前测试case 接口响应信息
        get_log.log_info('''Test Case Request Response Result：
        response : {0}
        actual   : {1}
        '''.format(response.get_json(),actual))

        # 接口请求实际结果与期望结果做校验
        try:
            self.assertEquals(case.expected,actual)
            read_excel.write_result('register',case.case_id,actual,'Pass')
            get_log.log_info('Test Result is Passed ! case_id is {0}，title is {1} '.format(case.case_id,case.title))
        except Exception as e:
            read_excel.write_result('register',case.case_id,actual,'Fail')
            get_log.log_info('Test Result is Failed ! case_id is {0}，title is {1} '.format(case.case_id,case.title))
            get_log.log_error('Error msg ：{0}'.format(e))
            raise e

        # 数据库校验
        if actual == '注册成功' or actual == '手机号码已被注册':
            sql_verify = 'SELECT MobilePhone FROM future.member WHERE MobilePhone = {0};'.format(data['mobilephone'])
            sql_result = MysqlUtil().fetch_one(sql_verify)
            if sql_result is not None:
                self.assertEquals(str(data['mobilephone']),sql_result['MobilePhone'])
                get_log.log_info('Test Result is Passed ! case_id is {0}，title is {1} '.format(case.case_id,case.title))
            else:
                get_log.log_info('Test Result is Failed ! case_id is {0}，title is {1} '.format(case.case_id,case.title))
                get_log.log_error('Error msg ：{0}'.format(AssertionError))
                raise AssertionError
        else:
            sql_verify = 'SELECT MobilePhone FROM future.member WHERE MobilePhone = {0};'.format(data['mobilephone'])
            sql_result = MysqlUtil().fetch_one(sql_verify)
            print(sql_result)
            try:
                self.assertEquals(None,sql_result)
                get_log.log_info('Test Result is Passed ! case_id is {0}，title is {1} '.format(case.case_id,case.title))
            except Exception as e:
                get_log.log_info('Test Result is Failed ! case_id is {0}，title is {1} '.format(case.case_id,case.title))
                get_log.log_error('Error msg ：{0}'.format(e))
                raise e