Add tests and fix check ordering in flask.py

This commit is contained in:
Endre Karlson
2013-03-28 12:44:52 +00:00
parent 1e699bf143
commit 6f9bb4d14f
2 changed files with 24 additions and 6 deletions

View File

@@ -1,6 +1,7 @@
import unittest
from flask import Flask, json
from wsmeext.flask import signature
from wsme.api import Response
from wsme.types import Base, text
@@ -51,6 +52,18 @@ def post_model(body):
return Model(name=body.name)
@test_app.route('/status_sig')
@signature(int, status_code=201)
def get_status_sig():
return 1
@test_app.route('/status_response')
@signature(int)
def get_status_response():
return Response(1, status_code=201)
class FlaskrTestCase(unittest.TestCase):
def setUp(self):
@@ -90,6 +103,14 @@ class FlaskrTestCase(unittest.TestCase):
)
assert resp.status_code == 200
def test_get_status_sig(self):
resp = self.app.get('/status_sig')
assert resp.status_code == 201
def test_get_status_response(self):
resp = self.app.get('/status_response')
assert resp.status_code == 201
def test_serversideerror(self):
r = self.app.get('/divide_by_zero')
assert r.status_code == 500