diff --git a/tests/pecantest/test/controllers/ws.py b/tests/pecantest/test/controllers/ws.py index d85ff55..05b640d 100644 --- a/tests/pecantest/test/controllers/ws.py +++ b/tests/pecantest/test/controllers/ws.py @@ -70,6 +70,10 @@ class AuthorsController(RestController): def get(self, id): if id == 999: raise wsme.exc.ClientSideError('Wrong ID') + + if id == 911: + return wsme.api.Response(Author(), + status_code=401) author = Author() author.id = id author.firstname = u"aname" diff --git a/tests/pecantest/test/tests/test_ws.py b/tests/pecantest/test/tests/test_ws.py index 506f0ef..af8942c 100644 --- a/tests/pecantest/test/tests/test_ws.py +++ b/tests/pecantest/test/tests/test_ws.py @@ -87,6 +87,14 @@ class TestWS(FunctionalTest): self.assertEqual(res.status, '400 Bad Request') assert 'Client' in res.body + def test_non_default_response(self): + res = self.app.get( + '/authors/911.json', + expect_errors=True + ) + self.assertEqual(res.status_int, 401) + self.assertEqual(res.status, '401 Unauthorized') + def test_serversideerror(self): res = self.app.get('/divide_by_zero.json', expect_errors=True) self.assertEqual(res.status, '500 Internal Server Error')