python測試api接口

在開發中，需要測試web-api的接口 spring mvc 使用單元測試非常方便，但是，受不了單元測試的啟動速度。用python寫了一個小腳本，用於測試接口，

測試腳本配置文件
api.yaml

server:
  url: http://127.0.0.1:9000/ihome/

api:
  name:
    #api-v2-neighbor-list.yaml
    - api/v2/neighbor/list   

api-v2-neighbor-list.yaml
接口配置文件

method:
  post
data:
  #post 的 body 的json
  postSid: a1
  userSid: u2

python 腳本

import requests, json, yaml, sys


def apiTest(apiName):
    f = open("api.yaml")
    obj = yaml.safe_load(f)
    f.close()

    if apiName != "":
        runApi(obj["server"]["url"] + apiName, apiName.replace("/", "-") + ".yaml")
        return;

    apis = obj['api']["name"]
    for api in apis:
        runApi(obj["server"]["url"] + api, api.replace("/", "-") + ".yaml")

def runApi(url, dataFile):
    headers = {'Content-Type' : 'application/json; charset=UTF-8',
                          'X-Requested-With' : 'XMLHttpRequest',
                          'Connection' : 'keep-alive',
                          'User-Agent' : 'Mozilla/5.0 (Windows NT 6.1; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/49.0.2623.110 Safari/537.36'
                    }

    f = open(dataFile)
    obj = yaml.safe_load(f)
    f.close()

    s = requests.session()

    if obj["method"] == "get":
        r = s.get(url= url, headers = headers)
    if obj["method"] == "post":
        r = s.post(url= url , data = json.dumps(obj["data"]), headers = headers)


    status = r.status_code
    if status != 200:
        print "\x1b[31mfail: status-> {} {}\x1b[0m".format(status, url)
        return

    resp = json.loads(r.text)
    if resp["success"] == 0:
        print "\x1b[32mpass: {}\x1b[0m".format(url)
    else:
        print "\x1b[31mfail: {} message: {}\x1b[0m".format(url, resp["message"].encode('utf-8'))


def main():
    apiName = ""
    if len(sys.argv) > 1:
        apiName = sys.argv[1]

    apiTest(apiName)

if __name__ == '__main__':
    main()