pytest接口自動化測試

pytest接口自動化測試

結合單元測試框架pytest+數據驅動模型+allure

目錄

api： 存儲測試接口

conftest.py :設置前置操作

             目前前置操作：1、獲取token並傳入headers，2、獲取命令行參數給到環境變量,指定運行環境

commmon：存儲封裝的公共方法

         connect_mysql.py：連接數據庫
         http_requests.py: 封裝自己的請求方法
         logger.py: 封裝輸出日志文件
         read_yaml.py：讀取yaml文件測試用例數據
         read_save_data.py：讀取保存的數據文件

case: 存放所有的測試用例
      
data：存放測試需要的數據
      save_data: 存放接口返回數據、接口下載文件
      test_data: 存放測試用例依賴數據
      upload_data: 存放上傳接口文件

logs: 存放輸出的日志文件

report: 存放測試輸出報告

getpathinfo.py :封裝項目測試路徑

pytest.int :配置文件

requirement.txt: 本地python包（pip install -r requirements.txt 安裝項目中所有python包）

run_main.py: 項目運行文件

結構設計

1.每一個接口用例組合在一個測試類里面生成一個py文件

2.將每個用例調用的接口封裝在一個測試類里面生成一個py文件

3.將測試數據存放在yml文件中通過parametrize進行參數化，實現數據驅動

4.通過allure生成測試報告

代碼展示

api/api_service.py　#需要測試的一類接口

'''
Code description:服務相關接口
Create time: 2020/12/3
Developer: 葉修
'''
import os
from common.http_requests import HttpRequests


class Api_Auth_Service(object):

    def __init__(self):
        self.headers = HttpRequests().headers

    def api_home_service_list(self):
        # 首頁服務列表
        # url = "http://192.168.2.199:9092/v1/auth/auth_service/findAuthService"
        url = os.environ["host"] + "/v1/auth/auth_service/findAuthService"  # 讀取conftest.py文件地址進行拼接
        response = HttpRequests().get(url, headers=self.headers, verify=False)
        # print(response.json())
        return response

    def get_service_id(self):
        #獲取銀行卡三要素認證服務id
        url = "http://192.168.2.199:9092/v1/auth/auth_service/findAuthService"
        #url = os.environ["host"] + "/v1/auth/auth_service/findAuthService"  # 讀取conftest.py文件地址進行拼接
        response = HttpRequests().get(url,headers=self.headers)
        #print(response.json()['data'][0]['service_list'][0]['id'])
        service_id = response.json()['data'][0]['service_list'][1]['id']
        return service_id

    def api_service_info(self,serviceId='0b6cf45bec757afa7ee7209d30012ce1',developerId=''):
        #服務詳情
        body = {
            "serviceId" :serviceId,
            "developerId":developerId
        }
        url = "http://192.168.2.199:9092/v1/auth/auth_service/findServiceDetail"
        #url = os.environ["host"] + "/v1/auth/auth_service/findServiceDetail"#讀取conftest.py文件地址進行拼接
        response = HttpRequests().get(url,headers=self.headers,params = body,verify=False)
        #print(response.json())
        return response

    def api_add_or_update_service(self,api_param_req,api_param_res,description,error_code,icon,id,interface_remarks,
                              name,product_info,request_method,sample_code,sort,type,url):
        #服務添加或者更新
        body={
                "api_param_req": api_param_req,
                "api_param_res": api_param_res,
                "description": description,
                "error_code": error_code,
                "icon": icon,
                "id": id,
                "interface_remarks": interface_remarks,
                "name": name,
                "product_info": product_info,
                "request_method": request_method,
                "sample_code": sample_code,
                "sort": sort,
                "type": type,
                "url": url,
        }
        #url = "http://192.168.2.199:9092/v1/auth/auth_service/insertOrUpdateService"
        url = os.environ["host"] + "/v1/auth/auth_service/insertOrUpdateService"  # 讀取conftest.py文件地址進行拼接
        response = HttpRequests().post(url,json=body,headers=self.headers,verify=False)
        return response

    def api_add_service_price(self,id,max_number,money,service_id,small_number):
        #服務價格添加
        body = {
            "id": id,
            "max_number": max_number,
            "money": money,
            "service_id": service_id,
            "small_number": small_number
        }
        # url = "http://192.168.2.199:9092/v1/auth/auth_service/insertServicePrice"
        url = os.environ["host"] + "/v1/auth/auth_service/insertServicePrice"  # 讀取conftest.py文件地址進行拼接
        response = HttpRequests().post(url, json=body, headers=self.headers, verify=False)
        return response

    def api_apply_service(self,developer_id,service_id):
        #申請服務
        body ={
            "developer_id": developer_id,
            "service_id": service_id
        }
        # url = "http://192.168.2.199:9092/v1/auth/auth_service/applyService"
        url = os.environ["host"] + "/v1/auth/auth_service/applyService"  # 讀取conftest.py文件地址進行拼接
        response = HttpRequests().post(url, json=body, headers=self.headers, verify=False)
        return response


