tests/regressiontests/test_client_regress/views.py

   1 from django.contrib.auth.decorators import login_required
   2 from django.http import HttpResponse, HttpResponseRedirect
   3 from django.core.exceptions import SuspiciousOperation
   4
   5 def no_template_view(request):
   6     "A simple view that expects a GET request, and returns a rendered template"
   7     return HttpResponse("No template used. Sample content: twice once twice. Content ends.")
   8
   9 def staff_only_view(request):
  10     "A view that can only be visited by staff. Non staff members get an exception"
  11     if request.user.is_staff:
  12         return HttpResponse('')
  13     else:
  14         raise SuspiciousOperation()
  15
  16 def get_view(request):
  17     "A simple login protected view"
  18     return HttpResponse("Hello world")
  19 get_view = login_required(get_view)
  20
  21 def view_with_argument(request, name):
  22     """A view that takes a string argument
  23
  24     The purpose of this view is to check that if a space is provided in
  25     the argument, the test framework unescapes the %20 before passing
  26     the value to the view.
  27     """
  28     if name == 'Arthur Dent':
  29         return HttpResponse('Hi, Arthur')
  30     else:
  31         return HttpResponse('Howdy, %s' % name)
  32
  33 def login_protected_redirect_view(request):
  34     "A view that redirects all requests to the GET view"
  35     return HttpResponseRedirect('/test_client_regress/get_view/')
  36 login_protected_redirect_view = login_required(login_protected_redirect_view)
  37
  38 def set_session_view(request):
  39     "A view that sets a session variable"
  40     request.session['session_var'] = 'YES'
  41     return HttpResponse('set_session')
  42
  43 def check_session_view(request):
  44     "A view that reads a session variable"
  45     return HttpResponse(request.session.get('session_var', 'NO'))