diff options
Diffstat (limited to 'websdk/werkzeug/testsuite/urls.py')
| -rw-r--r-- | websdk/werkzeug/testsuite/urls.py | 168 |
1 files changed, 168 insertions, 0 deletions
diff --git a/websdk/werkzeug/testsuite/urls.py b/websdk/werkzeug/testsuite/urls.py new file mode 100644 index 0000000..8c908b8 --- /dev/null +++ b/websdk/werkzeug/testsuite/urls.py @@ -0,0 +1,168 @@ +# -*- coding: utf-8 -*- +""" + werkzeug.testsuite.urls + ~~~~~~~~~~~~~~~~~~~~~~~ + + URL helper tests. + + :copyright: (c) 2011 by Armin Ronacher. + :license: BSD, see LICENSE for more details. +""" + +import unittest +from StringIO import StringIO + +from werkzeug.testsuite import WerkzeugTestCase + +from werkzeug.datastructures import OrderedMultiDict +from werkzeug import urls + + +class URLsTestCase(WerkzeugTestCase): + + def test_quoting(self): + assert urls.url_quote(u'\xf6\xe4\xfc') == '%C3%B6%C3%A4%C3%BC' + assert urls.url_unquote(urls.url_quote(u'#%="\xf6')) == u'#%="\xf6' + assert urls.url_quote_plus('foo bar') == 'foo+bar' + assert urls.url_unquote_plus('foo+bar') == 'foo bar' + assert urls.url_encode({'a': None, 'b': 'foo bar'}) == 'b=foo+bar' + assert urls.url_fix(u'http://de.wikipedia.org/wiki/Elf (Begriffsklärung)') == \ + 'http://de.wikipedia.org/wiki/Elf%20%28Begriffskl%C3%A4rung%29' + + def test_url_decoding(self): + x = urls.url_decode('foo=42&bar=23&uni=H%C3%A4nsel') + assert x['foo'] == '42' + assert x['bar'] == '23' + assert x['uni'] == u'Hänsel' + + x = urls.url_decode('foo=42;bar=23;uni=H%C3%A4nsel', separator=';') + assert x['foo'] == '42' + assert x['bar'] == '23' + assert x['uni'] == u'Hänsel' + + x = urls.url_decode('%C3%9Ch=H%C3%A4nsel', decode_keys=True) + assert x[u'Üh'] == u'Hänsel' + + def test_streamed_url_decoding(self): + item1 = 'a' * 100000 + item2 = 'b' * 400 + string = 'a=%s&b=%s&c=%s' % (item1, item2, item2) + gen = urls.url_decode_stream(StringIO(string), limit=len(string), + return_iterator=True) + self.assert_equal(gen.next(), ('a', item1)) + self.assert_equal(gen.next(), ('b', item2)) + self.assert_equal(gen.next(), ('c', item2)) + self.assert_raises(StopIteration, gen.next) + + def test_url_encoding(self): + assert urls.url_encode({'foo': 'bar 45'}) == 'foo=bar+45' + d = {'foo': 1, 'bar': 23, 'blah': u'Hänsel'} + assert urls.url_encode(d, sort=True) == 'bar=23&blah=H%C3%A4nsel&foo=1' + assert urls.url_encode(d, sort=True, separator=';') == 'bar=23;blah=H%C3%A4nsel;foo=1' + + def test_sorted_url_encode(self): + assert urls.url_encode({"a": 42, "b": 23, 1: 1, 2: 2}, sort=True) == '1=1&2=2&a=42&b=23' + assert urls.url_encode({'A': 1, 'a': 2, 'B': 3, 'b': 4}, sort=True, + key=lambda x: x[0].lower()) == 'A=1&a=2&B=3&b=4' + + def test_streamed_url_encoding(self): + out = StringIO() + urls.url_encode_stream({'foo': 'bar 45'}, out) + self.assert_equal(out.getvalue(), 'foo=bar+45') + + d = {'foo': 1, 'bar': 23, 'blah': u'Hänsel'} + out = StringIO() + urls.url_encode_stream(d, out, sort=True) + self.assert_equal(out.getvalue(), 'bar=23&blah=H%C3%A4nsel&foo=1') + out = StringIO() + urls.url_encode_stream(d, out, sort=True, separator=';') + self.assert_equal(out.getvalue(), 'bar=23;blah=H%C3%A4nsel;foo=1') + + gen = urls.url_encode_stream(d, sort=True) + self.assert_equal(gen.next(), 'bar=23') + self.assert_equal(gen.next(), 'blah=H%C3%A4nsel') + self.assert_equal(gen.next(), 'foo=1') + self.assert_raises(StopIteration, gen.next) + + def test_url_fixing(self): + x = urls.url_fix(u'http://de.wikipedia.org/wiki/Elf (Begriffskl\xe4rung)') + assert x == 'http://de.wikipedia.org/wiki/Elf%20%28Begriffskl%C3%A4rung%29' + + x = urls.url_fix('http://example.com/?foo=%2f%2f') + assert x == 'http://example.com/?foo=%2f%2f' + + def test_iri_support(self): + self.assert_raises(UnicodeError, urls.uri_to_iri, u'http://föö.com/') + self.assert_raises(UnicodeError, urls.iri_to_uri, 'http://föö.com/') + assert urls.uri_to_iri('http://xn--n3h.net/') == u'http://\u2603.net/' + assert urls.uri_to_iri('http://%C3%BCser:p%C3%A4ssword@xn--n3h.net/p%C3%A5th') == \ + u'http://\xfcser:p\xe4ssword@\u2603.net/p\xe5th' + assert urls.iri_to_uri(u'http://☃.net/') == 'http://xn--n3h.net/' + assert urls.iri_to_uri(u'http://üser:pässword@☃.net/påth') == \ + 'http://%C3%BCser:p%C3%A4ssword@xn--n3h.net/p%C3%A5th' + + assert urls.uri_to_iri('http://test.com/%3Fmeh?foo=%26%2F') == \ + u'http://test.com/%3Fmeh?foo=%26%2F' + + # this should work as well, might break on 2.4 because of a broken + # idna codec + assert urls.uri_to_iri('/foo') == u'/foo' + assert urls.iri_to_uri(u'/foo') == '/foo' + + def test_ordered_multidict_encoding(self): + d = OrderedMultiDict() + d.add('foo', 1) + d.add('foo', 2) + d.add('foo', 3) + d.add('bar', 0) + d.add('foo', 4) + assert urls.url_encode(d) == 'foo=1&foo=2&foo=3&bar=0&foo=4' + + def test_href(self): + x = urls.Href('http://www.example.com/') + assert x('foo') == 'http://www.example.com/foo' + assert x.foo('bar') == 'http://www.example.com/foo/bar' + assert x.foo('bar', x=42) == 'http://www.example.com/foo/bar?x=42' + assert x.foo('bar', class_=42) == 'http://www.example.com/foo/bar?class=42' + assert x.foo('bar', {'class': 42}) == 'http://www.example.com/foo/bar?class=42' + self.assert_raises(AttributeError, lambda: x.__blah__) + + x = urls.Href('blah') + assert x.foo('bar') == 'blah/foo/bar' + + self.assert_raises(TypeError, x.foo, {"foo": 23}, x=42) + + x = urls.Href('') + assert x('foo') == 'foo' + + def test_href_url_join(self): + x = urls.Href('test') + assert x('foo:bar') == 'test/foo:bar' + assert x('http://example.com/') == 'test/http://example.com/' + + if 0: + # stdlib bug? :( + def test_href_past_root(self): + base_href = urls.Href('http://www.blagga.com/1/2/3') + assert base_href('../foo') == 'http://www.blagga.com/1/2/foo' + assert base_href('../../foo') == 'http://www.blagga.com/1/foo' + assert base_href('../../../foo') == 'http://www.blagga.com/foo' + assert base_href('../../../../foo') == 'http://www.blagga.com/foo' + assert base_href('../../../../../foo') == 'http://www.blagga.com/foo' + assert base_href('../../../../../../foo') == 'http://www.blagga.com/foo' + + def test_url_unquote_plus_unicode(self): + # was broken in 0.6 + assert urls.url_unquote_plus(u'\x6d') == u'\x6d' + assert type(urls.url_unquote_plus(u'\x6d')) is unicode + + def test_quoting_of_local_urls(self): + rv = urls.iri_to_uri(u'/foo\x8f') + assert rv == '/foo%C2%8F' + assert type(rv) is str + + +def suite(): + suite = unittest.TestSuite() + suite.addTest(unittest.makeSuite(URLsTestCase)) + return suite |