怎樣從0開始搭建一個測試框架_10——數據生成器

怎樣從0開始搭建一個測試框架_10

有時候接口或UI上傳入的數據需要符合指定的格式，我們在參數化的過程中又不願意在excel中一遍遍去構造這樣的數據，這時我們可以加入生成器來爲我們產生符合某些固定格式的數據。
這裏我推薦一個挺有意思的庫，Faker，能夠爲你產生各種假數據

別忘了在ReadMe.md中添上你要用的庫。

在utils中創建一個generator.py，用來生成數據

"""一些生成器方法，生成隨機數，手機號，以及連續數字等"""
import random
from faker import Factory

fake = Factory().create('zh_CN')


def random_phone_number():
    """隨機手機號"""
    return fake.phone_number()


def random_name():
    """隨機姓名"""
    return fake.name()


def random_address():
    """隨機地址"""
    return fake.address()


def random_email():
    """隨機email"""
    return fake.email()


def random_ipv4():
    """隨機IPV4地址"""
    return fake.ipv4()


def random_str(min_chars=0, max_chars=8):
    """長度在最大值與最小值之間的隨機字符串"""
    return fake.pystr(min_chars=min_chars, max_chars=max_chars)


def factory_generate_ids(starting_id=1, increment=1):
    """ 返回一個生成器函數，調用這個函數產生生成器，從starting_id開始，步長爲increment。 """
    def generate_started_ids():
        val = starting_id
        local_increment = increment
        while True:
            yield val
            val += local_increment
    return generate_started_ids


def factory_choice_generator(values):
    """ 返回一個生成器函數，調用這個函數產生生成器，從給定的list中隨機取一項。 """
    def choice_generator():
        my_list = list(values)
        # rand = random.Random()
        while True:
            yield random.choice(my_list)
    return choice_generator


if __name__ == '__main__':
    print(random_phone_number())
    print(random_name())
    print(random_address())
    print(random_email())
    print(random_ipv4())
    print(random_str(min_chars=6, max_chars=8))
    id_gen = factory_generate_ids(starting_id=0, increment=2)()
    for i in range(5):
        print(next(id_gen))

    choices = ['John', 'Sam', 'Lily', 'Rose']
    choice_gen = factory_choice_generator(choices)()
    for i in range(5):
        print(next(choice_gen))

你還可以添加各種各樣的生成器，比如指定長度中文、英文、特殊字符的字符串，指定格式的json串等等，可以省去很多構造測試數據的煩惱。

所有的代碼我都放到了GITHUB上傳送，可以自己下載去學習，有什麼好的建議或者問題，可以留言或者加我的QQ羣:455478219討論。