Add more tests

tests/test_flask.py

@@ -1,6 +1,12 @@
 import unittest
 from flask import Flask
 from wsmeext.flask import signature
+from wsme.types import Base, text
+
+
+class Model(Base):
+    name = text
+
 
 test_app = Flask(__name__)
 
@@ -17,6 +23,24 @@ def divide_by_zero():
     return 1 / 0
 
 
+@test_app.route('/models')
+@signature([Model])
+def list_models():
+    return [Model(name=1)]
+
+
+@test_app.route('/models/<name>')
+@signature(Model)
+def get_model(name):
+    return Model(name=name)
+
+
+@test_app.route('/models', methods=['POST'])
+@signature(Model, body=Model)
+def post_model(body):
+    return Model(name=body.name)
+
+
 class FlaskrTestCase(unittest.TestCase):
 
     def setUp(self):
@@ -30,12 +54,27 @@ class FlaskrTestCase(unittest.TestCase):
         r = self.app.get('/multiply?a=2&b=5')
         assert r.data == '10'
 
+    def test_get_model(self):
+        resp = self.app.get('/models/test')
+        assert resp.status_code == 200
+
+    def test_list_models(self):
+        resp = self.app.get('/models')
+        assert resp.status_code == 200
+
+    def test_post_model(self):
+        resp = self.app.post('/models', data={"name": "test"})
+        import ipdb
+        ipdb.set_trace()
+        assert resp.status_code == 200
+
     def test_serversideerror(self):
         r = self.app.get('/divide_by_zero')
         assert r.status_code == 500
         self.assertEquals(
             r.data,
-            '{"debuginfo": null, "faultcode": "Server", "faultstring": "integer division or modulo by zero"}'
+            '{"debuginfo": null, "faultcode": "Server", "faultstring": '
+            '"integer division or modulo by zero"}'
         )
 
 if __name__ == '__main__':