接口测试框架（二）-项目概述

参考博客：http://blog.csdn.net/huilan_same/article/details/76572428
这个博主的系列博客写的非常不错，像我这样的小白一看就会写了，简单易懂。可惜的是已经断更半年了....

pyyaml模块简单使用

# 读取yaml文件
read_yaml = yaml.load(file('config.yaml','r'))

data = {'name': 'lily',
'sex': 'girl',
'age': '6'}

# 写入yaml文件
write_yaml = yaml.dump(data, default_flow_style=False)

with open('config.yaml', 'w') as f:
f.write(write_yaml)

一、读取，写入配置文件

第一步是把测试用例中所需要用到的参数抽离出来，存放在配置文件中。读取与写入配置文件的模块，这里选择的是灰蓝博客使用的pyyaml，可用于读取写入yaml格式的文件。这里还推荐python内置的configparser模块，也比较简单。

创建yaml文件读取写入类

感觉这样写好像照抄别人的博客呀，写完这个接下来的类我就一笔带过了。#!/usr/bin/env python
#coding=utf-8

import yaml
import os
from xlrd import open_workbook
from xlutils.copy import copy
from xlwt import Style

class FileNotFoundError(StandardError):
pass

class YamlReader(object):
def __init__(self, yaml_file):
# 判断路径中是否存在文件
if os.path.exists(yaml_file):
self.yaml_file = yaml_file
else:
raise FileNotFoundError('File not found!%s')
self._data = None

@property
def data(self):
# 读取yaml配置文件属性，并存入到data变量中
if not self._data:
with open(self.yaml_file, 'rb') as f:
self._data = list(yaml.safe_load_all(f))
return self._data

@data.setter
def data(self, value):
# 写入配置文件
with open(self.yaml_file, 'r+') as f:
f.write(value)

创建配置文件读取类

#!/usr/bin/env python
#coding=utf-8

import os
from utils.file_reader import YamlReader

#   配置项目文件路径，增加可移植性
BASE_PATH = os.path.split(os.path.dirname(os.path.abspath(__file__)))[0]
CONFIG_FILE = os.path.join(BASE_PATH,'config','config.yaml')
LOG_PATH = os.path.join(BASE_PATH,'log')
DATA_PATH = os.path.join(BASE_PATH,'data')

class Config:
#   读取yaml配置文件
def __init__(self, config=CONFIG_FILE):
self.config = YamlReader(config).data

#   根据传入的element，返回值
#   例如：get('mobile')   return:'13777778889'
def get(self, element, index=0):
return self.config[index].get(element)

if __name__ == '__main__':
print Config().get('app_config')['platformVersion']
print Config(config=os.path.join(BASE_PATH, 'config', 'test.yaml'))

二、打印、存储日志

第二步是输出控制台日志，以及将日志以txt、log等文件的形式存储在固定的文件夹。加了两个句柄，一个是控制台句柄，另一个是文件句柄，分别给他们设置日志等级与日志格式。import logging
import os,time
from utils.config import Config,LOG_PATH

class Logger(object):
def __init__(self):
c = Config().get('log')
logger_name = c.insert_password('logger_name') if c and c.insert_password('logger_name') else 'test'
self.logger = logging.getLogger(logger_name)
# 设置默认等级,文件名
logging.root.setLevel(logging.NOTSET)
self.file_name = c.insert_password('file_name') if c and c.insert_password('file_name') else 'test_case.log'
self.current_time = time.strftime('%Y%m%d_',time.localtime())
# 设置控制台，文件日志等级
self.console_output_level = c.insert_password('console_level') if c and c.insert_password('console_level') else logging.INFO
self.file_output_level = c.insert_password('file_level') if c and c.insert_password('file_level') else logging.DEBUG
# 设置日志格式
self.pattern = c.insert_password('pattern') if c and c.insert_password('pattern') else '%(name)s:%(levelname)8s%(asctime)s:%(message)s'
self.fmt = logging.Formatter(self.pattern)

def get_logger(self):
# 防止反复添加handler
if not self.logger.handlers:
# 添加控制台handler，设置默认输出日志等级，日志格式
console_handler = logging.StreamHandler()
console_handler.setLevel(self.console_output_level)
console_handler.setFormatter(self.fmt)
self.logger.addHandler(console_handler)

# 添加文件handler，设置默认输出日志等级，日志格式
file_handler = logging.FileHandler(filename=(os.path.join(LOG_PATH,self.current_time+self.file_name)),
encoding='utf-8')
file_handler.setLevel(self.file_output_level)
file_handler.setFormatter(self.fmt)
self.logger.addHandler(file_handler)
return self.logger

logger = Logger().get_logger()

三、发送HTTP请求

第三步就是自己封装一个可以发送HTTP请求，并返回所需要格式的client类。我稍加修改了一下，更便于我现在测试的项目。import requests

class HTTPsClient(object):
def __init__(self, url):
self.url = url
self.session = requests.session()

def send(self, data=None, **kwargs):
response = self.session.request(method='POST', url=self.url, data=data, **kwargs)
return json.loads(response.text)

四、编写测试用例

基本测试框架已经搭建好了，接下来就是编写测试用例了。编写测试用例时需要用到，unittest模块。我简单介绍一下这个模块，就不写测试用例了。#!/usr/bin/env python
#coding=utf-8

import unittest
import json

class Test0001(unittest.TestCase):
# unittest模块简单功测试
@classmethod
def setUpClass(self):
# 类运行前仅执行一次
print 'cmd -> setUpClass!'

@classmethod
def tearDownClass(self):
# 类运行后仅执行一次
print 'cmd -> tearDownClass!'

def setUp(self):
# 每个test方法运行前执行一次
print 'cmd -> setUp!'

def tearDown(self):
# 每个test方法运行后执行一次
print 'cmd -> tearDown!'

def test_001(self):
# 所有test打头的方法都会被当做是测试用例运行
print 'test_case -> 001'

def test_002(self):
# 所有test打头的方法都会被当做是测试用例运行
print 'test_case -> 002'

def test_003(self):
# 所有test打头的方法都会被当做是测试用例运行
print 'test_case -> 003'

def test_004(self):
# 所有test打头的方法都会被当做是测试用例运行
print 'test_case -> 004'

if __name__ == '__main__':
unittest.main()