Migrated requests doctests. Thanks to Stephan Jaekel.

git-svn-id: http://code.djangoproject.com/svn/django/trunk@13927 bcc190cf-cafb-0310-a4f2-bffc1f526a37

tests/regressiontests/requests/tests.py

@@ -1,81 +1,92 @@
-"""
+from datetime import datetime, timedelta
->>> from django.http import HttpRequest, HttpResponse
+import time
->>> print repr(HttpRequest())
+import unittest
-<HttpRequest
-GET:{},
-POST:{},
-COOKIES:{},
-META:{}>
 
->>> from django.core.handlers.wsgi import WSGIRequest
+from django.http import HttpRequest, HttpResponse, parse_cookie
->>> print repr(WSGIRequest({'PATH_INFO': 'bogus', 'REQUEST_METHOD': 'bogus'}))
+from django.core.handlers.wsgi import WSGIRequest
-<WSGIRequest
+from django.core.handlers.modpython import ModPythonRequest
-GET:<QueryDict: {}>,
+from django.utils.http import cookie_date
-POST:<QueryDict: {}>,
-COOKIES:{},
-META:{...}>
 
->>> from django.core.handlers.modpython import ModPythonRequest
+class RequestsTests(unittest.TestCase):
->>> class FakeModPythonRequest(ModPythonRequest):
-...    def __init__(self, *args, **kwargs):
-...        super(FakeModPythonRequest, self).__init__(*args, **kwargs)
-...        self._get = self._post = self._meta = self._cookies = {}
->>> class Dummy:
-...     def get_options(self):
-...         return {}
->>> req = Dummy()
->>> req.uri = 'bogus'
->>> print repr(FakeModPythonRequest(req))
-<ModPythonRequest
-path:bogus,
-GET:{},
-POST:{},
-COOKIES:{},
-META:{}>
 
->>> from django.http import parse_cookie
+    def test_httprequest(self):
->>> parse_cookie('invalid:key=true')
+        self.assertEquals(repr(HttpRequest()),
-{}
+            "<HttpRequest\n"\
+            "GET:{},\n"\
+            "POST:{},\n"\
+            "COOKIES:{},\n"\
+            "META:{}>"
+        )
 
->>> request = HttpRequest()
+    def test_wsgirequest(self):
->>> print request.build_absolute_uri(location="https://www.example.com/asdf")
+        self.assertEquals(repr(WSGIRequest({'PATH_INFO': 'bogus', 'REQUEST_METHOD': 'bogus'})),
-https://www.example.com/asdf
+            "<WSGIRequest\n"\
->>> request.get_host = lambda: 'www.example.com'
+            "GET:<QueryDict: {}>,\n"\
->>> request.path = ''
+            "POST:<QueryDict: {}>,\n"\
->>> print request.build_absolute_uri(location="/path/with:colons")
+            "COOKIES:{},\n"\
-http://www.example.com/path/with:colons
+            "META:{'PATH_INFO': u'bogus', 'REQUEST_METHOD': 'bogus', 'SCRIPT_NAME': u''}>"
+        )
 
+    def test_modpythonrequest(self):
+        class FakeModPythonRequest(ModPythonRequest):
+           def __init__(self, *args, **kwargs):
+               super(FakeModPythonRequest, self).__init__(*args, **kwargs)
+               self._get = self._post = self._meta = self._cookies = {}
 
-# Test cookie datetime expiration logic
+        class Dummy:
->>> from datetime import datetime, timedelta
+            def get_options(self):
->>> import time
+                return {}
->>> delta = timedelta(seconds=10)
->>> response = HttpResponse()
->>> expires = datetime.utcnow() + delta
 
-# There is a timing weakness in this test; The
+        req = Dummy()
-# expected result for max-age requires that there be
+        req.uri = 'bogus'
-# a very slight difference between the evaluated expiration
+        self.assertEquals(repr(FakeModPythonRequest(req)),
-# time, and the time evaluated in set_cookie(). If this
+            "<ModPythonRequest\n"\
-# difference doesn't exist, the cookie time will be
+            "path:bogus,\n"\
-# 1 second larger. To avoid the problem, put in a quick sleep,
+            "GET:{},\n"\
-# which guarantees that there will be a time difference.
+            "POST:{},\n"\
->>> time.sleep(0.001)
+            "COOKIES:{},\n"\
->>> response.set_cookie('datetime', expires=expires)
+            "META:{}>")
->>> datetime_cookie = response.cookies['datetime']
->>> datetime_cookie['max-age']
-10
->>> response.set_cookie('datetime', expires=datetime(2028, 1, 1, 4, 5, 6))
->>> response.cookies['datetime']['expires']
-'Sat, 01-Jan-2028 04:05:06 GMT'
 
-# Test automatically setting cookie expires if only max_age is provided
+    def test_parse_cookie(self):
->>> response.set_cookie('max_age', max_age=10)
+        self.assertEquals(parse_cookie('invalid:key=true'), {})
->>> max_age_cookie = response.cookies['max_age']
+
->>> max_age_cookie['max-age']
+    def test_httprequest_location(self):
-10
+        request = HttpRequest()
->>> from django.utils.http import cookie_date
+        self.assertEquals(request.build_absolute_uri(location="https://www.example.com/asdf"),
->>> import time
+            'https://www.example.com/asdf')
->>> max_age_cookie['expires'] == cookie_date(time.time()+10)
+
-True
+        request.get_host = lambda: 'www.example.com'
-"""
+        request.path = ''
+        self.assertEquals(request.build_absolute_uri(location="/path/with:colons"),
+            'http://www.example.com/path/with:colons')
+
+    def test_near_expiration(self):
+        "Cookie will expire when an near expiration time is provided"
+        response = HttpResponse()
+        # There is a timing weakness in this test; The
+        # expected result for max-age requires that there be
+        # a very slight difference between the evaluated expiration
+        # time, and the time evaluated in set_cookie(). If this
+        # difference doesn't exist, the cookie time will be
+        # 1 second larger. To avoid the problem, put in a quick sleep,
+        # which guarantees that there will be a time difference.
+        expires = datetime.utcnow() + timedelta(seconds=10)
+        time.sleep(0.001)
+        response.set_cookie('datetime', expires=expires)
+        datetime_cookie = response.cookies['datetime']
+        self.assertEquals(datetime_cookie['max-age'], 10)
+
+    def test_far_expiration(self):
+        "Cookie will expire when an distant expiration time is provided"
+        response = HttpResponse()
+        response.set_cookie('datetime', expires=datetime(2028, 1, 1, 4, 5, 6))
+        datetime_cookie = response.cookies['datetime']
+        self.assertEquals(datetime_cookie['expires'], 'Sat, 01-Jan-2028 04:05:06 GMT')
+
+    def test_max_age_expiration(self):
+        "Cookie will expire if max_age is provided"
+        response = HttpResponse()
+        response.set_cookie('max_age', max_age=10)
+        max_age_cookie = response.cookies['max_age']
+        self.assertEqual(max_age_cookie['max-age'], 10)
+        self.assertEqual(max_age_cookie['expires'], cookie_date(time.time()+10))