動態手機號處理

動態手機號處理思路

①編寫函式，生成隨機的手機號；

②將生成的手機號進行資料庫查詢；

③如手機號已存在，就重新生成手機號；

④如手機號不存在，就將此手機號替換測試用例中的手機號。

動態手機號處理

1註冊介面測試用例

在data/cases.xlsx中，新建register工作簿，填充註冊介面用例，其中mobile_phone是動態引數，如果寫死，在自動化過程中，會執行失敗，所以這裡用#new_phone#表示。

2動態生成手機號

在common目錄下，新建檔案helper.py，用於編寫輔助函式，實現特定的功能(類似於HttpRunner中的debugtalk.py)。

實現批量生成11位手機號，程式碼如下：

import random
def generate_mobile():
    """生成隨機手機號"""
    phone = "1" + random.choice(["3","5","7","8","9"])
    for i in range(0,9):
        num = random.randint(1,9)
        phone += str(num)
    return phone
if __name__ == '__main__':
    print(generate_mobile())

執行之後，結果為：

13889546979
 

資料庫查詢並替換

1replace()方法

描述：

replace()方法把字串中的 old（舊字串） 替換成 new(新字串)

2編寫註冊介面用例

接下來的註冊介面用例程式碼，大多數程式碼其實和登入用例一樣，只是新增了查詢資料庫操作。

大致思路如下：

①從excel中讀取用例資料；

②判斷用例資料中是否包含#new_phone#；

③如包含#new_phone#，則隨機生成手機號；

④如隨機生成的手機號在資料庫中存在，則重新生成；

⑤如隨機生成的手機號在資料庫中不存在，則用此手機號替換#new_phone#，進行註冊。

import json
import unittest
from common.db_handler import DBHandler
from common.helper import generate_mobile
from common.logger_handler import logger
from common.requests_handler import RequestHandler
from common.excel_handler import ExcelHandler
from config.setting import config
from libs import ddt
from middleware.yaml_handler import yaml_data
@ddt.ddt
class TestRegister(unittest.TestCase):
    # 讀取register sheet資料
    excel = ExcelHandler(config.data_path)
    data = excel.read_excel('register')
    def setUp(self):
        self.req = RequestHandler()
        self.db = DBHandler(host=yaml_data['mysql']['host'], port=yaml_data['mysql']['port'],
                            user=yaml_data['mysql']['user'], password=yaml_data['mysql']['password'],
                            database=yaml_data['mysql']['db'], charset=yaml_data['mysql']['charset'])
    def tearDown(self):
        self.req.close_session()
        self.db.close()
    @ddt.data(*data)
    def test_register(self,items):
        
        # 判斷#new_phone#是否在用例資料中
        if "#new_phone#" in items['payload']:
            while True:
                # 使用自動生成手機號的函式
                mobile = generate_mobile()
                # 從資料庫中查詢此手機號是否存在
                query_mobile = self.db.query("select * from member where mobile_phone=%s;",args=[mobile])
                # 如果不存在，就跳出迴圈
                if not query_mobile:
                    break
            # 將#new_phone#替換為生成的手機號        
            items['payload'] = items['payload'].replace('#new_phone#', mobile)
        logger.info('*'*30)
        logger.info('測試第{}條測試用例:{}'.format(items['case_id'],items['case_title']))
        logger.info('測試資料是:{}'.format(items))
        # 訪問註冊介面，獲取實際結果
        res = self.req.visit(items['method'],config.host+items['url'],
                             json=json.loads(items['payload']))
        # 斷言：預期結果與實際結果對比
        try:
            self.assertEqual(res['code'],items['expected_result'])
            logger.info(res)
            result = 'PASS'
        except AssertionError as e:
            logger.error("測試用例執行失敗{}".format(e))
            result = 'fail'
            raise e
        finally:
            TestRegister.excel.write_excel(config.data_path,'register',items['case_id']+1,8,res['code'])
            TestRegister.excel.write_excel(config.data_path,'register',items['case_id'] + 1,9, result)
if __name__ == '__main__':
    unittest.main()
 

執行結果為：

{
    'code': 0,
    'msg': 'OK',
    'data': {
        'leave_amount': 240.0,
        'mobile_phone': '155********',
    }
}