/tests/regressiontests/generic_views/base.py
https://code.google.com/p/mango-py/ · Python · 293 lines · 256 code · 24 blank · 13 comment · 0 complexity · 74ba569ae5c2065a24761afea8485036 MD5 · raw file
- import time
- import unittest
- from django.core.exceptions import ImproperlyConfigured
- from django.http import HttpResponse
- from django.test import TestCase, RequestFactory
- from django.utils import simplejson
- from django.views.generic import View, TemplateView, RedirectView
- class SimpleView(View):
- """
- A simple view with a docstring.
- """
- def get(self, request):
- return HttpResponse('This is a simple view')
- class SimplePostView(SimpleView):
- post = SimpleView.get
- class CustomizableView(SimpleView):
- parameter = {}
- def decorator(view):
- view.is_decorated = True
- return view
- class DecoratedDispatchView(SimpleView):
- @decorator
- def dispatch(self, request, *args, **kwargs):
- return super(DecoratedDispatchView, self).dispatch(request, *args, **kwargs)
- class AboutTemplateView(TemplateView):
- def get(self, request):
- return self.render_to_response({})
- def get_template_names(self):
- return ['generic_views/about.html']
- class AboutTemplateAttributeView(TemplateView):
- template_name = 'generic_views/about.html'
- def get(self, request):
- return self.render_to_response(context={})
- class InstanceView(View):
- def get(self, request):
- return self
- class ViewTest(unittest.TestCase):
- rf = RequestFactory()
- def _assert_simple(self, response):
- self.assertEqual(response.status_code, 200)
- self.assertEqual(response.content, 'This is a simple view')
- def test_no_init_kwargs(self):
- """
- Test that a view can't be accidentally instantiated before deployment
- """
- try:
- view = SimpleView(key='value').as_view()
- self.fail('Should not be able to instantiate a view')
- except AttributeError:
- pass
- def test_no_init_args(self):
- """
- Test that a view can't be accidentally instantiated before deployment
- """
- try:
- view = SimpleView.as_view('value')
- self.fail('Should not be able to use non-keyword arguments instantiating a view')
- except TypeError:
- pass
- def test_pathological_http_method(self):
- """
- The edge case of a http request that spoofs an existing method name is caught.
- """
- self.assertEqual(SimpleView.as_view()(
- self.rf.get('/', REQUEST_METHOD='DISPATCH')
- ).status_code, 405)
- def test_get_only(self):
- """
- Test a view which only allows GET doesn't allow other methods.
- """
- self._assert_simple(SimpleView.as_view()(self.rf.get('/')))
- self.assertEqual(SimpleView.as_view()(self.rf.post('/')).status_code, 405)
- self.assertEqual(SimpleView.as_view()(
- self.rf.get('/', REQUEST_METHOD='FAKE')
- ).status_code, 405)
- def test_get_and_post(self):
- """
- Test a view which only allows both GET and POST.
- """
- self._assert_simple(SimplePostView.as_view()(self.rf.get('/')))
- self._assert_simple(SimplePostView.as_view()(self.rf.post('/')))
- self.assertEqual(SimplePostView.as_view()(
- self.rf.get('/', REQUEST_METHOD='FAKE')
- ).status_code, 405)
- def test_invalid_keyword_argument(self):
- """
- Test that view arguments must be predefined on the class and can't
- be named like a HTTP method.
- """
- # Check each of the allowed method names
- for method in SimpleView.http_method_names:
- kwargs = dict(((method, "value"),))
- self.assertRaises(TypeError, SimpleView.as_view, **kwargs)
- # Check the case view argument is ok if predefined on the class...
- CustomizableView.as_view(parameter="value")
- # ...but raises errors otherwise.
- self.assertRaises(TypeError, CustomizableView.as_view, foobar="value")
- def test_calling_more_than_once(self):
- """
- Test a view can only be called once.
- """
- request = self.rf.get('/')
- view = InstanceView.as_view()
- self.assertNotEqual(view(request), view(request))
- def test_class_attributes(self):
- """
- Test that the callable returned from as_view() has proper
- docstring, name and module.
- """
- self.assertEqual(SimpleView.__doc__, SimpleView.as_view().__doc__)
- self.assertEqual(SimpleView.__name__, SimpleView.as_view().__name__)
- self.assertEqual(SimpleView.__module__, SimpleView.as_view().__module__)
- def test_dispatch_decoration(self):
- """
- Test that attributes set by decorators on the dispatch method
- are also present on the closure.
- """
- self.assertTrue(DecoratedDispatchView.as_view().is_decorated)
- class TemplateViewTest(TestCase):
- urls = 'regressiontests.generic_views.urls'
- rf = RequestFactory()
- def _assert_about(self, response):
- response.render()
- self.assertEqual(response.status_code, 200)
- self.assertContains(response, '<h1>About</h1>')
- def test_get(self):
- """
- Test a view that simply renders a template on GET
- """
- self._assert_about(AboutTemplateView.as_view()(self.rf.get('/about/')))
- def test_get_template_attribute(self):
- """
- Test a view that renders a template on GET with the template name as
- an attribute on the class.
- """
- self._assert_about(AboutTemplateAttributeView.as_view()(self.rf.get('/about/')))
- def test_get_generic_template(self):
- """
- Test a completely generic view that renders a template on GET
- with the template name as an argument at instantiation.
- """
- self._assert_about(TemplateView.as_view(template_name='generic_views/about.html')(self.rf.get('/about/')))
- def test_template_name_required(self):
- """
- A template view must provide a template name
- """
- self.assertRaises(ImproperlyConfigured, self.client.get, '/template/no_template/')
- def test_template_params(self):
- """
- A generic template view passes kwargs as context.
- """
- response = self.client.get('/template/simple/bar/')
- self.assertEqual(response.status_code, 200)
- self.assertEqual(response.context['params'], {'foo': 'bar'})
- def test_extra_template_params(self):
- """
- A template view can be customized to return extra context.
- """
- response = self.client.get('/template/custom/bar/')
- self.assertEqual(response.status_code, 200)
- self.assertEqual(response.context['params'], {'foo': 'bar'})
- self.assertEqual(response.context['key'], 'value')
- def test_cached_views(self):
- """
- A template view can be cached
- """
- response = self.client.get('/template/cached/bar/')
- self.assertEqual(response.status_code, 200)
- time.sleep(1.0)
- response2 = self.client.get('/template/cached/bar/')
- self.assertEqual(response2.status_code, 200)
- self.assertEqual(response.content, response2.content)
- time.sleep(2.0)
- # Let the cache expire and test again
- response2 = self.client.get('/template/cached/bar/')
- self.assertEqual(response2.status_code, 200)
- self.assertNotEqual(response.content, response2.content)
- class RedirectViewTest(unittest.TestCase):
- rf = RequestFactory()
- def test_no_url(self):
- "Without any configuration, returns HTTP 410 GONE"
- response = RedirectView.as_view()(self.rf.get('/foo/'))
- self.assertEqual(response.status_code, 410)
- def test_permanaent_redirect(self):
- "Default is a permanent redirect"
- response = RedirectView.as_view(url='/bar/')(self.rf.get('/foo/'))
- self.assertEqual(response.status_code, 301)
- self.assertEqual(response['Location'], '/bar/')
- def test_temporary_redirect(self):
- "Permanent redirects are an option"
- response = RedirectView.as_view(url='/bar/', permanent=False)(self.rf.get('/foo/'))
- self.assertEqual(response.status_code, 302)
- self.assertEqual(response['Location'], '/bar/')
- def test_include_args(self):
- "GET arguments can be included in the redirected URL"
- response = RedirectView.as_view(url='/bar/')(self.rf.get('/foo/'))
- self.assertEqual(response.status_code, 301)
- self.assertEqual(response['Location'], '/bar/')
- response = RedirectView.as_view(url='/bar/', query_string=True)(self.rf.get('/foo/?pork=spam'))
- self.assertEqual(response.status_code, 301)
- self.assertEqual(response['Location'], '/bar/?pork=spam')
- def test_parameter_substitution(self):
- "Redirection URLs can be parameterized"
- response = RedirectView.as_view(url='/bar/%(object_id)d/')(self.rf.get('/foo/42/'), object_id=42)
- self.assertEqual(response.status_code, 301)
- self.assertEqual(response['Location'], '/bar/42/')
- def test_redirect_POST(self):
- "Default is a permanent redirect"
- response = RedirectView.as_view(url='/bar/')(self.rf.post('/foo/'))
- self.assertEqual(response.status_code, 301)
- self.assertEqual(response['Location'], '/bar/')
- def test_redirect_HEAD(self):
- "Default is a permanent redirect"
- response = RedirectView.as_view(url='/bar/')(self.rf.head('/foo/'))
- self.assertEqual(response.status_code, 301)
- self.assertEqual(response['Location'], '/bar/')
- def test_redirect_OPTIONS(self):
- "Default is a permanent redirect"
- response = RedirectView.as_view(url='/bar/')(self.rf.options('/foo/'))
- self.assertEqual(response.status_code, 301)
- self.assertEqual(response['Location'], '/bar/')
- def test_redirect_PUT(self):
- "Default is a permanent redirect"
- response = RedirectView.as_view(url='/bar/')(self.rf.put('/foo/'))
- self.assertEqual(response.status_code, 301)
- self.assertEqual(response['Location'], '/bar/')
- def test_redirect_DELETE(self):
- "Default is a permanent redirect"
- response = RedirectView.as_view(url='/bar/')(self.rf.delete('/foo/'))
- self.assertEqual(response.status_code, 301)
- self.assertEqual(response['Location'], '/bar/')