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())

Str = 'coco愛讀書'

Str = 'coco愛讀書'
print(Str.replace('coco', 'vivi'))

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數(shù)據(jù)
    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#是否在用例數(shù)據(jù)中
        if "#new_phone#" in items['payload']:
            while True:
                # 使用自動生成手機號的函數(shù)
                mobile = generate_mobile()
                # 從數(shù)據(jù)庫中查詢此手機號是否存在
                query_mobile = self.db.query("select * from member where mobile_phone=%s;",args=[mobile])
                # 如果不存在，就跳出循環(huán)
                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('測試數(shù)據(jù)是:{}'.format(items))
        # 訪問注冊接口，獲取實際結(jié)果
        res = self.req.visit(items['method'],config.host+items['url'],
                             json=json.loads(items['payload']))
        # 斷言：預(yù)期結(jié)果與實際結(jié)果對比
        try:
            self.assertEqual(res['code'],items['expected_result'])
            logger.info(res)
            result = 'PASS'
        except AssertionError as e:
            logger.error("測試用例執(zhí)行失敗{}".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()