def test_base_url(self):
"""
Tests that the base_url always returns the
correct shit.
"""
app = Flask('myapp')
d = FlaskDispatcher(app)
with app.test_request_context():
self.assertEqual(d.base_url, 'http://localhost/')
d = FlaskDispatcher(app, url_prefix='someprefix', auto_options_name='Options2')
with app.test_request_context():
self.assertEqual(d.base_url, 'http://localhost/someprefix')
def test_flask_dispatch_wrapper_fail(self):
"""
Tests the response when their is a failure in
dispatching a request that is not a RestException
"""
adapter_class = self.get_mock_adapter_class()
def fake(*args, **kwargs):
raise Exception
d = FlaskDispatcher(self.app)
d.register_adapters(adapter_class)
view_func = flask_dispatch_wrapper(d, fake)
self.assertEqual(view_func.__name__, fake.__name__)
with self.app.test_request_context('/myresource'):
self.assertRaises(Exception, view_func)
def test_register_route(self):
d = FlaskDispatcher(self.app)
def fake():
pass
endpoint = 'someendpoint'
values = dict(endpoint=endpoint, endpoint_func=fake,
route='/myresource', methods=['GET'])
d.register_route(**values)
self.assertIn(endpoint, self.app.url_map._rules_by_endpoint)
rule = self.app.url_map._rules_by_endpoint[endpoint][0]
self.assertIn('GET', rule.methods)
self.assertEqual(rule.rule, '/myresource')
self.assertEqual(endpoint, rule.endpoint)
# They won't actually be the same function since it gets wrapped.
self.assertEqual(self.app.view_functions[endpoint].__name__, fake.__name__)
def test_register_route_invalid_options(self):
"""
Tests registering a route with options not accepted
by Flask.
"""
app = Flask('myapp')
d = FlaskDispatcher(app)
def fake():
return
d.register_route('/ha', route='/ha', endpoint_func=fake, options=dict(invalid='option'))
for rul in app.url_map._rules:
if rul.endpoint == '/ha':
break
else:
assert False
def test_blueprint_base_url(self):
"""
Tests that dispatcher return the right shit
when a blueprint is passed in.
"""
with self.app.test_request_context():
bp = Blueprint('name', __name__)
d = FlaskDispatcher(bp)
self.assertEqual(d.base_url, 'http://localhost/')
d = FlaskDispatcher(bp, url_prefix='someprefix')
self.assertEqual(d.base_url, 'http://localhost/someprefix')
bp2 = Blueprint('name', __name__, url_prefix='another')
d = FlaskDispatcher(bp2)
self.assertEqual(d.base_url, 'http://localhost/another/')
d = FlaskDispatcher(bp2, url_prefix='again')
self.assertEqual(d.base_url, 'http://localhost/another/again')
def test_flask_dispatch_wrapper(self):
"""
Tests the wrapper to ensure that it properly calls the
apimethod
"""
adapter_class = self.get_mock_adapter_class()
def fake(*args, **kwargs):
return mock.Mock()
d = FlaskDispatcher(self.app)
d.register_adapters(adapter_class)
view_func = flask_dispatch_wrapper(d, fake)
self.assertEqual(view_func.__name__, fake.__name__)
with self.app.test_request_context('/myresource'):
response = view_func()
self.assertEqual(response.content_type, 'fake')
self.assertEqual(response.status_code, 600)
self.assertEqual(response.data.decode('utf8'), 'some body')
def test_flask_dispatch_wrapper_fail_restexception(self):
"""
Tests the response when their is a failure in
dispatching a request. In particular when a RestException
is raised.
"""
adapter_class = self.get_mock_adapter_class()
def fake(*args, **kwargs):
raise RestException
d = FlaskDispatcher(self.app)
d.register_adapters(adapter_class)
view_func = flask_dispatch_wrapper(d, fake)
self.assertEqual(view_func.__name__, fake.__name__)
with self.app.test_request_context('/myresource'):
response = view_func()
self.assertEqual(response.status_code, 500)
self.assertEqual(response.data.decode('utf8'), 'error')
self.assertEqual(response.content_type, 'fake')
