Python自动生成测试数据:Faker库的使用

Python013

Python自动生成测试数据:Faker库的使用,第1张

在进行测试时,经常需要使用一些测试数据。如果手动创建数据,就会花费大量的时间,使用faker模块就可以帮助我们自动创建随机模拟数据。

faker是一个开源的python库,安装完成后只需要调用Facker库,就可以帮助我们创建需要的数据。

源码地址: https://github.com/joke2k/faker

faker默认使用的是英语,可以通过设置地域更改语言。

可支持设置的语言如下:

比如设置成中文:f = Faker("zh_CN")

Faker把Provider对象做成了Faker对象的“插件”,Faker通过内置的Provider对象,实现了生成某项数据的能力。

可以通过查看源码内的Provider查看可以哪些方法 https://github.com/joke2k/faker/tree/master/faker/providers

date_time

用于生成时间相关数据,用法如下:

Lorem

生成假文字数据。用法如下:

Address

生成和地址相关的数据

company

生成公司信息

person

生成名字等

生成数字类信息

生成随机集合类型

设置seed

设置seed可以保证每次取到的数字是一样的,添加Faker.seed()实现,用法如下:

测试过程中,经常需要批量去造数据,方法有很多,最简单方便的应该是使用python 的一个三方库Faker。

脚本作用,批量生成10000条基本数据,写入本地的文件:test_data.csv  

在linux服务器上执行,把test_data.csv 拿到本地windows,excel打开可能中文乱码,可使用:数据--从文本/CSV 导入数据,修改下编码格式,就能正常显示中文

以下为详细脚本:

import csv

from faker import Faker

import datetime

fake = Faker(['zh_CN'])

file = open("test_data.csv","w",newline="")

# 创建文件,分别是文件名、w打开方式(w代表新建,如果已存在,就删除重写)、newline(如果不加,每行数据就会多一空白行)

fwrite = csv.writer(file)

# 获取写文件的对象

fwrite.writerow(["name", "phone", "Card_id", "公司", "地址", "信用卡", "职位", "email"])

# 写入标题头

for i in range(9999):

    user_name = fake.name()

    phone = fake.phone_number()

    card_id = fake.ssn()

    company = fake.company()

    addr = fake.address()

    bank_card = fake.credit_card_number()

    title = fake.job()

    email = fake.email()

    fwrite.writerow([user_name, phone, card_id, company, addr, bank_card, title, email])

# 写入一行一行的数据

file.close()

保存为makedata.py

注意,需要先安装python3和faker,如下

#yum install python3.x86_64

#pip3 install faker

#coding=utf-8

import sys

reload(sys)

sys.setdefaultencoding('utf8')

import requests

import xlrd

import os

import time

from seleniumimport webdriver

from selenium.common.exceptionsimport *

class Runmain:

def run_main(self,url,data,params,headers):

data = xlrd.open_workbook(os.getcwd() +"\\login.xlsx")#打开步骤一中的表格

table = data.sheet_by_name("Sheet1")#选取表格中的某一页,如图所示

rows=table.nrows#计算出该表中有多少行数据

        print ('行数:'+str(rows))

for i in range(1,rows):#遍历每一行数据,除去第一行

            tabledata = table.row_values(i)#计算出第i行数据

            username,password,method,loginurl= tabledata[0],tabledata[1],tabledata[2],tabledata[3]#找出第i行索引值

            print username,',',int(password),',',method,',',loginurl

response =None

            if method =='GET':

# response=self.send_get(url,params,headers)

                print ("method:GET")

else:

print ("method:POST")

# response=self.send_post(url,data,headers)

            driver=webdriver.Firefox()

driver.get(loginurl)

urllogonbefore=driver.current_url

print ("urllogonbefore:",urllogonbefore)

driver.find_element_by_xpath(".//*[@id='username']").send_keys(username)

driver.find_element_by_xpath(".//*[@id='password']").send_keys(int(password))

driver.find_element_by_xpath(".//*[@id='content']/div/section/div/div/div[2]/div/form/div[3]/div/button").click()

time.sleep(2)

try:

if driver.switch_to_alert():

#driver.switch_to_alert().accept()#接受警告框,点击警告框中的确定按钮

                  ErrorMsg=driver.switch_to_alert().text#获取警告框弹出内容

                  print ('ErrorMsg:',ErrorMsg)

print (ErrorMsg[0:5])#打印警告框弹出内容,用分片进行操作

                  print  ("登录账号或密码错误!")

driver.quit()

except NoAlertPresentExceptionas e:#如果没有抛出"NoAlertPresentException"异常,则表示没有弹出警告框,登录成功

                print ('login success,because no ',e,"see!!!")

urlloginend = driver.current_url

print ("urlloginend:", urlloginend)

driver.quit()

Rmain=Runmain()

Rmain.run_main(1,2,3,4)