if __name__ == '__main__':
    #Auth_Service().api_home_service_list()
    Api_Auth_Service().get_service_id()
    #Auth_Service().api_service_info()

 api/get_token.py#獲取登錄token

'''
Code description：獲取token
Create time：2020-12-03
Developer：葉修
'''
import os

import urllib3
from common.http_requests import HttpRequests


class Get_Token(object):

    def get_token(self,account='****',password='****'):
        #url = "http://192.168.2.199:9092/v1/auth/developer/accountLogin"
        url = os.environ["host"]+"/v1/auth/developer/accountLogin"
        body = {
            "account": account,
            "password": password,
        }
        urllib3.disable_warnings()
        r = HttpRequests().post(url, json=body,verify=False)
        #print(r.json())
        token = r.json()['data']['token']
        params = {
            "access_token": token
        }
        HttpRequests().params.update(params)#更新token到session
        return token

if __name__ == '__main__':
    print(Get_Token().get_token())

case/test_service_info.py #上面接口某一測試用例

'''
Code description: 服務詳情
Create time: 2020/12/3
Developer: 葉修
'''
import sys
import allure
import pytest
from common.logger import Log
from common.read_yaml import ReadYaml
from api.api_auth_service.api_auth_service import Api_Auth_Service

testdata = ReadYaml("auth_service.yml").get_yaml_data()  # 讀取數據


@allure.feature('服務詳情')
class Test_Service_Info(object):
    log = Log()

    @pytest.mark.process
    @pytest.mark.parametrize('serviceId,developerId,expect', testdata['service_info'],
                             ids=['服務詳情'])
    def test_service_info(self,serviceId,developerId,expect):
        self.log.info('%s{%s}' % ((sys._getframe().f_code.co_name,'------服務詳情接口-----')))
        with allure.step('獲取服務id'):
            serviceId = Api_Auth_Service().get_service_id()
        with allure.step('服務詳情'):
            msg = Api_Auth_Service().api_service_info(serviceId,developerId)
        self.log.info('%s:%s' % ((sys._getframe().f_code.co_name, '獲取請求結果：%s' % msg.json())))
        # 斷言
        assert msg.json()["result_message"] == expect['result_message']
        assert msg.json()['result_code'] == expect['result_code']
        assert 'url' in msg.json()['data']

 conftest.py

'''
Code description:配置信息
Create time: 2020/12/3
Developer: 葉修
'''
import os
import pytest
from api.get_token import Get_Token
from common.http_requests import HttpRequests


@pytest.fixture(scope="session")
def get_token():
    '''前置操作獲取token並傳入headers'''
    Get_Token().get_token()
    if not HttpRequests().params.get("access_token", ""):#沒有get到token，跳出用例
        pytest.skip("未獲取token跳過用例")
    yield HttpRequests().req
    HttpRequests().req.close()

def pytest_addoption(parser):
    parser.addoption(
        "--cmdhost", action="store", default="http://192.168.1.54:32099",
        help="my option: type1 or type2"
    )
@pytest.fixture(scope="session",autouse=True)
def host(request):
    '''獲取命令行參數'''
    #獲取命令行參數給到環境變量
    #pytest --cmdhost 運行指定環境
    os.environ["host"] = request.config.getoption("--cmdhost")
    print("當前用例運行測試環境:%s" % os.environ["host"])

 common/connect_mysql.py

'''
Code description: 配置連接數據庫
Create time: 2020/12/3
Developer: 葉修
'''
import pymysql

