/tests/regressiontests/test_client_regress/views.py
https://code.google.com/p/mango-py/ · Python · 102 lines · 79 code · 15 blank · 8 comment · 7 complexity · d5af1e268daeb093ca485280e533abc6 MD5 · raw file
- from django.conf import settings
- from django.contrib.auth.decorators import login_required
- from django.http import HttpResponse, HttpResponseRedirect
- from django.core.exceptions import SuspiciousOperation
- from django.shortcuts import render_to_response
- from django.utils import simplejson
- from django.utils.encoding import smart_str
- from django.core.serializers.json import DjangoJSONEncoder
- from django.test.client import CONTENT_TYPE_RE
- from django.template import RequestContext
- def no_template_view(request):
- "A simple view that expects a GET request, and returns a rendered template"
- return HttpResponse("No template used. Sample content: twice once twice. Content ends.")
- def staff_only_view(request):
- "A view that can only be visited by staff. Non staff members get an exception"
- if request.user.is_staff:
- return HttpResponse('')
- else:
- raise SuspiciousOperation()
- def get_view(request):
- "A simple login protected view"
- return HttpResponse("Hello world")
- get_view = login_required(get_view)
- def request_data(request, template='base.html', data='sausage'):
- "A simple view that returns the request data in the context"
- return render_to_response(template, {
- 'get-foo':request.GET.get('foo',None),
- 'get-bar':request.GET.get('bar',None),
- 'post-foo':request.POST.get('foo',None),
- 'post-bar':request.POST.get('bar',None),
- 'request-foo':request.REQUEST.get('foo',None),
- 'request-bar':request.REQUEST.get('bar',None),
- 'data': data,
- })
- def view_with_argument(request, name):
- """A view that takes a string argument
- The purpose of this view is to check that if a space is provided in
- the argument, the test framework unescapes the %20 before passing
- the value to the view.
- """
- if name == 'Arthur Dent':
- return HttpResponse('Hi, Arthur')
- else:
- return HttpResponse('Howdy, %s' % name)
- def login_protected_redirect_view(request):
- "A view that redirects all requests to the GET view"
- return HttpResponseRedirect('/test_client_regress/get_view/')
- login_protected_redirect_view = login_required(login_protected_redirect_view)
- def set_session_view(request):
- "A view that sets a session variable"
- request.session['session_var'] = 'YES'
- return HttpResponse('set_session')
- def check_session_view(request):
- "A view that reads a session variable"
- return HttpResponse(request.session.get('session_var', 'NO'))
- def request_methods_view(request):
- "A view that responds with the request method"
- return HttpResponse('request method: %s' % request.method)
- def return_unicode(request):
- return render_to_response('unicode.html')
- def return_json_file(request):
- "A view that parses and returns a JSON string as a file."
- match = CONTENT_TYPE_RE.match(request.META['CONTENT_TYPE'])
- if match:
- charset = match.group(1)
- else:
- charset = settings.DEFAULT_CHARSET
- # This just checks that the uploaded data is JSON
- obj_dict = simplejson.loads(request.raw_post_data.decode(charset))
- obj_json = simplejson.dumps(obj_dict, encoding=charset,
- cls=DjangoJSONEncoder,
- ensure_ascii=False)
- response = HttpResponse(smart_str(obj_json, encoding=charset), status=200,
- mimetype='application/json; charset=' + charset)
- response['Content-Disposition'] = 'attachment; filename=testfile.json'
- return response
- def check_headers(request):
- "A view that responds with value of the X-ARG-CHECK header"
- return HttpResponse('HTTP_X_ARG_CHECK: %s' % request.META.get('HTTP_X_ARG_CHECK', 'Undefined'))
- def raw_post_data(request):
- "A view that is requested with GET and accesses request.raw_post_data. Refs #14753."
- return HttpResponse(request.raw_post_data)
- def request_context_view(request):
- # Special attribute that won't be present on a plain HttpRequest
- request.special_path = request.path
- return render_to_response('request_context.html', context_instance=RequestContext(request, {}))