diff options
Diffstat (limited to 'tests/test_dec.py')
| -rw-r--r-- | tests/test_dec.py | 267 |
1 files changed, 267 insertions, 0 deletions
diff --git a/tests/test_dec.py b/tests/test_dec.py new file mode 100644 index 0000000..0f65590 --- /dev/null +++ b/tests/test_dec.py @@ -0,0 +1,267 @@ +import unittest +from webob import Request +from webob import Response +from webob.dec import _format_args +from webob.dec import _func_name +from webob.dec import wsgify + +class DecoratorTests(unittest.TestCase): + def _testit(self, app, req): + if isinstance(req, basestring): + req = Request.blank(req) + resp = req.get_response(app) + return resp + + def test_wsgify(self): + resp_str = 'hey, this is a test: %s' + @wsgify + def test_app(req): + return resp_str % req.url + resp = self._testit(test_app, '/a url') + self.assertEqual(resp.body, resp_str % 'http://localhost/a%20url') + self.assertEqual(resp.content_length, 45) + self.assertEqual(resp.content_type, 'text/html') + self.assertEqual(resp.charset, 'UTF-8') + self.assertEqual('%r' % (test_app,), 'wsgify(tests.test_dec.test_app)') + + def test_wsgify_empty_repr(self): + self.assertEqual('%r' % (wsgify(),), 'wsgify()') + + def test_wsgify_args(self): + resp_str = 'hey hey my my' + @wsgify(args=(resp_str,)) + def test_app(req, strarg): + return strarg + resp = self._testit(test_app, '/a url') + self.assertEqual(resp.body, resp_str) + self.assertEqual(resp.content_length, 13) + self.assertEqual(resp.content_type, 'text/html') + self.assertEqual(resp.charset, 'UTF-8') + self.assertEqual('%r' % (test_app,), + "wsgify(tests.test_dec.test_app, args=('%s',))" % resp_str) + + def test_wsgify_kwargs(self): + resp_str = 'hey hey my my' + @wsgify(kwargs=dict(strarg=resp_str)) + def test_app(req, strarg=''): + return strarg + resp = self._testit(test_app, '/a url') + self.assertEqual(resp.body, resp_str) + self.assertEqual(resp.content_length, 13) + self.assertEqual(resp.content_type, 'text/html') + self.assertEqual(resp.charset, 'UTF-8') + self.assertEqual('%r' % test_app, + "wsgify(tests.test_dec.test_app, " + "kwargs={'strarg': '%s'})" % resp_str) + + def test_wsgify_raise_httpexception(self): + from webob.exc import HTTPBadRequest + @wsgify + def test_app(req): + import sys + if sys.version_info < (2,5): + raise HTTPBadRequest(None).exception + raise HTTPBadRequest + resp = self._testit(test_app, '/a url') + self.assert_(resp.body.startswith('400 Bad Request')) + self.assertEqual(resp.content_type, 'text/plain') + self.assertEqual(resp.charset, 'UTF-8') + self.assertEqual('%r' % test_app, + "wsgify(tests.test_dec.test_app)") + + def test_wsgify_no___get__(self): + # use a class instance instead of a fn so we wrap something w/ + # no __get__ + class TestApp(object): + def __call__(self, req): + return 'nothing to see here' + test_app = wsgify(TestApp()) + resp = self._testit(test_app, '/a url') + self.assertEqual(resp.body, 'nothing to see here') + self.assert_(test_app.__get__(test_app) is test_app) + + def test_wsgify_args_no_func(self): + test_app = wsgify(None, args=(1,)) + self.assertRaises(TypeError, self._testit, test_app, '/a url') + + def test_wsgify_wrong_sig(self): + @wsgify + def test_app(req): + return 'What have you done for me lately?' + req = dict() + self.assertRaises(TypeError, test_app, req, 1, 2) + self.assertRaises(TypeError, test_app, req, 1, key='word') + + def test_wsgify_none_response(self): + @wsgify + def test_app(req): + return + resp = self._testit(test_app, '/a url') + self.assertEqual(resp.body, '') + self.assertEqual(resp.content_type, 'text/html') + self.assertEqual(resp.content_length, 0) + + def test_wsgify_get(self): + resp_str = "What'choo talkin' about, Willis?" + @wsgify + def test_app(req): + return Response(resp_str) + resp = test_app.get('/url/path') + self.assertEqual(resp.body, resp_str) + + def test_wsgify_post(self): + post_dict = dict(speaker='Robin', + words='Holy test coverage, Batman!') + @wsgify + def test_app(req): + return Response('%s: %s' % (req.POST['speaker'], + req.POST['words'])) + resp = test_app.post('/url/path', post_dict) + self.assertEqual(resp.body, '%s: %s' % (post_dict['speaker'], + post_dict['words'])) + + def test_wsgify_request_method(self): + resp_str = 'Nice body!' + @wsgify + def test_app(req): + self.assertEqual(req.method, 'PUT') + return Response(req.body) + resp = test_app.request('/url/path', method='PUT', + body=resp_str) + self.assertEqual(resp.body, resp_str) + self.assertEqual(resp.content_length, 10) + self.assertEqual(resp.content_type, 'text/html') + + def test_wsgify_undecorated(self): + def test_app(req): + return Response('whoa') + wrapped_test_app = wsgify(test_app) + self.assert_(wrapped_test_app.undecorated is test_app) + + def test_wsgify_custom_request(self): + resp_str = 'hey, this is a test: %s' + class MyRequest(Request): + pass + @wsgify(RequestClass=MyRequest) + def test_app(req): + return resp_str % req.url + resp = self._testit(test_app, '/a url') + self.assertEqual(resp.body, resp_str % 'http://localhost/a%20url') + self.assertEqual(resp.content_length, 45) + self.assertEqual(resp.content_type, 'text/html') + self.assertEqual(resp.charset, 'UTF-8') + self.assertEqual('%r' % (test_app,), "wsgify(tests.test_dec.test_app, " + "RequestClass=<class 'tests.test_dec.MyRequest'>)") + + def test_middleware(self): + resp_str = "These are the vars: %s" + @wsgify.middleware + def set_urlvar(req, app, **vars): + req.urlvars.update(vars) + return app(req) + from webob.dec import _MiddlewareFactory + self.assert_(set_urlvar.__class__ is _MiddlewareFactory) + repr = '%r' % (set_urlvar,) + self.assert_(repr.startswith('wsgify.middleware(<function set_urlvar at ')) + @wsgify + def show_vars(req): + return resp_str % (sorted(req.urlvars.items())) + show_vars2 = set_urlvar(show_vars, a=1, b=2) + self.assertEqual('%r' % (show_vars2,), + 'wsgify.middleware(tests.test_dec.set_urlvar)' + '(wsgify(tests.test_dec.show_vars), a=1, b=2)') + resp = self._testit(show_vars2, '/path') + self.assertEqual(resp.body, resp_str % "[('a', 1), ('b', 2)]") + self.assertEqual(resp.content_type, 'text/html') + self.assertEqual(resp.charset, 'UTF-8') + self.assertEqual(resp.content_length, 40) + + def test_unbound_middleware(self): + @wsgify + def test_app(req): + return Response('Say wha!?') + unbound = wsgify.middleware(None, test_app, some='thing') + from webob.dec import _UnboundMiddleware + self.assert_(unbound.__class__ is _UnboundMiddleware) + self.assertEqual(unbound.kw, dict(some='thing')) + self.assertEqual('%r' % (unbound,), + "wsgify.middleware(wsgify(tests.test_dec.test_app), " + "some='thing')") + @unbound + def middle(req, app, **kw): + return app(req) + self.assert_(middle.__class__ is wsgify) + self.assertEqual('%r' % (middle,), + "wsgify.middleware(tests.test_dec.middle)" + "(wsgify(tests.test_dec.test_app), some='thing')") + + def test_unbound_middleware_no_app(self): + unbound = wsgify.middleware(None, None) + from webob.dec import _UnboundMiddleware + self.assert_(unbound.__class__ is _UnboundMiddleware) + self.assertEqual(unbound.kw, dict()) + self.assertEqual('%r' % (unbound,), + "wsgify.middleware()") + + def test_classapp(self): + class HostMap(dict): + @wsgify + def __call__(self, req): + return self[req.host.split(':')[0]] + app = HostMap() + app['example.com'] = Response('1') + app['other.com'] = Response('2') + resp = Request.blank('http://example.com/').get_response(wsgify(app)) + self.assertEqual(resp.content_type, 'text/html') + self.assertEqual(resp.charset, 'UTF-8') + self.assertEqual(resp.content_length, 1) + self.assertEqual(resp.body, '1') + + def test__func_name(self): + def func(): + pass + name = _func_name(func) + self.assertEqual(name, 'tests.test_dec.func') + name = _func_name('a') + self.assertEqual(name, "'a'") + class Klass(object): + @classmethod + def classmeth(cls): + pass + def meth(self): + pass + name = _func_name(Klass) + self.assertEqual(name, 'tests.test_dec.Klass') + k = Klass() + kname = _func_name(k) + self.assert_(kname.startswith('<tests.test_dec.Klass object at 0x')) + name = _func_name(k.meth) + self.assert_(name.startswith('tests.test_dec.%s' % kname)) + self.assert_(name.endswith('>.meth')) + name = _func_name(Klass.meth) + self.assertEqual(name, 'tests.test_dec.Klass.meth') + name = _func_name(Klass.classmeth) + self.assertEqual(name, "tests.test_dec.<class " + "'tests.test_dec.Klass'>.classmeth") + + def test__format_args(self): + args_rep = _format_args() + self.assertEqual(args_rep, '') + kw = dict(a=4, b=5, c=6) + args_rep = _format_args(args=(1, 2, 3), kw=kw) + self.assertEqual(args_rep, '1, 2, 3, a=4, b=5, c=6') + args_rep = _format_args(args=(1, 2, 3), kw=kw, leading_comma=True) + self.assertEqual(args_rep, ', 1, 2, 3, a=4, b=5, c=6') + class Klass(object): + a = 1 + b = 2 + c = 3 + names = ['a', 'b', 'c'] + obj = Klass() + self.assertRaises(AssertionError, _format_args, names=names) + args_rep = _format_args(obj=obj, names='a') + self.assertEqual(args_rep, 'a=1') + args_rep = _format_args(obj=obj, names=names) + self.assertEqual(args_rep, 'a=1, b=2, c=3') + args_rep = _format_args(kw=kw, defaults=dict(a=4, b=5)) + self.assertEqual(args_rep, 'c=6') |