diff options
Diffstat (limited to 'tests/test_client_regress/views.py')
-rw-r--r-- | tests/test_client_regress/views.py | 108 |
1 files changed, 108 insertions, 0 deletions
diff --git a/tests/test_client_regress/views.py b/tests/test_client_regress/views.py new file mode 100644 index 0000000000..9b0654806b --- /dev/null +++ b/tests/test_client_regress/views.py @@ -0,0 +1,108 @@ +import json + +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.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 = json.loads(request.body.decode(charset)) + obj_json = json.dumps(obj_dict, cls=DjangoJSONEncoder, ensure_ascii=False) + response = HttpResponse(obj_json.encode(charset), status=200, + content_type='application/json; charset=%s' % 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 body(request): + "A view that is requested with GET and accesses request.body. Refs #14753." + return HttpResponse(request.body) + +def read_all(request): + "A view that is requested with accesses request.read()." + return HttpResponse(request.read()) + +def read_buffer(request): + "A view that is requested with accesses request.read(LARGE_BUFFER)." + return HttpResponse(request.read(99999)) + +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, {})) |