python中的unittest测试,确实很方便。但是只对每个api都要定义TestCase,这就很繁琐。
那要是针对每个api,做单独的配置,让测试自动运行就方便很多。
下面,我们用requests库来搭建自己的测试框架。

1.首先,定义需要配置的api的list。

url_prex='http://127.0.0.1:5000'

    [{'request_id': 1,
         'request_name': 'api_url',
         'request_method': 'GET',#或者是post,put之类的
         'request_url': url_prex + '/api/test',
         'request_data': None,
         'respone_status_code': 200,
         'respone_data': "success"
         },
        {'request_id': 2,
         'request_name': 'api_url',
         'request_method': 'POST',#或者是post,put之类的
         'request_url': url_prex + '/api/test',
         'request_data': {'name':'testname'},
         'respone_status_code': 200,
         'respone_data': "success"
         },
    ]

2.定义post方法需要的header

    headers = {
        'Accept': 'application/json',
        'Content-Type': 'application/json,application/x-www-form-urlencoded'
    }

3,先进行api登录,登录之后才可以对api进行调用

    req = requests.session()
    req.get('http://127.0.0.1:5000/auth/login',
            data={'userName': 'admin', 'password': "admin"})

    url_prex = 'http://99.1.15.23:8080'

4.封装requests的方法,实现单一的get,post,delete,put等方法的统一传值和调用。

def url_call(method, url, data):
        try:
            if not method or method.upper() == 'GET':
                response = req.request('GET',
                                       url,
                                       params=data)
            else:
                response = req.request(method,
                                       url,
                                       headers=headers,
                                       data=data)
        except Exception as e:
            raise e
        return response

5,针对list的每一个api,进行测试

def test_result(url):
        try:
            rs = url_call(url['request_method'], url['request_url'],
                          url['request_data'])
            if "<title>404 Not Found</title>" in rs.text:
                return u'%s 测试结果为: %s!,原因:request_url 地址错误! request_id: %s,url:%s method:%s request_data:%s' % (
                    url['request_name'], 'FAIL', url['request_id'],
                    url['request_url'], url['request_method'],
                    change_request_data_str(url['request_data'])), False
            if "Traceback <em>(most recent call last)" in rs.text:
                return u'%s 测试结果为: %s!,原因:被调用的api发生了Exception! request_id: %s,url:%s method:%s request_data:%s' % (
                    url['request_name'], 'Api Exception', url['request_id'],
                    url['request_url'], url['request_method'],
                    change_request_data_str(url['request_data'])), False
            if url['respone_status_code'] and rs.status_code != url[
                'respone_status_code']:
                return u'%s 测试结果为: %s!,原因:status_code不一致! request_id: %s, url:%s method:%s request_data:%s' % (
                    url['request_name'], 'FAIL', url['request_id'],
                    url['request_url'], url['request_method'],
                    change_request_data_str(url['request_data'])), False
            if url['respone_data'] and url['respone_data'] not in rs.text:
                return u'%s 测试结果为: %s!,原因:期待的数据不在结果中! request_id: %s ,url:%s method:%s request_data:%s' % (
                    url['request_name'], 'FAIL', url['request_id'],
                    url['request_url'], url['request_method'],
                    change_request_data_str(url['request_data'])), False
        except Exception as e:
            return u'%s 测试结果为: %s!,原因:测试程序发生了内部错误,%s! request_id: %s ,url:%s method:%s request_data:%s' % (
                url['request_name'], 'FAIL', e.__str__(), url['request_id'],
                url['request_url'], url['request_method'],
                change_request_data_str(url['request_data'])), False
        return u'%s 测试结果为: %s! request_id: %s ,url:%s method:%s request_data:%s' % (
            url['request_name'], 'Success', url['request_id'], url['request_url'],
            url['request_method'],
            change_request_data_str(url['request_data'])), True

6,对list进行遍历测试,并对结果进行统计

    def url_test():
        result_false_list = []
        begin_time = time.time()
        for url in url_map:
            result_str, result_status = test_result(url)
            print result_str
            if not result_status:
                result_false_list.append(result_str)
        percent = float(
            url_map.__len__() - result_false_list.__len__()) * 100 / float(
            url_map.__len__())
        print u"总用时:%s ,测试总数:%s,失败个数:%s,测试成功率为:%.2f%%" % (
            time.time() - begin_time, url_map.__len__(),
            result_false_list.__len__(), percent)
        return result_false_list

7,用flask_script进行app封装,和调用,并把错误记录到fail_log日志

#!/usr/bin/env python
    # -*- coding: UTF-8 -*-

    import sys

    from flask import Flask
    from flask_script import Manager

    reload(sys)
    sys.setdefaultencoding('utf-8')

    app = Flask(__name__)
    manager = Manager(app)

    @manager.command
    def api():
        from tests.api.requests_help import url_test
        result = url_test()
        if result:
            import time
            date_str = time.strftime('%Y-%m-%d %H:%M:%S',
                                     time.localtime(time.time())) + '\r\n'
            log_str = '\r\n'.join(result) + '\r\n'
            open('fail_log', 'a').write(date_str + log_str)

    if __name__ == "__main__":
        manager.run(default_command='runserver')

8,用命令python main.py api 进行api自动化测试,可以得到如下测试结果

    test_ip_url 测试结果为: Success! request_id: 13 ,url:http://127.0.0.1:5000/api/test method:GET request_data:{'command': 'true', 'Name': 'office_pooxl'}                                                       
    test_ip_url 测试结果为: FAIL!,原因:status_code不一致! request_id: 14, url:http://127.0.0.1:5000/api/test/name method:GET request_data:{'virtualName': 'NameServer', 'ip': '110.65.139.196'} 

    总用时:24.8880000114 ,测试总数:2,失败个数:1,测试成功率为:50.00% 

最后,只要对list进行配置,就可以进行api的自动化测试,并得到想要的测试报告

**完整代码和具体的框架我放到了我的资源中,供大家免费使用。地址:http://download.csdn.net/detail/u010445540/9800522
封装了requests 写法,unittest写法 和页面selenium的测试,请进入我的github进行下载:。。。。。。。。**

Logo

权威|前沿|技术|干货|国内首个API全生命周期开发者社区

更多推荐