从数据库获取最大手机号+1注册新用户

Posted 2020-11-27 sophia-985935365

import unittest
from common.get_request import GetRequest
from common.read_data import ReadData
from ddt import ddt, data
from common import constant
from common.read_mysql import ReadMysql
import json


@ddt
class TestRegister(unittest.TestCase):
    """注册"""
    get_request = GetRequest()
    read_data = ReadData(constant.data_file)
    register_data = read_data.read_data(‘register‘)
    read_mysql = ReadMysql()


    @data(*register_data)
    def test_register(self, case):
        sql = ‘select max(MobilePhone) from member where MobilePhone like "185112958%"‘
        mobilephone = self.read_mysql.fetch_one(sql)[0]    # 从数据库取出来是字符串


        dict_data = json.loads(case.data)                  # 字符串转字典
        if dict_data[‘mobilephone‘] == ‘${mobilephone}‘:   # 从excel取出来是字符串
            dict_data[‘mobilephone‘] = int(mobilephone)+1  # 从数据库获取最大手机号+1注册新用户


        resp = self.get_request.get_request(case.method, case.url, dict_data)
        try:
            self.assertTrue(case.expected == resp.json()[‘msg‘])
            case.result = ‘SUCCESS‘
        except AssertionError as e:
            case.result = ‘FAILED‘
            print(‘断言失败{}‘.format(e))
        finally:
            self.read_data.write_back(case.case_id + 1, 7, str(resp.json())) #excel只支持字符串格式存入，不支持json格式的数据
            self.read_data.write_back(case.case_id + 1, 8, case.result)

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