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