dbinfo = {
    "host":"******",
    "user":"root",
    "password":"******",
    "port":31855
}
class DbConnect():
    def __init__(self,db_conf,database=""):
        self.db_conf = db_conf
        #打開數據庫
        self.db = pymysql.connect(database = database,
                                  cursorclass = pymysql.cursors.DictCursor,
                                  **db_conf)
        #使用cursor()方式獲取操作游標
        self.cursor = self.db.cursor()

    def select(self,sql):
        #sql查詢
        self.cursor.execute(sql)#執行sql
        results = self.cursor.fetchall()
        return results

    def execute(self,sql):
        #sql 刪除 提示 修改
        try:
            self.cursor.execute(sql)#執行sql
            self.db.commit()#提交修改
        except:
            #發生錯誤時回滾
            self.db.rollback()

    def close(self):
        self.db.close()#關閉連接

def select_sql(select_sql):
    '''查詢數據庫'''
    db = DbConnect(dbinfo,database='auth_platform')
    result = db.select(select_sql)
    db.close()
    return result

def execute_sql(sql):
    '''執行SQL'''
    db = DbConnect(dbinfo,database='auth_platform')
    db.execute(sql)
    db.close()

if __name__ == '__main__':
    sql = "SELECT * FROM auth_platform.auth_service where name='四要素認證'"
    sel = select_sql(sql)[0]['name']
    print(sel)

 

 common/http_requests.py

'''
Code description: 封裝自己的請求類型
Create time: 2020/12/3
Developer: 葉修
'''

import requests


# 定義一個HttpRequests的類
class HttpRequests(object):

    req = requests.session()#定義session會話
    # 定義公共請求頭
    headers = {
        'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/84.0.4147.125 Safari/537.36',
        'cookie':''
    }
    params = {
        'access_token':''
    }
    # 封裝自己的get請求,獲取資源
    def get(self, url='', params='', data='', headers=None, cookies=None,stream=None,verify=None):
        response = self.req.get(url,params=params,data=data,headers=headers,cookies=cookies,stream=stream,verify=verify)
        return response

    # 封裝自己的post方法，創建資源
    def post(self, url='', params='',data='', json='', headers=None, cookies=None,stream=None,verify=None):
        response = self.req.post(url,params=params,data=data,json=json,headers=headers,cookies=cookies,stream=stream,verify=verify)
        return response

    # 封裝自己的put方法，更新資源
    def put(self, url='', params='', data='', headers=None, cookies=None,verify=None):
        response = self.req.put(url, params=params, data=data, headers=headers, cookies=cookies,verify=verify)
        return response

    # 封裝自己的delete方法，刪除資源
    def delete(self, url='', params='', data='', headers=None, cookies=None,verify=None):
        response = self.req.delete(url, params=params, data=data, headers=headers, cookies=cookies,verify=verify)
        return response

 common/logger.py

'''
Code description: 封裝輸出日志文件
Create time: 2020/12/3
Developer: 葉修
'''
import logging,time
import os
import getpathinfo

path = getpathinfo.get_path()#獲取本地路徑
log_path = os.path.join(path,'logs')# log_path是存放日志的路徑
# 如果不存在這個logs文件夾，就自動創建一個
if not os.path.exists(log_path):os.mkdir(log_path)

class Log():
    def __init__(self):
        #文件的命名
        self.logname = os.path.join(log_path,'%s.log'%time.strftime('%Y_%m_%d'))
        self.logger = logging.getLogger()
        self.logger.setLevel(logging.DEBUG)
        #日志輸出格式
        self.formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')

    def __console(self,level,message):
        #創建一個fileHander，用於寫入本地
        fh = logging.FileHandler(self.logname,'a',encoding='utf-8')
        fh.setLevel(logging.DEBUG)
        fh.setFormatter(self.formatter)
        self.logger.addHandler(fh)

        #創建一個StreamHandler,用於輸入到控制台
        ch = logging.StreamHandler()
        ch.setLevel(logging.DEBUG)
        ch.setFormatter(self.formatter)
        self.logger.addHandler(ch)

        if level == 'info':
            self.logger.info(message)
        elif level == 'debug':
            self.logger.debug(message)
        elif level == 'warning':
            self.logger.warning(message)
        elif level == 'error':
            self.logger.error(message)
        #避免日志重復
        self.logger.removeHandler(fh)
        self.logger.removeHandler(ch)
        #關閉打開文件
        fh.close()

    def debug(self,message):
        self.__console('debug',message)

    def info(self,message):
        self.__console('info',message)

    def warning(self,message):
        self.__console('warning',message)

    def error(self,message):
        self.__console('error',message)

