updated test for jsonify status code argument

This commit is contained in:
elishowk 2011-06-01 18:09:43 +02:00
parent f195459cf1
commit 29169e52ee

View file

@ -815,18 +815,20 @@ class JSONTestCase(unittest.TestCase):
def test_jsonify(self): def test_jsonify(self):
d = dict(a=23, b=42, c=[1, 2, 3]) d = dict(a=23, b=42, c=[1, 2, 3])
status=200
app = flask.Flask(__name__) app = flask.Flask(__name__)
@app.route('/kw') @app.route('/kw')
def return_kwargs(): def return_kwargs():
return flask.jsonify(**d) return flask.jsonify(status, **d)
@app.route('/dict') @app.route('/dict')
def return_dict(): def return_dict():
return flask.jsonify(d) return flask.jsonify(status, d)
c = app.test_client() c = app.test_client()
for url in '/kw', '/dict': for url in '/kw', '/dict':
rv = c.get(url) rv = c.get(url)
assert rv.mimetype == 'application/json' assert rv.mimetype == 'application/json'
assert flask.json.loads(rv.data) == d assert flask.json.loads(rv.data) == d
assert rv.status_code == status
def test_json_attr(self): def test_json_attr(self):
app = flask.Flask(__name__) app = flask.Flask(__name__)