PageRenderTime 44ms CodeModel.GetById 11ms RepoModel.GetById 0ms app.codeStats 0ms

/tests/regressiontests/httpwrappers/tests.py

https://code.google.com/p/mango-py/
Python | 298 lines | 284 code | 4 blank | 10 comment | 0 complexity | 325916362e34419706601a21abd5df39 MD5 | raw file
Possible License(s): BSD-3-Clause
  1. import copy
  2. import pickle
  3. from django.core.exceptions import SuspiciousOperation
  4. from django.http import (QueryDict, HttpResponse, HttpResponseRedirect,
  5. HttpResponsePermanentRedirect,
  6. SimpleCookie, BadHeaderError,
  7. parse_cookie)
  8. from django.utils import unittest
  9. class QueryDictTests(unittest.TestCase):
  10. def test_missing_key(self):
  11. q = QueryDict('')
  12. self.assertRaises(KeyError, q.__getitem__, 'foo')
  13. def test_immutability(self):
  14. q = QueryDict('')
  15. self.assertRaises(AttributeError, q.__setitem__, 'something', 'bar')
  16. self.assertRaises(AttributeError, q.setlist, 'foo', ['bar'])
  17. self.assertRaises(AttributeError, q.appendlist, 'foo', ['bar'])
  18. self.assertRaises(AttributeError, q.update, {'foo': 'bar'})
  19. self.assertRaises(AttributeError, q.pop, 'foo')
  20. self.assertRaises(AttributeError, q.popitem)
  21. self.assertRaises(AttributeError, q.clear)
  22. def test_immutable_get_with_default(self):
  23. q = QueryDict('')
  24. self.assertEqual(q.get('foo', 'default'), 'default')
  25. def test_immutable_basic_operations(self):
  26. q = QueryDict('')
  27. self.assertEqual(q.getlist('foo'), [])
  28. self.assertEqual(q.has_key('foo'), False)
  29. self.assertEqual('foo' in q, False)
  30. self.assertEqual(q.items(), [])
  31. self.assertEqual(q.lists(), [])
  32. self.assertEqual(q.items(), [])
  33. self.assertEqual(q.keys(), [])
  34. self.assertEqual(q.values(), [])
  35. self.assertEqual(len(q), 0)
  36. self.assertEqual(q.urlencode(), '')
  37. def test_single_key_value(self):
  38. """Test QueryDict with one key/value pair"""
  39. q = QueryDict('foo=bar')
  40. self.assertEqual(q['foo'], 'bar')
  41. self.assertRaises(KeyError, q.__getitem__, 'bar')
  42. self.assertRaises(AttributeError, q.__setitem__, 'something', 'bar')
  43. self.assertEqual(q.get('foo', 'default'), 'bar')
  44. self.assertEqual(q.get('bar', 'default'), 'default')
  45. self.assertEqual(q.getlist('foo'), ['bar'])
  46. self.assertEqual(q.getlist('bar'), [])
  47. self.assertRaises(AttributeError, q.setlist, 'foo', ['bar'])
  48. self.assertRaises(AttributeError, q.appendlist, 'foo', ['bar'])
  49. self.assertTrue(q.has_key('foo'))
  50. self.assertTrue('foo' in q)
  51. self.assertFalse(q.has_key('bar'))
  52. self.assertFalse('bar' in q)
  53. self.assertEqual(q.items(), [(u'foo', u'bar')])
  54. self.assertEqual(q.lists(), [(u'foo', [u'bar'])])
  55. self.assertEqual(q.keys(), ['foo'])
  56. self.assertEqual(q.values(), ['bar'])
  57. self.assertEqual(len(q), 1)
  58. self.assertRaises(AttributeError, q.update, {'foo': 'bar'})
  59. self.assertRaises(AttributeError, q.pop, 'foo')
  60. self.assertRaises(AttributeError, q.popitem)
  61. self.assertRaises(AttributeError, q.clear)
  62. self.assertRaises(AttributeError, q.setdefault, 'foo', 'bar')
  63. self.assertEqual(q.urlencode(), 'foo=bar')
  64. def test_urlencode(self):
  65. q = QueryDict('', mutable=True)
  66. q['next'] = '/a&b/'
  67. self.assertEqual(q.urlencode(), 'next=%2Fa%26b%2F')
  68. self.assertEqual(q.urlencode(safe='/'), 'next=/a%26b/')
  69. q = QueryDict('', mutable=True)
  70. q['next'] = u'/t\xebst&key/'
  71. self.assertEqual(q.urlencode(), 'next=%2Ft%C3%ABst%26key%2F')
  72. self.assertEqual(q.urlencode(safe='/'), 'next=/t%C3%ABst%26key/')
  73. def test_mutable_copy(self):
  74. """A copy of a QueryDict is mutable."""
  75. q = QueryDict('').copy()
  76. self.assertRaises(KeyError, q.__getitem__, "foo")
  77. q['name'] = 'john'
  78. self.assertEqual(q['name'], 'john')
  79. def test_mutable_delete(self):
  80. q = QueryDict('').copy()
  81. q['name'] = 'john'
  82. del q['name']
  83. self.assertFalse('name' in q)
  84. def test_basic_mutable_operations(self):
  85. q = QueryDict('').copy()
  86. q['name'] = 'john'
  87. self.assertEqual(q.get('foo', 'default'), 'default')
  88. self.assertEqual(q.get('name', 'default'), 'john')
  89. self.assertEqual(q.getlist('name'), ['john'])
  90. self.assertEqual(q.getlist('foo'), [])
  91. q.setlist('foo', ['bar', 'baz'])
  92. self.assertEqual(q.get('foo', 'default'), 'baz')
  93. self.assertEqual(q.getlist('foo'), ['bar', 'baz'])
  94. q.appendlist('foo', 'another')
  95. self.assertEqual(q.getlist('foo'), ['bar', 'baz', 'another'])
  96. self.assertEqual(q['foo'], 'another')
  97. self.assertTrue(q.has_key('foo'))
  98. self.assertTrue('foo' in q)
  99. self.assertEqual(q.items(), [(u'foo', u'another'), (u'name', u'john')])
  100. self.assertEqual(q.lists(), [(u'foo', [u'bar', u'baz', u'another']), (u'name', [u'john'])])
  101. self.assertEqual(q.keys(), [u'foo', u'name'])
  102. self.assertEqual(q.values(), [u'another', u'john'])
  103. self.assertEqual(len(q), 2)
  104. q.update({'foo': 'hello'})
  105. self.assertEqual(q['foo'], 'hello')
  106. self.assertEqual(q.get('foo', 'not available'), 'hello')
  107. self.assertEqual(q.getlist('foo'), [u'bar', u'baz', u'another', u'hello'])
  108. self.assertEqual(q.pop('foo'), [u'bar', u'baz', u'another', u'hello'])
  109. self.assertEqual(q.pop('foo', 'not there'), 'not there')
  110. self.assertEqual(q.get('foo', 'not there'), 'not there')
  111. self.assertEqual(q.setdefault('foo', 'bar'), 'bar')
  112. self.assertEqual(q['foo'], 'bar')
  113. self.assertEqual(q.getlist('foo'), ['bar'])
  114. self.assertEqual(q.urlencode(), 'foo=bar&name=john')
  115. q.clear()
  116. self.assertEqual(len(q), 0)
  117. def test_multiple_keys(self):
  118. """Test QueryDict with two key/value pairs with same keys."""
  119. q = QueryDict('vote=yes&vote=no')
  120. self.assertEqual(q['vote'], u'no')
  121. self.assertRaises(AttributeError, q.__setitem__, 'something', 'bar')
  122. self.assertEqual(q.get('vote', 'default'), u'no')
  123. self.assertEqual(q.get('foo', 'default'), 'default')
  124. self.assertEqual(q.getlist('vote'), [u'yes', u'no'])
  125. self.assertEqual(q.getlist('foo'), [])
  126. self.assertRaises(AttributeError, q.setlist, 'foo', ['bar', 'baz'])
  127. self.assertRaises(AttributeError, q.setlist, 'foo', ['bar', 'baz'])
  128. self.assertRaises(AttributeError, q.appendlist, 'foo', ['bar'])
  129. self.assertEqual(q.has_key('vote'), True)
  130. self.assertEqual('vote' in q, True)
  131. self.assertEqual(q.has_key('foo'), False)
  132. self.assertEqual('foo' in q, False)
  133. self.assertEqual(q.items(), [(u'vote', u'no')])
  134. self.assertEqual(q.lists(), [(u'vote', [u'yes', u'no'])])
  135. self.assertEqual(q.keys(), [u'vote'])
  136. self.assertEqual(q.values(), [u'no'])
  137. self.assertEqual(len(q), 1)
  138. self.assertRaises(AttributeError, q.update, {'foo': 'bar'})
  139. self.assertRaises(AttributeError, q.pop, 'foo')
  140. self.assertRaises(AttributeError, q.popitem)
  141. self.assertRaises(AttributeError, q.clear)
  142. self.assertRaises(AttributeError, q.setdefault, 'foo', 'bar')
  143. self.assertRaises(AttributeError, q.__delitem__, 'vote')
  144. def test_invalid_input_encoding(self):
  145. """
  146. QueryDicts must be able to handle invalid input encoding (in this
  147. case, bad UTF-8 encoding).
  148. """
  149. q = QueryDict('foo=bar&foo=\xff')
  150. self.assertEqual(q['foo'], u'\ufffd')
  151. self.assertEqual(q.getlist('foo'), [u'bar', u'\ufffd'])
  152. def test_pickle(self):
  153. q = QueryDict('')
  154. q1 = pickle.loads(pickle.dumps(q, 2))
  155. self.assertEqual(q == q1, True)
  156. q = QueryDict('a=b&c=d')
  157. q1 = pickle.loads(pickle.dumps(q, 2))
  158. self.assertEqual(q == q1, True)
  159. q = QueryDict('a=b&c=d&a=1')
  160. q1 = pickle.loads(pickle.dumps(q, 2))
  161. self.assertEqual(q == q1 , True)
  162. def test_update_from_querydict(self):
  163. """Regression test for #8278: QueryDict.update(QueryDict)"""
  164. x = QueryDict("a=1&a=2", mutable=True)
  165. y = QueryDict("a=3&a=4")
  166. x.update(y)
  167. self.assertEqual(x.getlist('a'), [u'1', u'2', u'3', u'4'])
  168. def test_non_default_encoding(self):
  169. """#13572 - QueryDict with a non-default encoding"""
  170. q = QueryDict('sbb=one', encoding='rot_13')
  171. self.assertEqual(q.encoding , 'rot_13' )
  172. self.assertEqual(q.items() , [(u'foo', u'bar')] )
  173. self.assertEqual(q.urlencode() , 'sbb=one' )
  174. q = q.copy()
  175. self.assertEqual(q.encoding , 'rot_13' )
  176. self.assertEqual(q.items() , [(u'foo', u'bar')] )
  177. self.assertEqual(q.urlencode() , 'sbb=one' )
  178. self.assertEqual(copy.copy(q).encoding , 'rot_13' )
  179. self.assertEqual(copy.deepcopy(q).encoding , 'rot_13')
  180. class HttpResponseTests(unittest.TestCase):
  181. def test_unicode_headers(self):
  182. r = HttpResponse()
  183. # If we insert a unicode value it will be converted to an ascii
  184. r['value'] = u'test value'
  185. self.assertTrue(isinstance(r['value'], str))
  186. # An error is raised ~hen a unicode object with non-ascii is assigned.
  187. self.assertRaises(UnicodeEncodeError, r.__setitem__, 'value', u't\xebst value')
  188. # An error is raised when a unicode object with non-ASCII format is
  189. # passed as initial mimetype or content_type.
  190. self.assertRaises(UnicodeEncodeError, HttpResponse,
  191. mimetype=u't\xebst value')
  192. # HttpResponse headers must be convertible to ASCII.
  193. self.assertRaises(UnicodeEncodeError, HttpResponse,
  194. content_type=u't\xebst value')
  195. # The response also converts unicode keys to strings.)
  196. r[u'test'] = 'testing key'
  197. l = list(r.items())
  198. l.sort()
  199. self.assertEqual(l[1], ('test', 'testing key'))
  200. # It will also raise errors for keys with non-ascii data.
  201. self.assertRaises(UnicodeEncodeError, r.__setitem__, u't\xebst key', 'value')
  202. def test_newlines_in_headers(self):
  203. # Bug #10188: Do not allow newlines in headers (CR or LF)
  204. r = HttpResponse()
  205. self.assertRaises(BadHeaderError, r.__setitem__, 'test\rstr', 'test')
  206. self.assertRaises(BadHeaderError, r.__setitem__, 'test\nstr', 'test')
  207. def test_unsafe_redirects(self):
  208. bad_urls = [
  209. 'data:text/html,<script>window.alert("xss")</script>',
  210. 'mailto:test@example.com',
  211. 'file:///etc/passwd',
  212. ]
  213. for url in bad_urls:
  214. self.assertRaises(SuspiciousOperation,
  215. HttpResponseRedirect, url)
  216. self.assertRaises(SuspiciousOperation,
  217. HttpResponsePermanentRedirect, url)
  218. class CookieTests(unittest.TestCase):
  219. def test_encode(self):
  220. """
  221. Test that we don't output tricky characters in encoded value
  222. """
  223. # Python 2.4 compatibility note: Python 2.4's cookie implementation
  224. # always returns Set-Cookie headers terminating in semi-colons.
  225. # That's not the bug this test is looking for, so ignore it.
  226. c = SimpleCookie()
  227. c['test'] = "An,awkward;value"
  228. self.assertTrue(";" not in c.output().rstrip(';')) # IE compat
  229. self.assertTrue("," not in c.output().rstrip(';')) # Safari compat
  230. def test_decode(self):
  231. """
  232. Test that we can still preserve semi-colons and commas
  233. """
  234. c = SimpleCookie()
  235. c['test'] = "An,awkward;value"
  236. c2 = SimpleCookie()
  237. c2.load(c.output())
  238. self.assertEqual(c['test'].value, c2['test'].value)
  239. def test_decode_2(self):
  240. """
  241. Test that we haven't broken normal encoding
  242. """
  243. c = SimpleCookie()
  244. c['test'] = "\xf0"
  245. c2 = SimpleCookie()
  246. c2.load(c.output())
  247. self.assertEqual(c['test'].value, c2['test'].value)
  248. def test_nonstandard_keys(self):
  249. """
  250. Test that a single non-standard cookie name doesn't affect all cookies. Ticket #13007.
  251. """
  252. self.assertTrue('good_cookie' in parse_cookie('good_cookie=yes;bad:cookie=yes').keys())