python之mock web接口

在日常的测试工作中,有时会有需要调用外部接口,拿到返回数据用以满足当前的测试任务的需求。但是当外部接口不可用,或者没有提供测试用环境时,我们就需要自己来mock一个接口的返回内容了,先让我们看一看下面的这段代码

from wsgiref.simple_server import make_server

def application(environ,response):
    response('200 OK', [('Content-Type', 'text/html')])
    res = b'BestTest'
    return [res]

http = make_server('',8000,application)
print('server in 8000....')
http.serve_forever()

上面的application()函数就是符合WSGI标准的一个HTTP处理函数,它接收两个参数:

  • environ:一个包含所有HTTP请求信息的dict对象;
  • response:一个发送HTTP响应的函数。

response('200 OK', [('Content-Type', 'text/html')])则是将http响应的header发出去,同时需要注意的是只能发送一次。response()函数会接收两个参数,一个是HTTP响应码,也就是'200 OK',一个是一组list表示的HTTP Header,每个Header用一个包含两个str的tuple表示。

直接运行这段代码,同时请求我们本机的ip:8000 或127.0.0.1:8000这个地址,在这时会发现我们设置的返回结果出现在了屏幕上。

《python之mock web接口》

但是问题就来了,有的人会问:我要的接口返回的是一段json返回呀,你这个只是一个字符串的返回我根本用不了啊,那么让我们来试试下面这一段

from wsgiref.simple_server import make_server
import json

def application(environ,response):
    response('200 OK', [('Content-Type', 'text/json')])
    res = {'data':[{"class":'BestTest性能测试',"Teacher":'安大叔'},
                   {"class":'BestTest性能测试',"Teacher":'liml'}]}
    return [json.dumps(res).encode('utf-8')]

http = make_server('',8000,application)
print('server in 8000....')
http.serve_forever()

我们再来运行一下试试看

《python之mock web接口》

好了,这下返回的结果可以使用了。

发表评论

电子邮件地址不会被公开。