【发布时间】：2016-07-12 04:51:00
【问题描述】：
我正在尝试测试控制器做出的响应。当我通过 gunicorn 与 testFramework
运行它时，响应是不同的我的server.py 如下所示：
app = Flask(__name__, static_url_path='/pub')
api = Api(app, catch_all_404s=True)
connect('myapp', host='mongomock://localhost')
api.add_resource(IndexController, "/")
if __name__ == '__main__':
app.run(host='0.0.0.0', debug=debug)
我的IndexController 看起来像：
from flask_restful import Resource
from myapp.models.User import User
class IndexController(Resource):
"""
This controller shows all the Users
"""
@classmethod
def get(cls):
"""
Simple GET call for /
"""
data = []
for user in User.objects:
data.append({
"name": user.name,
"location": user.location
})
User 对象：
from mongoengine import *
import datetime
class User(Document):
name = StringField(max_length=200, required=True, unique=True)
location = PointField(default=[0,0])
我的IndexControllerTest：
class IndexControllerTest(unittest.TestCase):
"""
This is the default controller Base class which
sets up the server app and has handy asserts to make
life easier when testing the controllers
"""
def setUp(self):
"""
Set up a global instance of the controller
"""
self.app = server.app.test_client()
def tearDown(self):
conn = get_connection()
conn.myapp.user.drop()
#Have to do this because it does not clear the unique index
def assertGet(self, path, data):
"""
This will run a GET request and test the output
"""
response = self.app.get(path)
#print str(response.get_data())
#print flask.json.dumps(response.data)
parsed = flask.json.loads(response.data)
self.assertEqual(parsed, data)
def test_get_returns_one_user(self):
"""
When I call a get I should get Users with thier data
"""
user = User(name="muse")
user.save()
self.assertGet(
"/",
[
{
"name": "muse",
"location": [ 0, 0 ]
}
],
status_code=200
)
通过gunicorn 输出，这就是我想要的！：
[
{
"name": "Hello",
"location": [
52.201962,
0.128145
]
},
{
"name": "World",
"location": [
0,
0
]
}
]
我的测试输出：
...
First differing element 0:
{u'name': u'muse', u'location': {u'type': u'Point', u'coordinates': [0, 0]}}
{'name': 'muse', 'location': [0, 0]}
- [{u'location': {u'coordinates': [0, 0], u'type': u'Point'}, u'name': u'muse'}]
+ [{'location': [0, 0], 'name': 'muse'}]
什么？为什么？在哪里？谁？ @£$@$%&£$(%&£$((&疯狂！
我希望flask_restfull API 确保两种情况下的输出相同
【问题讨论】：
-
仅供参考，mogomock 标签指的是github.com/vmalloc/mongomock，mongoengine 将其用作模拟框架
标签： python mongoengine flask-restful mongomock