Python做接口自动化测试框架

# env config
ENV = ‘test‘

APP_VERSION = ‘2.3.7‘

HEADERS = {‘content-type‘: ‘application/x-www-form-urlencoded; charset=UTF-8‘}

# test url test config
API_TEST_BASE_URL = "http://api.abc.com"

# redis config
REDIS_HOST = ‘‘
REDIS_PORT = ‘‘

# mysql config
DB_HOST = ‘‘
DB_PORT = ‘‘
DB_USER = ‘‘
DB_PASSWORD = ‘‘

# -*- coding:utf-8 -*-
import json,requests
import settings


class BaseApi(object):
    url = ""
    base_url= settings.API_TEST_BASE_URL

    def __init__(self,url_params=None):
        if not url_params:
            url_params = []
        self.url_params = url_params
        self.response = None
        self.base_url = self.base_url

    # 拼接url
    def api_url(self):
        if not self.url:
            raise RuntimeError("no url been set")
        return self._get_url()

    def _get_url(self):
        format_url = self.url.format(self.url_params)
        return "{0}{1}".format(self.base_url, format_url)

    # 封装POST请求类型
    def post(self, data=None):
        if not data:
            data = {}
        base_param = self.build_base_param()
        custom_param = self.build_custom_param(data)
        data.update(base_param)
        data.update(custom_param)
        self.response = requests.post(url=self.api_url(), data=data, headers=settings.HEADERS)
        return self.response

    # 封装GET请求类型
    def get(self,data=None):
        if not data:
            data={}
        base_param = self.build_base_param()
        custom_param = self.build_custom_param(data)
        data.update(base_param)
        data.update(custom_param)
        response = requests.get(url=self.api_url(),params=data)
        return self.response

    # 获取回参中状态码
    def get_code(self):
        if self.response:
            return json.loads(self.response.text)[‘code‘]

    # 获取HTTP状态码
    def get_status_code(self):
        if self.response:
            return self.response.status_code

    # 获取回参中message
    def get_response_message(self):
        if self.response:
            return json.loads(self.response.text)[‘msg‘]

    # 所有接口共有的入参，比如：app_version、token等
    def build_base_param(self):
        return {
                "app_version": SETTINGS.APP_VERSION,
                "token":""
        }

    # 被测接口除公共参数外所需的其余参数
    def build_custom_param(self, data):
        return {}

import pymysql.cursors,settings


def execute(sql, params=None, db=‘‘, is_fetchone=True):
    # Connect to the database
    connection = pymysql.connect(host=settings.DB_HOST,
                                 port=settings.DB_PORT,
                                 user=settings.DB_USER,
                                 password=settings.DB_PASSWORD,
                                 db=db,
                                 autocommit=True,
                                 charset=‘utf8mb4‘,
                                 cursorclass=pymysql.cursors.DictCursor)
    try:
        with connection.cursor() as cursor:
            cursor.execute(sql, params)
            if is_fetchone:
                return cursor.fetchone()
            else:
                return cursor.fetchall()
    finally:
        connection.close()

# -*- coding:utf-8 -*-
import db

def get_sms_captcha(mobile):
    # 获取短信验证码
    sms_captcha = db.execute(‘select code from send_sms_code where mobile=%s order by id desc‘,params=(mobile))
    return sms_captcha[‘code‘]

def delete_user(mobile):
    # 删除用户
    db.execute(‘delete from user where mobile=%s‘,params=(mobile))

# -*- coding:utf-8 -*-
from base_api.base_api import BaseApi
import settings

class RegisterApi(BaseApi):
    url = ‘/home/register‘
    
    #对BaseApi类中build_custom_param方法重写
    def build_custom_param(self, data):
        return {‘login_name‘:data[‘login_name‘],‘password‘:data[‘password‘],‘code‘:data[‘code‘],‘nickname‘:data[‘nickname‘]}

# -*- coding:utf-8 -*-
from base_api.base_api import BaseApi
import settings


class SendSmsCaptcha(BaseApi):
    url = ‘/user/sendsms‘

    def build_custom_param(self, data):
        return {‘type‘: data[‘type‘], ‘phone‘: data[‘phone‘]}

# -*- coding:utf-8 -*-
from unittest import TestCase
from base_api.register_api import RegisterApi
from base_api.send_sms_code_api import SendSmsCaptcha
from utilities import user
import settings,json



class TestRegisterApi(TestCase):
    new_mobile = ‘13000000001‘
    password = ‘123abc‘
    nick_name = ‘XiangXi‘


    def test_register_success(self):
        # 调用发送短信验证码接口
        send_sms_code_api = SendSmsCaptcha()
        send_sms_code_api.post({‘type‘:‘register‘,‘phone‘:self.new_mobile})

        # 校验发送短信验证码接口HTTP状态码为200
        self.assertEqual(send_sms_code_api.get_status_code(),200)

        # 校验发送短信验证码接口反参中code为0，代表成功（不同项目该字段值不一定为0）
        self.assertEqual(send_sms_code_api.get_code(),0)

        # 通过数据库获取短信验证码
        sms_code = user.get_sms_captcha(self.new_mobile)

        # 调用注册接口
        register_api = RegisterApi()
        response = register_api.post({‘login_name‘:self.new_mobile,‘password‘:self.password,‘code‘:sms_code,‘nickname‘:self.nick_name})

        # 校验注册接口HTTP状态码为200
        self.assertEqual(send_sms_code_api.get_status_code(),200)

        # 校验注册接口反参中code为0
        self.assertEqual(register_api.get_code(),0)

        # 校验反参中手机号、登录名与用户昵称是否与入参值一样
        identity_obj = json.loads(response.content)[‘result‘][‘identity_obj‘]
        self.assertEqual(identity_obj[‘nickname‘],self.nick_name)
        self.assertEqual(identity_obj[‘login_name‘],self.new_mobile)
        self.assertEqual(identity_obj[‘mobilephone‘],self.new_mobile)

    def tearDown(self):
        user.delete_user(self.new_mobile)

# -*- coding:utf-8 -*-
from base_api.base_api import BaseApi
from base_api.login_api import LoginApi

class LoginBaseApi(BaseApi):
    def __init__(self, login_name,password, *args, **kwargs):
        super(LoginBaseApi, self).__init__(*args, **kwargs)
        self.login_name = login_name
        self.password = password

    def build_base_param(self):
        base_param = super(LoginBaseApi, self).build_base_param()
        response = LoginApi().post(self.login_name, self.password)
        base_param[‘token‘] = token
        return base_param