if __name__ == '__main__':
    log = Log()
    log.info('測試')
    log.debug('測試')
    log.warning('測試')
    log.error('測試')

 common/read_save_data.py

'''
Code description: 讀取保存數據
Create time: 2020/12/8
Developer: 葉修
'''
import os
import yaml
import getpathinfo
class Read_Save_Date():
    def __init__(self):
        path = getpathinfo.get_path()#獲取本地路徑
        self.head_img_path = os.path.join(path,'data','save_data')+"/"+'head_img_path.txt'# head_img_path文件地址
        self.order_id_path = os.path.join(path, 'data', 'save_data') + "/" + 'order_id.txt'  # order_id.txt文件地址

    def get_head_img_path(self):
        # 獲取head_img_path
        with open(self.head_img_path, "r", encoding="utf-8")as f:
            return f.read()

    def get_order_id(self):
        # 獲取order_id
        with open(self.order_id_path, "r", encoding="utf-8")as f:
            return f.read()

if __name__ == '__main__':
    print(Read_Save_Date().get_head_img_path())
    print(Read_Save_Date().get_order_id())

 common/read_yaml.py

'''
Code description: 讀取yml文件測試數據
Create time: 2020/12/3
Developer: 葉修
'''
import os
import yaml
import getpathinfo
class ReadYaml():
    def __init__(self,filename):
        path = getpathinfo.get_path()#獲取本地路徑
        self.filepath = os.path.join(path,'data','test_data')+"/"+filename#拼接定位到data文件夾

    def get_yaml_data(self):
        with open(self.filepath, "r", encoding="utf-8")as f:
            # 調用load方法加載文件流
            return yaml.load(f,Loader=yaml.FullLoader)

if __name__ == '__main__':
    data = ReadYaml("auth_service.yml").get_yaml_data()['add_or_update_service']
    print(data)

data/

 data/test_data/auth_service.yml

home_service_list:
  - [{'result_code': '0', 'result_message': '處理成功'}]
service_info:
  - ['','',{'result_code': '0', 'result_message': '處理成功'}]
add_or_update_service:
  - [['1','1','1','1','1','123456','1','測試','1','1','1','1','1','1'],{'result_code': '0', 'result_message': '處理成功'}]
add_service_price:
  - ['123456789','10','0','','0',{'result_code': '0', 'result_message': '處理成功'}]
apply_service:
  - ['','',{'result_code': '0', 'result_message': '處理成功'}]

 logs/

 report/

 getpathinfo.py

'''
Code description:配置文件路徑
Create time: 2020/12/3
Developer: 葉修
'''
import os

def get_path():
    # 獲取當前路徑
    curpath = os.path.dirname(os.path.realpath(__file__))
    return curpath

if __name__ == '__main__':# 執行該文件，測試下是否OK
    print('測試路徑是否OK,路徑為：', get_path())

pytest.ini

#pytest.ini
[pytest]
markers = process
addopts = -p no:warnings
#addopts = -v --reruns 1 --html=./report/report.html --self-contained-html
#addopts = -v --reruns 1 --alluredir ./report/allure_raw
#addopts = -v -s -p no:warnings --reruns  1 --pytest_report ./report/Pytest_Report.html

requirements.txt

allure-pytest==2.8.18
allure-python-commons==2.8.18
BeautifulReport==0.1.3
beautifulsoup4==4.9.3
ddt==1.4.1
Faker==4.18.0
Flask==1.1.1
httpie==1.0.3
httplib2==0.9.2
HttpRunner==1.5.8
py==1.9.0
PyMySQL==0.10.1
pytest==6.1.1
pytest-base-url==1.4.2
pytest-cov==2.10.1
pytest-forked==1.3.0
pytest-html==2.1.1
pytest-instafail==0.4.2
pytest-metadata==1.10.0
pytest-mock==3.3.1
pywin32==228
PyYAML==5.3.1
requests==2.22.0
requests-oauthlib==1.3.0
requests-toolbelt==0.9.1

run_main.py

'''
Code description: 運行主流程測試用例
Create time: 2020/11/5
Developer: 葉修
'''
import os
import pytest
if __name__ == '__main__':
    pytest.main(['-m','process', '-s','--alluredir', 'report/tmp'])#-m運行mark標記文件
    os.system('allure generate report/tmp -o report/html --clean') # /report/tmp 為存放報告的源文件目錄