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')