def test_api_rval_response_list(self):
""" Tests that werkzeug rvals are not changed """
self.mock(auth, 'authenticate').using(lambda: models.User())
self.mock(api.APIResource, 'as_view').using(staticmethod(
lambda _: lambda *args, **kwargs: [1, 2, 3]))
with self.app.test_request_context('/api/v1/'):
api_wrapper = urls.register_api(api.APIResource, 'fake_api', 'fake')
response = api_wrapper(test=True)
self.assertEqual(response.status_code, 200)
def test_api_rval_response(self):
""" Tests that werkzeug rvals are not changed """
rval = Response()
self.mock(auth, 'authenticate').using(lambda: models.User())
self.mock(api.APIResource, 'as_view').using(staticmethod(
lambda _: lambda *args, **kwargs: rval))
with self.app.test_request_context('/api/v1/'):
self.assertTrue(isinstance(rval, Response))
self.assertFalse(isinstance(rval, flask.Response))
api_wrapper = urls.register_api(api.APIResource, 'fake_api', 'fake')
response = api_wrapper()
self.assertTrue(isinstance(rval, Response))
self.assertFalse(isinstance(rval, flask.Response))
self.assertEqual(response, rval)
def api_wrapper(self): """ Returns nested function, api_wrapper in register_api """ return urls.register_api(api.APIResource, 'fake_api', 'fake')
