当前位置：首页 > news >正文

使用python+flask设置挡板

news 2025/9/27 7:43:50

在测试过程中，可能会需要使用挡板来模拟外部系统接口。比如涉及到外部系统接口调用时，可能会需要特定的返回，比如某个字段为指定值。或者在压测过程中，为了更准确的了解系统的性能，需要模拟外部系统的接口返回。

使用python+flask设置挡板，具体代码如下

```python
from flask import Flask, request
from xml.etree.ElementTree import fromstring, Element,SubElement,tostring

app = Flask(__name__)

@app.route('/directResponse', methods=['POST'])
def directResponse():
# 直接返回
response = {"status": 0, "message": "Success", "data": {"col1": "value1", "col2": "value2"}}
return response

@app.route('/baseOnRequest/response', methods=['POST'])
def baseOnRequest_response():
# 根据请求参数的不同，返回不同的响应
request_data = request.get_json() # 获取json格式的请求体
age = request_data.get('age')
name = request_data.get('name')

isadult = False
if int(age) >= 18:
isadult = True

response = {"status": 0, "message": "Success", "data": {"username": name, "isadult": isadult}}
return response

@app.route('/getMessage/<msg>', methods=['GET', 'POST'])
def getMessage(msg):
# 请求路径中带有参数的接口
userid = request.args.get('userid', default=None)
if None == userid:
return {"errorMessage": "用户id为空，请输入！"}
if msg == 'idcard':
response = {"status": 0, "message": "Success", "userid": userid, "idcard": "110110199901011234"}
elif msg == 'mobile':
response = {"status": 0, "message": "Success", "userid": userid,
"mobiles": [{"phoneNo": "13512341234"}, {"phoneNo": "13523452345"}]}
elif msg == 'name':
response = {"status": 0, "message": "Success", "userid": userid, "relname": "真实姓名"}
else:
response = {"status": -1, "message": "没有您要查询的信息", }
return response

@app.route('/xmlResponse',methods=['POST'])
def xmlResponse():
# 构建soup协议的XML响应
envelope = Element('soap:Envelope')
envelope.set('xmlns:soap','http:www.w3.org/2003/05/soap-envelope')
body = SubElement(envelope,'soap:Body')
response = SubElement(body,'xmlResponse')
response.set('xmlns','http://example.com/soap-service')

user = SubElement(response, 'user')
SubElement(user,'userid').text = '11111'
return tostring(envelope,encoding='unicode',xml_declaration=True)

if __name__ == '__main__':
app.run(host='127.0.0.1', port='8087', debug=True)

```

调用的结果分别如下

/directResponse