def test_format_suffix_url_rules_calls_original_endpoint_without_format_suffix_urls(self): with mock.MagicMock() as targetview: def check_request_effect(*args,**kwargs): self.assertEqual(flask.request.endpoint, 'test_function') self.assertEqual(flask.request.url, 'http://localhost/test') targetview.side_effect = check_request_effect app = flask.Flask(__name__) app.add_url_rule("/test", 'test_function', api_view(targetview)) format_suffix_url_rules(app) with app.test_request_context('/test.json'): app.full_dispatch_request() targetview.assert_called_once_with()
def test_format_suffix_url_rules(self): with mock.MagicMock() as targetview: def check_request_effect(*args,**kwargs): #Should only accept json, nothing else self.assertTrue(flask.request.accept_mimetypes.accept_json) self.assertFalse(flask.request.accept_mimetypes.accept_html) targetview.side_effect = check_request_effect app = flask.Flask(__name__) app.add_url_rule("/test", 'test_function', api_view(targetview)) format_suffix_url_rules(app) with app.test_request_context('/test.json', headers={'Accept':'application/noexist'}): app.full_dispatch_request() targetview.assert_called_once_with()
def test_format_suffix_url_rules_calls_original_endpoint_without_format_suffix_urls( self): with mock.MagicMock() as targetview: def check_request_effect(*args, **kwargs): self.assertEqual(flask.request.endpoint, 'test_function') self.assertEqual(flask.request.url, 'http://localhost/test') targetview.side_effect = check_request_effect app = flask.Flask(__name__) app.add_url_rule("/test", 'test_function', api_view(targetview)) format_suffix_url_rules(app) with app.test_request_context('/test.json'): app.full_dispatch_request() targetview.assert_called_once_with()
def test_format_suffix_url_rules(self): with mock.MagicMock() as targetview: def check_request_effect(*args, **kwargs): #Should only accept json, nothing else self.assertTrue(flask.request.accept_mimetypes.accept_json) self.assertFalse(flask.request.accept_mimetypes.accept_html) targetview.side_effect = check_request_effect app = flask.Flask(__name__) app.add_url_rule("/test", 'test_function', api_view(targetview)) format_suffix_url_rules(app) with app.test_request_context( '/test.json', headers={'Accept': 'application/noexist'}): app.full_dispatch_request() targetview.assert_called_once_with()