python unittest实现api自动化测试_python-Python教程-PHP中文网

python unittest实现api自动化测试_python

不言

发布： 2018-04-04 16:44:55

原创

1959 人浏览过

这篇文章主要为大家详细介绍了python unittest实现api自动化测试的方法，具有一定的参考价值，感兴趣的小伙伴们可以参考一下

项目测试对于一个项目的重要性，大家应该都知道吧，写python的朋友，应该都写过自动化测试脚本。
最近正好负责公司项目中的api测试，下面写了一个简单的例子，对API 测试进行梳理。

首先，编写restful api接口文件 testpost.py，包含了get，post，put方法

#!/usr/bin/env python
# -*- coding: utf-8 -*-
from flask import request
from flask_restful import Resource
from flask_restful import reqparse


test_praser = reqparse.RequestParser()
test_praser.add_argument(&#39;ddos&#39;)


class TestPost(Resource):
  def post(self, PostData):
    data = request.get_json()
    user = User(&#39;wangjing&#39;)
    if data[&#39;ddos&#39;]:
     return {&#39;hello&#39;: &#39;uese&#39;, "PostData": PostData, &#39;ddos&#39;: &#39;data[\&#39;ddos\&#39;]&#39;}
    return {&#39;hello&#39;: &#39;uese&#39;, "PostData": PostData}

  def get(self, PostData):
    data = request.args
    if data and data[&#39;ddos&#39;]:
      return "hello" + PostData + data[&#39;ddos&#39;], 200
    return {&#39;hello&#39;: &#39;uese&#39;, "PostData": PostData}

  def put(self, PostData):
    data = test_praser.parse_args()
    if data and data[&#39;ddos&#39;]:
      return "hello" + PostData + data[&#39;ddos&#39;], 200
    return {&#39;hello&#39;: &#39;uese&#39;, "PostData": PostData}

登录后复制

ps:对于request的取值，我这里定义了常用的三种方法：

post方法：request.get_json(),在调用API时，传值是json方式
get和put方法:request.args 或者reqparse.RequestParser(),调用API时，传的是字符串

其次，定义Blueprint（蓝图）文件 init.py

#!/usr/bin/env python
  # -*- coding: utf-8 -*-
  from flask import Blueprint
  from flask_restful import Api
  from testpost import TestPost

  testPostb = Blueprint(&#39;testPostb&#39;, __name__)
  api = Api(testPostb)
  api.add_resource(TestPost, &#39;/<string:PostData>/postMeth&#39;)

登录后复制

然后,编写测试脚本testPostM.py

#!/usr/bin/env python
  # -*- coding: utf-8 -*-
  import unittest
  import json
  from secautoApp.api.testPostMeth import api
  from flask import url_for
  from run import app
  from secautoApp.api.testPostMeth import TestPost

  headers = {&#39;Accept&#39;: &#39;application/json&#39;,
        &#39;Content-Type&#39;: &#39;application/json&#39;
        }

  class APITestCase(unittest.TestCase):
    def setUp(self):
      # self.app = create_app(os.getenv("SECAUTOCFG") or &#39;default&#39;)
      self.app = app
      #   self.app_context = self.app.app_context()
      #   self.app_context.push()
      self.client = self.app.test_client()

    #
    # def tearDown(self):
    #   self.app_context.pop()

    def test_post(self):
      # with app.test_request_context():

      response = self.client.get(api.url_for(TestPost, PostData=&#39;adb&#39;, ddos=&#39;123&#39;))
      self.assertTrue(response.status_code == 200)

      response = self.client.get(url_for(&#39;testPostb.testpost&#39;, PostData=&#39;adb&#39;, ddos=&#39;123&#39;))
      self.assertTrue(response.status_code == 200)  
      self.assertTrue(json.loads(response.data)[&#39;PostData&#39;] ==&#39;adb&#39;)

      response = self.client.post(url_for(&#39;testPostb.testpost&#39;, PostData=&#39;adb&#39;), headers=headers,
                    data=json.dumps({"ddos": &#39;123&#39;}))
      print json.loads(response.data)
      self.assertTrue(response.status_code == 200)

      response = self.client.put(url_for(&#39;testPostb.testpost&#39;, PostData=&#39;adb&#39;, ddos=&#39;123&#39;))
      self.assertTrue(json.loads(response.data) == &#39;helloadb123&#39;)

      response = self.client.put(url_for(&#39;testPostb.testpost&#39;, PostData=&#39;adb&#39;))
      print json.loads(response.data)[&#39;PostData&#39;]
      self.assertTrue(response.status_code == 200)

登录后复制

ps:调用的api url 主要用的是flask_restful 的api.url_for,或者是flask的url_for,下面我来说下这2种方法的具体使用

flask_restful 的api.url_for说明

api.url_for(TestPost,PostData='adb'),这里的TestPost指的是restful api接口文件中定义的class，因为我们在api蓝图中，已经通过api.add_resource(TestPost, ‘//postMeth')添加类的方式定义过

flask的url_for的使用说明

url_for(‘testPostb.testpost', PostData='adb', ddos='123'),'testPostb.testpost'这个字符串中
testPostb指的是蓝图的名称，也就是testPostb = Blueprint(‘testPostb', name)中Blueprint(‘testPostb',name)中的testPostb。
testpost指的是蓝图下endpoit的端点名称,flask_restful中，指的是api.add_resource(TestPost, ‘//postMeth')中类名TestPost的小写

启动测试脚本：

C:\secauto3>python run.py test
test_post (testPostM.APITestCase) ... ok

----------------------------------------------------------------------
Ran 1 test in 0.056s

OK

登录后复制

小总结：url_for的传值和request中的取值是有对应关系的，最后就是flask_restful中端点的方式，一定要是api.add_resource中类名的小写。