2012-05-31 23:21:13 +08:00
# -*- coding: utf-8 -*-
"""
Regression tests for the Test Client , especially the customized assertions .
"""
2012-06-08 00:08:47 +08:00
from __future__ import unicode_literals
2012-05-31 23:21:13 +08:00
import os
2013-10-29 03:34:09 +08:00
import itertools
2014-12-14 17:17:18 +08:00
import warnings
2012-05-31 23:21:13 +08:00
2014-01-14 23:43:27 +08:00
from django . core . urlresolvers import reverse , NoReverseMatch
2014-11-16 03:58:26 +08:00
from django . template import TemplateSyntaxError , Context , Template
2013-12-23 23:01:13 +08:00
from django . test import Client , TestCase , override_settings
2014-10-17 21:46:42 +08:00
from django . test . client import RedirectCycleError , RequestFactory , encode_file
2013-12-23 23:01:13 +08:00
from django . test . utils import ContextList , str_prefix
2012-05-31 23:21:13 +08:00
from django . template . response import SimpleTemplateResponse
2012-12-08 18:13:52 +08:00
from django . utils . _os import upath
2014-12-14 17:17:18 +08:00
from django . utils . deprecation import RemovedInDjango20Warning
2012-09-19 02:58:40 +08:00
from django . utils . translation import ugettext_lazy
2012-05-31 23:21:13 +08:00
from django . http import HttpResponse
2013-06-05 00:37:44 +08:00
from django . contrib . auth . signals import user_logged_out , user_logged_in
from django . contrib . auth . models import User
2012-05-31 23:21:13 +08:00
2013-06-05 00:37:44 +08:00
from . models import CustomUser
2013-05-16 07:14:28 +08:00
from . views import CustomTestException
2012-05-31 23:21:13 +08:00
2013-11-03 05:34:05 +08:00
2014-12-18 05:51:42 +08:00
@override_settings ( ROOT_URLCONF = ' test_client_regress.urls ' )
2012-05-31 23:21:13 +08:00
class AssertContainsTests ( TestCase ) :
2014-01-14 23:43:27 +08:00
2012-05-31 23:21:13 +08:00
def test_contains ( self ) :
" Responses can be inspected for content, including counting repeated substrings "
2014-01-14 23:43:27 +08:00
response = self . client . get ( ' /no_template_view/ ' )
2012-05-31 23:21:13 +08:00
self . assertNotContains ( response , ' never ' )
self . assertContains ( response , ' never ' , 0 )
self . assertContains ( response , ' once ' )
self . assertContains ( response , ' once ' , 1 )
self . assertContains ( response , ' twice ' )
self . assertContains ( response , ' twice ' , 2 )
try :
self . assertContains ( response , ' text ' , status_code = 999 )
except AssertionError as e :
self . assertIn ( " Couldn ' t retrieve content: Response code was 200 (expected 999) " , str ( e ) )
try :
self . assertContains ( response , ' text ' , status_code = 999 , msg_prefix = ' abc ' )
except AssertionError as e :
self . assertIn ( " abc: Couldn ' t retrieve content: Response code was 200 (expected 999) " , str ( e ) )
try :
self . assertNotContains ( response , ' text ' , status_code = 999 )
except AssertionError as e :
self . assertIn ( " Couldn ' t retrieve content: Response code was 200 (expected 999) " , str ( e ) )
try :
self . assertNotContains ( response , ' text ' , status_code = 999 , msg_prefix = ' abc ' )
except AssertionError as e :
self . assertIn ( " abc: Couldn ' t retrieve content: Response code was 200 (expected 999) " , str ( e ) )
try :
self . assertNotContains ( response , ' once ' )
except AssertionError as e :
self . assertIn ( " Response should not contain ' once ' " , str ( e ) )
try :
self . assertNotContains ( response , ' once ' , msg_prefix = ' abc ' )
except AssertionError as e :
self . assertIn ( " abc: Response should not contain ' once ' " , str ( e ) )
try :
self . assertContains ( response , ' never ' , 1 )
except AssertionError as e :
self . assertIn ( " Found 0 instances of ' never ' in response (expected 1) " , str ( e ) )
try :
self . assertContains ( response , ' never ' , 1 , msg_prefix = ' abc ' )
except AssertionError as e :
self . assertIn ( " abc: Found 0 instances of ' never ' in response (expected 1) " , str ( e ) )
try :
self . assertContains ( response , ' once ' , 0 )
except AssertionError as e :
self . assertIn ( " Found 1 instances of ' once ' in response (expected 0) " , str ( e ) )
try :
self . assertContains ( response , ' once ' , 0 , msg_prefix = ' abc ' )
except AssertionError as e :
self . assertIn ( " abc: Found 1 instances of ' once ' in response (expected 0) " , str ( e ) )
try :
self . assertContains ( response , ' once ' , 2 )
except AssertionError as e :
self . assertIn ( " Found 1 instances of ' once ' in response (expected 2) " , str ( e ) )
try :
self . assertContains ( response , ' once ' , 2 , msg_prefix = ' abc ' )
except AssertionError as e :
self . assertIn ( " abc: Found 1 instances of ' once ' in response (expected 2) " , str ( e ) )
try :
self . assertContains ( response , ' twice ' , 1 )
except AssertionError as e :
self . assertIn ( " Found 2 instances of ' twice ' in response (expected 1) " , str ( e ) )
try :
self . assertContains ( response , ' twice ' , 1 , msg_prefix = ' abc ' )
except AssertionError as e :
self . assertIn ( " abc: Found 2 instances of ' twice ' in response (expected 1) " , str ( e ) )
try :
self . assertContains ( response , ' thrice ' )
except AssertionError as e :
self . assertIn ( " Couldn ' t find ' thrice ' in response " , str ( e ) )
try :
self . assertContains ( response , ' thrice ' , msg_prefix = ' abc ' )
except AssertionError as e :
self . assertIn ( " abc: Couldn ' t find ' thrice ' in response " , str ( e ) )
try :
self . assertContains ( response , ' thrice ' , 3 )
except AssertionError as e :
self . assertIn ( " Found 0 instances of ' thrice ' in response (expected 3) " , str ( e ) )
try :
self . assertContains ( response , ' thrice ' , 3 , msg_prefix = ' abc ' )
except AssertionError as e :
self . assertIn ( " abc: Found 0 instances of ' thrice ' in response (expected 3) " , str ( e ) )
def test_unicode_contains ( self ) :
" Unicode characters can be found in template context "
2013-11-03 05:02:56 +08:00
# Regression test for #10183
2014-01-14 23:43:27 +08:00
r = self . client . get ( ' /check_unicode/ ' )
2012-06-08 00:08:47 +08:00
self . assertContains ( r , ' さかき ' )
2012-05-31 23:21:13 +08:00
self . assertContains ( r , b ' \xe5 \xb3 \xa0 ' . decode ( ' utf-8 ' ) )
def test_unicode_not_contains ( self ) :
" Unicode characters can be searched for, and not found in template context "
2013-11-03 05:02:56 +08:00
# Regression test for #10183
2014-01-14 23:43:27 +08:00
r = self . client . get ( ' /check_unicode/ ' )
2012-06-08 00:08:47 +08:00
self . assertNotContains ( r , ' はたけ ' )
2012-05-31 23:21:13 +08:00
self . assertNotContains ( r , b ' \xe3 \x81 \xaf \xe3 \x81 \x9f \xe3 \x81 \x91 ' . decode ( ' utf-8 ' ) )
2013-04-11 16:36:12 +08:00
def test_binary_contains ( self ) :
2014-01-14 23:43:27 +08:00
r = self . client . get ( ' /check_binary/ ' )
2013-04-13 02:00:49 +08:00
self . assertContains ( r , b ' % PDF-1.4 \r \n % \x93 \x8c \x8b \x9e ' )
2013-04-11 16:36:12 +08:00
with self . assertRaises ( AssertionError ) :
2013-04-13 02:00:49 +08:00
self . assertContains ( r , b ' % PDF-1.4 \r \n % \x93 \x8c \x8b \x9e ' , count = 2 )
def test_binary_not_contains ( self ) :
2014-01-14 23:43:27 +08:00
r = self . client . get ( ' /check_binary/ ' )
2013-04-13 02:00:49 +08:00
self . assertNotContains ( r , b ' % ODF-1.4 \r \n % \x93 \x8c \x8b \x9e ' )
with self . assertRaises ( AssertionError ) :
self . assertNotContains ( r , b ' % PDF-1.4 \r \n % \x93 \x8c \x8b \x9e ' )
2013-04-11 16:36:12 +08:00
2012-09-19 02:58:40 +08:00
def test_nontext_contains ( self ) :
2014-01-14 23:43:27 +08:00
r = self . client . get ( ' /no_template_view/ ' )
2012-09-19 02:58:40 +08:00
self . assertContains ( r , ugettext_lazy ( ' once ' ) )
def test_nontext_not_contains ( self ) :
2014-01-14 23:43:27 +08:00
r = self . client . get ( ' /no_template_view/ ' )
2012-09-19 02:58:40 +08:00
self . assertNotContains ( r , ugettext_lazy ( ' never ' ) )
2012-05-31 23:21:13 +08:00
def test_assert_contains_renders_template_response ( self ) :
""" Test that we can pass in an unrendered SimpleTemplateReponse
without throwing an error .
Refs #15826.
"""
response = SimpleTemplateResponse ( Template ( ' Hello ' ) , status = 200 )
self . assertContains ( response , ' Hello ' )
def test_assert_contains_using_non_template_response ( self ) :
""" Test that auto-rendering does not affect responses that aren ' t
instances ( or subclasses ) of SimpleTemplateResponse .
Refs #15826.
"""
response = HttpResponse ( ' Hello ' )
self . assertContains ( response , ' Hello ' )
def test_assert_not_contains_renders_template_response ( self ) :
""" Test that we can pass in an unrendered SimpleTemplateReponse
without throwing an error .
Refs #15826.
"""
response = SimpleTemplateResponse ( Template ( ' Hello ' ) , status = 200 )
self . assertNotContains ( response , ' Bye ' )
def test_assert_not_contains_using_non_template_response ( self ) :
""" Test that auto-rendering does not affect responses that aren ' t
instances ( or subclasses ) of SimpleTemplateResponse .
Refs #15826.
"""
response = HttpResponse ( ' Hello ' )
self . assertNotContains ( response , ' Bye ' )
2013-11-03 05:34:05 +08:00
2014-04-05 14:04:46 +08:00
@override_settings ( PASSWORD_HASHERS = ( ' django.contrib.auth.hashers.SHA1PasswordHasher ' , ) ,
ROOT_URLCONF = ' test_client_regress.urls ' , )
2012-05-31 23:21:13 +08:00
class AssertTemplateUsedTests ( TestCase ) :
fixtures = [ ' testdata.json ' ]
def test_no_context ( self ) :
" Template usage assertions work then templates aren ' t in use "
2014-01-14 23:43:27 +08:00
response = self . client . get ( ' /no_template_view/ ' )
2012-05-31 23:21:13 +08:00
# Check that the no template case doesn't mess with the template assertions
self . assertTemplateNotUsed ( response , ' GET Template ' )
try :
self . assertTemplateUsed ( response , ' GET Template ' )
except AssertionError as e :
self . assertIn ( " No templates used to render the response " , str ( e ) )
try :
self . assertTemplateUsed ( response , ' GET Template ' , msg_prefix = ' abc ' )
except AssertionError as e :
self . assertIn ( " abc: No templates used to render the response " , str ( e ) )
2014-04-15 03:13:49 +08:00
with self . assertRaises ( AssertionError ) as context :
self . assertTemplateUsed ( response , ' GET Template ' , count = 2 )
self . assertIn (
" No templates used to render the response " ,
str ( context . exception ) )
2012-05-31 23:21:13 +08:00
def test_single_context ( self ) :
" Template assertions work when there is a single context "
2014-01-14 23:43:27 +08:00
response = self . client . get ( ' /post_view/ ' , { } )
2012-05-31 23:21:13 +08:00
try :
self . assertTemplateNotUsed ( response , ' Empty GET Template ' )
except AssertionError as e :
self . assertIn ( " Template ' Empty GET Template ' was used unexpectedly in rendering the response " , str ( e ) )
try :
self . assertTemplateNotUsed ( response , ' Empty GET Template ' , msg_prefix = ' abc ' )
except AssertionError as e :
self . assertIn ( " abc: Template ' Empty GET Template ' was used unexpectedly in rendering the response " , str ( e ) )
try :
self . assertTemplateUsed ( response , ' Empty POST Template ' )
except AssertionError as e :
self . assertIn ( " Template ' Empty POST Template ' was not a template used to render the response. Actual template(s) used: Empty GET Template " , str ( e ) )
try :
self . assertTemplateUsed ( response , ' Empty POST Template ' , msg_prefix = ' abc ' )
except AssertionError as e :
self . assertIn ( " abc: Template ' Empty POST Template ' was not a template used to render the response. Actual template(s) used: Empty GET Template " , str ( e ) )
2014-04-15 03:13:49 +08:00
with self . assertRaises ( AssertionError ) as context :
self . assertTemplateUsed ( response , ' Empty GET Template ' , count = 2 )
self . assertIn (
" Template ' Empty GET Template ' was expected to be rendered 2 "
" time(s) but was actually rendered 1 time(s). " ,
str ( context . exception ) )
with self . assertRaises ( AssertionError ) as context :
self . assertTemplateUsed (
response , ' Empty GET Template ' , msg_prefix = ' abc ' , count = 2 )
self . assertIn (
" abc: Template ' Empty GET Template ' was expected to be rendered 2 "
" time(s) but was actually rendered 1 time(s). " ,
str ( context . exception ) )
2012-05-31 23:21:13 +08:00
def test_multiple_context ( self ) :
" Template assertions work when there are multiple contexts "
post_data = {
' text ' : ' Hello World ' ,
' email ' : ' foo@example.com ' ,
' value ' : 37 ,
' single ' : ' b ' ,
2013-10-27 03:15:03 +08:00
' multi ' : ( ' b ' , ' c ' , ' e ' )
2012-05-31 23:21:13 +08:00
}
2014-01-14 23:43:27 +08:00
response = self . client . post ( ' /form_view_with_template/ ' , post_data )
2012-05-31 23:21:13 +08:00
self . assertContains ( response , ' POST data OK ' )
try :
self . assertTemplateNotUsed ( response , " form_view.html " )
except AssertionError as e :
self . assertIn ( " Template ' form_view.html ' was used unexpectedly in rendering the response " , str ( e ) )
try :
self . assertTemplateNotUsed ( response , ' base.html ' )
except AssertionError as e :
self . assertIn ( " Template ' base.html ' was used unexpectedly in rendering the response " , str ( e ) )
try :
self . assertTemplateUsed ( response , " Valid POST Template " )
except AssertionError as e :
self . assertIn ( " Template ' Valid POST Template ' was not a template used to render the response. Actual template(s) used: form_view.html, base.html " , str ( e ) )
2014-04-15 03:13:49 +08:00
with self . assertRaises ( AssertionError ) as context :
self . assertTemplateUsed ( response , ' base.html ' , count = 2 )
self . assertIn (
" Template ' base.html ' was expected to be rendered 2 "
" time(s) but was actually rendered 1 time(s). " ,
str ( context . exception ) )
def test_template_rendered_multiple_times ( self ) :
""" Template assertions work when a template is rendered multiple times. """
response = self . client . get ( ' /render_template_multiple_times/ ' )
self . assertTemplateUsed ( response , ' base.html ' , count = 2 )
2013-11-03 05:34:05 +08:00
2014-04-05 14:04:46 +08:00
@override_settings ( ROOT_URLCONF = ' test_client_regress.urls ' )
2012-05-31 23:21:13 +08:00
class AssertRedirectsTests ( TestCase ) :
2014-01-14 23:43:27 +08:00
2012-05-31 23:21:13 +08:00
def test_redirect_page ( self ) :
" An assertion is raised if the original page couldn ' t be retrieved as expected "
# This page will redirect with code 301, not 302
2014-01-14 23:43:27 +08:00
response = self . client . get ( ' /permanent_redirect_view/ ' )
2012-05-31 23:21:13 +08:00
try :
2014-01-14 23:43:27 +08:00
self . assertRedirects ( response , ' /get_view/ ' )
2012-05-31 23:21:13 +08:00
except AssertionError as e :
self . assertIn ( " Response didn ' t redirect as expected: Response code was 301 (expected 302) " , str ( e ) )
try :
2014-01-14 23:43:27 +08:00
self . assertRedirects ( response , ' /get_view/ ' , msg_prefix = ' abc ' )
2012-05-31 23:21:13 +08:00
except AssertionError as e :
self . assertIn ( " abc: Response didn ' t redirect as expected: Response code was 301 (expected 302) " , str ( e ) )
def test_lost_query ( self ) :
" An assertion is raised if the redirect location doesn ' t preserve GET parameters "
2014-01-14 23:43:27 +08:00
response = self . client . get ( ' /redirect_view/ ' , { ' var ' : ' value ' } )
2012-05-31 23:21:13 +08:00
try :
2014-01-14 23:43:27 +08:00
self . assertRedirects ( response , ' /get_view/ ' )
2012-05-31 23:21:13 +08:00
except AssertionError as e :
2014-01-14 23:43:27 +08:00
self . assertIn ( " Response redirected to ' http://testserver/get_view/?var=value ' , expected ' http://testserver/get_view/ ' " , str ( e ) )
2012-05-31 23:21:13 +08:00
try :
2014-01-14 23:43:27 +08:00
self . assertRedirects ( response , ' /get_view/ ' , msg_prefix = ' abc ' )
2012-05-31 23:21:13 +08:00
except AssertionError as e :
2014-01-14 23:43:27 +08:00
self . assertIn ( " abc: Response redirected to ' http://testserver/get_view/?var=value ' , expected ' http://testserver/get_view/ ' " , str ( e ) )
2012-05-31 23:21:13 +08:00
def test_incorrect_target ( self ) :
" An assertion is raised if the response redirects to another target "
2014-01-14 23:43:27 +08:00
response = self . client . get ( ' /permanent_redirect_view/ ' )
2012-05-31 23:21:13 +08:00
try :
# Should redirect to get_view
2014-01-14 23:43:27 +08:00
self . assertRedirects ( response , ' /some_view/ ' )
2012-05-31 23:21:13 +08:00
except AssertionError as e :
self . assertIn ( " Response didn ' t redirect as expected: Response code was 301 (expected 302) " , str ( e ) )
def test_target_page ( self ) :
" An assertion is raised if the response redirect target cannot be retrieved as expected "
2014-01-14 23:43:27 +08:00
response = self . client . get ( ' /double_redirect_view/ ' )
2012-05-31 23:21:13 +08:00
try :
# The redirect target responds with a 301 code, not 200
2014-01-14 23:43:27 +08:00
self . assertRedirects ( response , ' http://testserver/permanent_redirect_view/ ' )
2012-05-31 23:21:13 +08:00
except AssertionError as e :
2014-01-14 23:43:27 +08:00
self . assertIn ( " Couldn ' t retrieve redirection page ' /permanent_redirect_view/ ' : response code was 301 (expected 200) " , str ( e ) )
2012-05-31 23:21:13 +08:00
try :
# The redirect target responds with a 301 code, not 200
2014-01-14 23:43:27 +08:00
self . assertRedirects ( response , ' http://testserver/permanent_redirect_view/ ' , msg_prefix = ' abc ' )
2012-05-31 23:21:13 +08:00
except AssertionError as e :
2014-01-14 23:43:27 +08:00
self . assertIn ( " abc: Couldn ' t retrieve redirection page ' /permanent_redirect_view/ ' : response code was 301 (expected 200) " , str ( e ) )
2012-05-31 23:21:13 +08:00
def test_redirect_chain ( self ) :
" You can follow a redirect chain of multiple redirects "
2014-01-14 23:43:27 +08:00
response = self . client . get ( ' /redirects/further/more/ ' , { } , follow = True )
self . assertRedirects ( response , ' /no_template_view/ ' ,
2012-05-31 23:21:13 +08:00
status_code = 301 , target_status_code = 200 )
self . assertEqual ( len ( response . redirect_chain ) , 1 )
2014-01-14 23:43:27 +08:00
self . assertEqual ( response . redirect_chain [ 0 ] , ( ' http://testserver/no_template_view/ ' , 301 ) )
2012-05-31 23:21:13 +08:00
def test_multiple_redirect_chain ( self ) :
" You can follow a redirect chain of multiple redirects "
2014-01-14 23:43:27 +08:00
response = self . client . get ( ' /redirects/ ' , { } , follow = True )
self . assertRedirects ( response , ' /no_template_view/ ' ,
2012-05-31 23:21:13 +08:00
status_code = 301 , target_status_code = 200 )
self . assertEqual ( len ( response . redirect_chain ) , 3 )
2014-01-14 23:43:27 +08:00
self . assertEqual ( response . redirect_chain [ 0 ] , ( ' http://testserver/redirects/further/ ' , 301 ) )
self . assertEqual ( response . redirect_chain [ 1 ] , ( ' http://testserver/redirects/further/more/ ' , 301 ) )
self . assertEqual ( response . redirect_chain [ 2 ] , ( ' http://testserver/no_template_view/ ' , 301 ) )
2012-05-31 23:21:13 +08:00
def test_redirect_chain_to_non_existent ( self ) :
" You can follow a chain to a non-existent view "
2014-01-14 23:43:27 +08:00
response = self . client . get ( ' /redirect_to_non_existent_view2/ ' , { } , follow = True )
self . assertRedirects ( response , ' /non_existent_view/ ' ,
2012-05-31 23:21:13 +08:00
status_code = 301 , target_status_code = 404 )
def test_redirect_chain_to_self ( self ) :
" Redirections to self are caught and escaped "
2014-10-17 21:46:42 +08:00
with self . assertRaises ( RedirectCycleError ) as context :
self . client . get ( ' /redirect_to_self/ ' , { } , follow = True )
response = context . exception . last_response
2012-05-31 23:21:13 +08:00
# The chain of redirects stops once the cycle is detected.
2014-01-14 23:43:27 +08:00
self . assertRedirects ( response , ' /redirect_to_self/ ' ,
2012-05-31 23:21:13 +08:00
status_code = 301 , target_status_code = 301 )
self . assertEqual ( len ( response . redirect_chain ) , 2 )
2014-10-17 21:46:42 +08:00
def test_redirect_to_self_with_changing_query ( self ) :
" Redirections don ' t loop forever even if query is changing "
with self . assertRaises ( RedirectCycleError ) :
self . client . get ( ' /redirect_to_self_with_changing_query_view/ ' , { ' counter ' : ' 0 ' } , follow = True )
2012-05-31 23:21:13 +08:00
def test_circular_redirect ( self ) :
" Circular redirect chains are caught and escaped "
2014-10-17 21:46:42 +08:00
with self . assertRaises ( RedirectCycleError ) as context :
self . client . get ( ' /circular_redirect_1/ ' , { } , follow = True )
response = context . exception . last_response
2012-05-31 23:21:13 +08:00
# The chain of redirects will get back to the starting point, but stop there.
2014-01-14 23:43:27 +08:00
self . assertRedirects ( response , ' /circular_redirect_2/ ' ,
2012-05-31 23:21:13 +08:00
status_code = 301 , target_status_code = 301 )
self . assertEqual ( len ( response . redirect_chain ) , 4 )
def test_redirect_chain_post ( self ) :
" A redirect chain will be followed from an initial POST post "
2014-01-14 23:43:27 +08:00
response = self . client . post ( ' /redirects/ ' ,
2012-05-31 23:21:13 +08:00
{ ' nothing ' : ' to_send ' } , follow = True )
self . assertRedirects ( response ,
2014-01-14 23:43:27 +08:00
' /no_template_view/ ' , 301 , 200 )
2012-05-31 23:21:13 +08:00
self . assertEqual ( len ( response . redirect_chain ) , 3 )
def test_redirect_chain_head ( self ) :
" A redirect chain will be followed from an initial HEAD request "
2014-01-14 23:43:27 +08:00
response = self . client . head ( ' /redirects/ ' ,
2012-05-31 23:21:13 +08:00
{ ' nothing ' : ' to_send ' } , follow = True )
self . assertRedirects ( response ,
2014-01-14 23:43:27 +08:00
' /no_template_view/ ' , 301 , 200 )
2012-05-31 23:21:13 +08:00
self . assertEqual ( len ( response . redirect_chain ) , 3 )
def test_redirect_chain_options ( self ) :
" A redirect chain will be followed from an initial OPTIONS request "
2014-01-14 23:43:27 +08:00
response = self . client . options ( ' /redirects/ ' ,
2012-05-31 23:21:13 +08:00
follow = True )
self . assertRedirects ( response ,
2014-01-14 23:43:27 +08:00
' /no_template_view/ ' , 301 , 200 )
2012-05-31 23:21:13 +08:00
self . assertEqual ( len ( response . redirect_chain ) , 3 )
def test_redirect_chain_put ( self ) :
" A redirect chain will be followed from an initial PUT request "
2014-01-14 23:43:27 +08:00
response = self . client . put ( ' /redirects/ ' ,
2012-05-31 23:21:13 +08:00
follow = True )
self . assertRedirects ( response ,
2014-01-14 23:43:27 +08:00
' /no_template_view/ ' , 301 , 200 )
2012-05-31 23:21:13 +08:00
self . assertEqual ( len ( response . redirect_chain ) , 3 )
def test_redirect_chain_delete ( self ) :
" A redirect chain will be followed from an initial DELETE request "
2014-01-14 23:43:27 +08:00
response = self . client . delete ( ' /redirects/ ' ,
2012-05-31 23:21:13 +08:00
follow = True )
self . assertRedirects ( response ,
2014-01-14 23:43:27 +08:00
' /no_template_view/ ' , 301 , 200 )
2012-05-31 23:21:13 +08:00
self . assertEqual ( len ( response . redirect_chain ) , 3 )
def test_redirect_to_different_host ( self ) :
" The test client will preserve scheme, host and port changes "
2014-01-14 23:43:27 +08:00
response = self . client . get ( ' /redirect_other_host/ ' , follow = True )
2012-05-31 23:21:13 +08:00
self . assertRedirects ( response ,
2014-01-14 23:43:27 +08:00
' https://otherserver:8443/no_template_view/ ' ,
2012-05-31 23:21:13 +08:00
status_code = 301 , target_status_code = 200 )
# We can't use is_secure() or get_host()
# because response.request is a dictionary, not an HttpRequest
self . assertEqual ( response . request . get ( ' wsgi.url_scheme ' ) , ' https ' )
self . assertEqual ( response . request . get ( ' SERVER_NAME ' ) , ' otherserver ' )
self . assertEqual ( response . request . get ( ' SERVER_PORT ' ) , ' 8443 ' )
def test_redirect_chain_on_non_redirect_page ( self ) :
" An assertion is raised if the original page couldn ' t be retrieved as expected "
# This page will redirect with code 301, not 302
2014-01-14 23:43:27 +08:00
response = self . client . get ( ' /get_view/ ' , follow = True )
2012-05-31 23:21:13 +08:00
try :
2014-01-14 23:43:27 +08:00
self . assertRedirects ( response , ' /get_view/ ' )
2012-05-31 23:21:13 +08:00
except AssertionError as e :
self . assertIn ( " Response didn ' t redirect as expected: Response code was 200 (expected 302) " , str ( e ) )
try :
2014-01-14 23:43:27 +08:00
self . assertRedirects ( response , ' /get_view/ ' , msg_prefix = ' abc ' )
2012-05-31 23:21:13 +08:00
except AssertionError as e :
self . assertIn ( " abc: Response didn ' t redirect as expected: Response code was 200 (expected 302) " , str ( e ) )
def test_redirect_on_non_redirect_page ( self ) :
" An assertion is raised if the original page couldn ' t be retrieved as expected "
# This page will redirect with code 301, not 302
2014-01-14 23:43:27 +08:00
response = self . client . get ( ' /get_view/ ' )
2012-05-31 23:21:13 +08:00
try :
2014-01-14 23:43:27 +08:00
self . assertRedirects ( response , ' /get_view/ ' )
2012-05-31 23:21:13 +08:00
except AssertionError as e :
self . assertIn ( " Response didn ' t redirect as expected: Response code was 200 (expected 302) " , str ( e ) )
try :
2014-01-14 23:43:27 +08:00
self . assertRedirects ( response , ' /get_view/ ' , msg_prefix = ' abc ' )
2012-05-31 23:21:13 +08:00
except AssertionError as e :
self . assertIn ( " abc: Response didn ' t redirect as expected: Response code was 200 (expected 302) " , str ( e ) )
2013-10-29 03:34:09 +08:00
def test_redirect_scheme ( self ) :
" An assertion is raised if the response doesn ' t have the scheme specified in expected_url "
# Assure that original request scheme is preserved if no scheme specified in the redirect location
2014-01-14 23:43:27 +08:00
response = self . client . get ( ' /redirect_view/ ' , secure = True )
self . assertRedirects ( response , ' https://testserver/get_view/ ' )
2013-10-29 03:34:09 +08:00
# For all possible True/False combinations of follow and secure
for follow , secure in itertools . product ( [ True , False ] , repeat = 2 ) :
# always redirects to https
2014-01-14 23:43:27 +08:00
response = self . client . get ( ' /https_redirect_view/ ' , follow = follow , secure = secure )
2013-10-29 03:34:09 +08:00
# no scheme to compare too, always succeeds
2014-01-14 23:43:27 +08:00
self . assertRedirects ( response , ' /secure_view/ ' , status_code = 301 )
2013-10-29 03:34:09 +08:00
# the goal scheme is https
2014-01-14 23:43:27 +08:00
self . assertRedirects ( response , ' https://testserver/secure_view/ ' , status_code = 301 )
2013-10-29 03:34:09 +08:00
with self . assertRaises ( AssertionError ) :
2014-01-14 23:43:27 +08:00
self . assertRedirects ( response , ' http://testserver/secure_view/ ' , status_code = 301 )
2013-10-29 03:34:09 +08:00
2012-05-31 23:21:13 +08:00
2014-04-05 14:04:46 +08:00
@override_settings ( ROOT_URLCONF = ' test_client_regress.urls ' )
2012-05-31 23:21:13 +08:00
class AssertFormErrorTests ( TestCase ) :
2014-01-14 23:43:27 +08:00
2012-05-31 23:21:13 +08:00
def test_unknown_form ( self ) :
" An assertion is raised if the form name is unknown "
post_data = {
' text ' : ' Hello World ' ,
' email ' : ' not an email address ' ,
' value ' : 37 ,
' single ' : ' b ' ,
2013-10-27 03:15:03 +08:00
' multi ' : ( ' b ' , ' c ' , ' e ' )
2012-05-31 23:21:13 +08:00
}
2014-01-14 23:43:27 +08:00
response = self . client . post ( ' /form_view/ ' , post_data )
2012-05-31 23:21:13 +08:00
self . assertEqual ( response . status_code , 200 )
self . assertTemplateUsed ( response , " Invalid POST Template " )
try :
self . assertFormError ( response , ' wrong_form ' , ' some_field ' , ' Some error. ' )
except AssertionError as e :
self . assertIn ( " The form ' wrong_form ' was not used to render the response " , str ( e ) )
try :
self . assertFormError ( response , ' wrong_form ' , ' some_field ' , ' Some error. ' , msg_prefix = ' abc ' )
except AssertionError as e :
self . assertIn ( " abc: The form ' wrong_form ' was not used to render the response " , str ( e ) )
def test_unknown_field ( self ) :
" An assertion is raised if the field name is unknown "
post_data = {
' text ' : ' Hello World ' ,
' email ' : ' not an email address ' ,
' value ' : 37 ,
' single ' : ' b ' ,
2013-10-27 03:15:03 +08:00
' multi ' : ( ' b ' , ' c ' , ' e ' )
2012-05-31 23:21:13 +08:00
}
2014-01-14 23:43:27 +08:00
response = self . client . post ( ' /form_view/ ' , post_data )
2012-05-31 23:21:13 +08:00
self . assertEqual ( response . status_code , 200 )
self . assertTemplateUsed ( response , " Invalid POST Template " )
try :
self . assertFormError ( response , ' form ' , ' some_field ' , ' Some error. ' )
except AssertionError as e :
self . assertIn ( " The form ' form ' in context 0 does not contain the field ' some_field ' " , str ( e ) )
try :
self . assertFormError ( response , ' form ' , ' some_field ' , ' Some error. ' , msg_prefix = ' abc ' )
except AssertionError as e :
self . assertIn ( " abc: The form ' form ' in context 0 does not contain the field ' some_field ' " , str ( e ) )
def test_noerror_field ( self ) :
" An assertion is raised if the field doesn ' t have any errors "
post_data = {
' text ' : ' Hello World ' ,
' email ' : ' not an email address ' ,
' value ' : 37 ,
' single ' : ' b ' ,
2013-10-27 03:15:03 +08:00
' multi ' : ( ' b ' , ' c ' , ' e ' )
2012-05-31 23:21:13 +08:00
}
2014-01-14 23:43:27 +08:00
response = self . client . post ( ' /form_view/ ' , post_data )
2012-05-31 23:21:13 +08:00
self . assertEqual ( response . status_code , 200 )
self . assertTemplateUsed ( response , " Invalid POST Template " )
try :
self . assertFormError ( response , ' form ' , ' value ' , ' Some error. ' )
except AssertionError as e :
self . assertIn ( " The field ' value ' on form ' form ' in context 0 contains no errors " , str ( e ) )
try :
self . assertFormError ( response , ' form ' , ' value ' , ' Some error. ' , msg_prefix = ' abc ' )
except AssertionError as e :
self . assertIn ( " abc: The field ' value ' on form ' form ' in context 0 contains no errors " , str ( e ) )
def test_unknown_error ( self ) :
" An assertion is raised if the field doesn ' t contain the provided error "
post_data = {
' text ' : ' Hello World ' ,
' email ' : ' not an email address ' ,
' value ' : 37 ,
' single ' : ' b ' ,
2013-10-27 03:15:03 +08:00
' multi ' : ( ' b ' , ' c ' , ' e ' )
2012-05-31 23:21:13 +08:00
}
2014-01-14 23:43:27 +08:00
response = self . client . post ( ' /form_view/ ' , post_data )
2012-05-31 23:21:13 +08:00
self . assertEqual ( response . status_code , 200 )
self . assertTemplateUsed ( response , " Invalid POST Template " )
try :
self . assertFormError ( response , ' form ' , ' email ' , ' Some error. ' )
except AssertionError as e :
2012-09-26 20:14:51 +08:00
self . assertIn ( str_prefix ( " The field ' email ' on form ' form ' in context 0 does not contain the error ' Some error. ' (actual errors: [ %(_)s ' Enter a valid email address. ' ]) " ) , str ( e ) )
2012-05-31 23:21:13 +08:00
try :
self . assertFormError ( response , ' form ' , ' email ' , ' Some error. ' , msg_prefix = ' abc ' )
except AssertionError as e :
2012-09-26 20:14:51 +08:00
self . assertIn ( str_prefix ( " abc: The field ' email ' on form ' form ' in context 0 does not contain the error ' Some error. ' (actual errors: [ %(_)s ' Enter a valid email address. ' ]) " ) , str ( e ) )
2012-05-31 23:21:13 +08:00
def test_unknown_nonfield_error ( self ) :
"""
Checks that an assertion is raised if the form ' s non field errors
doesn ' t contain the provided error.
"""
post_data = {
' text ' : ' Hello World ' ,
' email ' : ' not an email address ' ,
' value ' : 37 ,
' single ' : ' b ' ,
2013-10-27 03:15:03 +08:00
' multi ' : ( ' b ' , ' c ' , ' e ' )
2012-05-31 23:21:13 +08:00
}
2014-01-14 23:43:27 +08:00
response = self . client . post ( ' /form_view/ ' , post_data )
2012-05-31 23:21:13 +08:00
self . assertEqual ( response . status_code , 200 )
self . assertTemplateUsed ( response , " Invalid POST Template " )
try :
self . assertFormError ( response , ' form ' , None , ' Some error. ' )
except AssertionError as e :
self . assertIn ( " The form ' form ' in context 0 does not contain the non-field error ' Some error. ' (actual errors: ) " , str ( e ) )
try :
self . assertFormError ( response , ' form ' , None , ' Some error. ' , msg_prefix = ' abc ' )
except AssertionError as e :
self . assertIn ( " abc: The form ' form ' in context 0 does not contain the non-field error ' Some error. ' (actual errors: ) " , str ( e ) )
2013-11-03 05:34:05 +08:00
2014-04-05 14:04:46 +08:00
@override_settings ( ROOT_URLCONF = ' test_client_regress.urls ' )
2013-02-09 07:45:26 +08:00
class AssertFormsetErrorTests ( TestCase ) :
msg_prefixes = [ ( " " , { } ) , ( " abc: " , { " msg_prefix " : " abc " } ) ]
2013-10-22 18:21:07 +08:00
2013-02-09 07:45:26 +08:00
def setUp ( self ) :
""" Makes response object for testing field and non-field errors """
# For testing field and non-field errors
self . response_form_errors = self . getResponse ( {
' form-TOTAL_FORMS ' : ' 2 ' ,
' form-INITIAL_FORMS ' : ' 2 ' ,
' form-0-text ' : ' Raise non-field error ' ,
' form-0-email ' : ' not an email address ' ,
' form-0-value ' : 37 ,
' form-0-single ' : ' b ' ,
2013-10-27 03:15:03 +08:00
' form-0-multi ' : ( ' b ' , ' c ' , ' e ' ) ,
2013-02-09 07:45:26 +08:00
' form-1-text ' : ' Hello World ' ,
' form-1-email ' : ' email@domain.com ' ,
' form-1-value ' : 37 ,
' form-1-single ' : ' b ' ,
2013-10-27 03:15:03 +08:00
' form-1-multi ' : ( ' b ' , ' c ' , ' e ' ) ,
2013-02-09 07:45:26 +08:00
} )
# For testing non-form errors
self . response_nonform_errors = self . getResponse ( {
' form-TOTAL_FORMS ' : ' 2 ' ,
' form-INITIAL_FORMS ' : ' 2 ' ,
' form-0-text ' : ' Hello World ' ,
' form-0-email ' : ' email@domain.com ' ,
' form-0-value ' : 37 ,
' form-0-single ' : ' b ' ,
2013-10-27 03:15:03 +08:00
' form-0-multi ' : ( ' b ' , ' c ' , ' e ' ) ,
2013-02-09 07:45:26 +08:00
' form-1-text ' : ' Hello World ' ,
' form-1-email ' : ' email@domain.com ' ,
' form-1-value ' : 37 ,
' form-1-single ' : ' b ' ,
2013-10-27 03:15:03 +08:00
' form-1-multi ' : ( ' b ' , ' c ' , ' e ' ) ,
2013-02-09 07:45:26 +08:00
} )
def getResponse ( self , post_data ) :
2014-01-14 23:43:27 +08:00
response = self . client . post ( ' /formset_view/ ' , post_data )
2013-02-09 07:45:26 +08:00
self . assertEqual ( response . status_code , 200 )
self . assertTemplateUsed ( response , " Invalid POST Template " )
return response
def test_unknown_formset ( self ) :
" An assertion is raised if the formset name is unknown "
for prefix , kwargs in self . msg_prefixes :
with self . assertRaises ( AssertionError ) as cm :
self . assertFormsetError ( self . response_form_errors ,
' wrong_formset ' ,
0 ,
' Some_field ' ,
' Some error. ' ,
* * kwargs )
self . assertIn ( prefix + " The formset ' wrong_formset ' was not "
" used to render the response " ,
str ( cm . exception ) )
def test_unknown_field ( self ) :
" An assertion is raised if the field name is unknown "
for prefix , kwargs in self . msg_prefixes :
with self . assertRaises ( AssertionError ) as cm :
self . assertFormsetError ( self . response_form_errors ,
' my_formset ' ,
0 ,
' Some_field ' ,
' Some error. ' ,
* * kwargs )
self . assertIn ( prefix + " The formset ' my_formset ' , "
" form 0 in context 0 "
" does not contain the field ' Some_field ' " ,
str ( cm . exception ) )
def test_no_error_field ( self ) :
" An assertion is raised if the field doesn ' t have any errors "
for prefix , kwargs in self . msg_prefixes :
with self . assertRaises ( AssertionError ) as cm :
self . assertFormsetError ( self . response_form_errors ,
' my_formset ' ,
1 ,
' value ' ,
' Some error. ' ,
* * kwargs )
self . assertIn ( prefix + " The field ' value ' "
" on formset ' my_formset ' , form 1 "
" in context 0 contains no errors " ,
str ( cm . exception ) )
def test_unknown_error ( self ) :
" An assertion is raised if the field doesn ' t contain the specified error "
for prefix , kwargs in self . msg_prefixes :
with self . assertRaises ( AssertionError ) as cm :
self . assertFormsetError ( self . response_form_errors ,
' my_formset ' ,
0 ,
' email ' ,
' Some error. ' ,
* * kwargs )
self . assertIn ( str_prefix ( prefix + " The field ' email ' "
" on formset ' my_formset ' , form 0 in context 0 does not "
" contain the error ' Some error. ' (actual errors: "
" [ %(_)s ' Enter a valid email address. ' ]) " ) ,
str ( cm . exception ) )
def test_field_error ( self ) :
" No assertion is raised if the field contains the provided error "
for prefix , kwargs in self . msg_prefixes :
self . assertFormsetError ( self . response_form_errors ,
' my_formset ' ,
0 ,
' email ' ,
[ ' Enter a valid email address. ' ] ,
* * kwargs )
def test_no_nonfield_error ( self ) :
" An assertion is raised if the formsets non-field errors doesn ' t contain any errors. "
for prefix , kwargs in self . msg_prefixes :
with self . assertRaises ( AssertionError ) as cm :
self . assertFormsetError ( self . response_form_errors ,
' my_formset ' ,
1 ,
None ,
' Some error. ' ,
* * kwargs )
self . assertIn ( prefix + " The formset ' my_formset ' , form 1 in "
" context 0 does not contain any "
" non-field errors. " ,
str ( cm . exception ) )
def test_unknown_nonfield_error ( self ) :
" An assertion is raised if the formsets non-field errors doesn ' t contain the provided error. "
for prefix , kwargs in self . msg_prefixes :
with self . assertRaises ( AssertionError ) as cm :
self . assertFormsetError ( self . response_form_errors ,
' my_formset ' ,
0 ,
None ,
' Some error. ' ,
* * kwargs )
self . assertIn ( str_prefix ( prefix +
" The formset ' my_formset ' , form 0 in context 0 does not "
" contain the non-field error ' Some error. ' (actual errors: "
" [ %(_)s ' Non-field error. ' ]) " ) , str ( cm . exception ) )
def test_nonfield_error ( self ) :
" No assertion is raised if the formsets non-field errors contains the provided error. "
for prefix , kwargs in self . msg_prefixes :
self . assertFormsetError ( self . response_form_errors ,
' my_formset ' ,
0 ,
None ,
' Non-field error. ' ,
* * kwargs )
def test_no_nonform_error ( self ) :
" An assertion is raised if the formsets non-form errors doesn ' t contain any errors. "
for prefix , kwargs in self . msg_prefixes :
with self . assertRaises ( AssertionError ) as cm :
self . assertFormsetError ( self . response_form_errors ,
' my_formset ' ,
None ,
None ,
' Some error. ' ,
* * kwargs )
self . assertIn ( prefix + " The formset ' my_formset ' in context 0 "
" does not contain any non-form errors. " ,
str ( cm . exception ) )
def test_unknown_nonform_error ( self ) :
" An assertion is raised if the formsets non-form errors doesn ' t contain the provided error. "
for prefix , kwargs in self . msg_prefixes :
with self . assertRaises ( AssertionError ) as cm :
self . assertFormsetError ( self . response_nonform_errors ,
' my_formset ' ,
None ,
None ,
' Some error. ' ,
* * kwargs )
self . assertIn ( str_prefix ( prefix +
" The formset ' my_formset ' in context 0 does not contain the "
" non-form error ' Some error. ' (actual errors: [ %(_)s ' Forms "
" in a set must have distinct email addresses. ' ]) " ) , str ( cm . exception ) )
def test_nonform_error ( self ) :
" No assertion is raised if the formsets non-form errors contains the provided error. "
for prefix , kwargs in self . msg_prefixes :
self . assertFormsetError ( self . response_nonform_errors ,
' my_formset ' ,
None ,
None ,
' Forms in a set must have distinct email '
' addresses. ' ,
* * kwargs )
2013-11-03 05:34:05 +08:00
2014-04-05 14:04:46 +08:00
@override_settings ( PASSWORD_HASHERS = ( ' django.contrib.auth.hashers.SHA1PasswordHasher ' , ) ,
ROOT_URLCONF = ' test_client_regress.urls ' , )
2012-05-31 23:21:13 +08:00
class LoginTests ( TestCase ) :
fixtures = [ ' testdata ' ]
def test_login_different_client ( self ) :
" Check that using a different test client doesn ' t violate authentication "
# Create a second client, and log in.
c = Client ( )
login = c . login ( username = ' testclient ' , password = ' password ' )
self . assertTrue ( login , ' Could not log in ' )
# Get a redirection page with the second client.
2014-01-14 23:43:27 +08:00
response = c . get ( " /login_protected_redirect_view/ " )
2012-05-31 23:21:13 +08:00
# At this points, the self.client isn't logged in.
# Check that assertRedirects uses the original client, not the
# default client.
2014-01-14 23:43:27 +08:00
self . assertRedirects ( response , " http://testserver/get_view/ " )
2012-05-31 23:21:13 +08:00
2012-05-31 23:41:31 +08:00
@override_settings (
PASSWORD_HASHERS = ( ' django.contrib.auth.hashers.SHA1PasswordHasher ' , ) ,
2014-04-05 14:04:46 +08:00
SESSION_ENGINE = ' test_client_regress.session ' ,
ROOT_URLCONF = ' test_client_regress.urls ' ,
2012-05-31 23:41:31 +08:00
)
2012-05-31 23:21:13 +08:00
class SessionEngineTests ( TestCase ) :
fixtures = [ ' testdata ' ]
def test_login ( self ) :
" A session engine that modifies the session key can be used to log in "
login = self . client . login ( username = ' testclient ' , password = ' password ' )
self . assertTrue ( login , ' Could not log in ' )
# Try to access a login protected page.
2014-01-14 23:43:27 +08:00
response = self . client . get ( " /login_protected_view/ " )
2012-05-31 23:21:13 +08:00
self . assertEqual ( response . status_code , 200 )
self . assertEqual ( response . context [ ' user ' ] . username , ' testclient ' )
2014-04-05 14:04:46 +08:00
@override_settings ( ROOT_URLCONF = ' test_client_regress.urls ' , )
2012-05-31 23:21:13 +08:00
class URLEscapingTests ( TestCase ) :
2014-01-14 23:43:27 +08:00
2012-05-31 23:21:13 +08:00
def test_simple_argument_get ( self ) :
" Get a view that has a simple string argument "
response = self . client . get ( reverse ( ' arg_view ' , args = [ ' Slartibartfast ' ] ) )
self . assertEqual ( response . status_code , 200 )
self . assertEqual ( response . content , b ' Howdy, Slartibartfast ' )
def test_argument_with_space_get ( self ) :
" Get a view that has a string argument that requires escaping "
response = self . client . get ( reverse ( ' arg_view ' , args = [ ' Arthur Dent ' ] ) )
self . assertEqual ( response . status_code , 200 )
self . assertEqual ( response . content , b ' Hi, Arthur ' )
def test_simple_argument_post ( self ) :
" Post for a view that has a simple string argument "
response = self . client . post ( reverse ( ' arg_view ' , args = [ ' Slartibartfast ' ] ) )
self . assertEqual ( response . status_code , 200 )
self . assertEqual ( response . content , b ' Howdy, Slartibartfast ' )
def test_argument_with_space_post ( self ) :
" Post for a view that has a string argument that requires escaping "
response = self . client . post ( reverse ( ' arg_view ' , args = [ ' Arthur Dent ' ] ) )
self . assertEqual ( response . status_code , 200 )
self . assertEqual ( response . content , b ' Hi, Arthur ' )
2013-11-03 05:34:05 +08:00
2014-04-05 14:04:46 +08:00
@override_settings ( PASSWORD_HASHERS = ( ' django.contrib.auth.hashers.SHA1PasswordHasher ' , ) ,
ROOT_URLCONF = ' test_client_regress.urls ' , )
2012-05-31 23:21:13 +08:00
class ExceptionTests ( TestCase ) :
fixtures = [ ' testdata.json ' ]
def test_exception_cleared ( self ) :
" #5836 - A stale user exception isn ' t re-raised by the test client. "
2013-10-27 03:15:03 +08:00
login = self . client . login ( username = ' testclient ' , password = ' password ' )
2012-05-31 23:21:13 +08:00
self . assertTrue ( login , ' Could not log in ' )
try :
2014-01-14 23:43:27 +08:00
self . client . get ( " /staff_only/ " )
2012-05-31 23:21:13 +08:00
self . fail ( " General users should not be able to visit this page " )
2013-05-16 07:14:28 +08:00
except CustomTestException :
2012-05-31 23:21:13 +08:00
pass
# At this point, an exception has been raised, and should be cleared.
# This next operation should be successful; if it isn't we have a problem.
login = self . client . login ( username = ' staff ' , password = ' password ' )
self . assertTrue ( login , ' Could not log in ' )
try :
2014-01-14 23:43:27 +08:00
self . client . get ( " /staff_only/ " )
2013-05-16 07:14:28 +08:00
except CustomTestException :
2012-05-31 23:21:13 +08:00
self . fail ( " Staff should be able to visit this page " )
2012-05-31 23:41:31 +08:00
2014-04-05 14:04:46 +08:00
@override_settings ( ROOT_URLCONF = ' test_client_regress.urls ' )
2012-05-31 23:21:13 +08:00
class TemplateExceptionTests ( TestCase ) :
2014-01-14 23:43:27 +08:00
2014-12-18 05:51:42 +08:00
@override_settings ( TEMPLATES = [ {
' BACKEND ' : ' django.template.backends.django.DjangoTemplates ' ,
' DIRS ' : [ os . path . join ( os . path . dirname ( upath ( __file__ ) ) , ' bad_templates ' ) ] ,
} ] )
2012-05-31 23:21:13 +08:00
def test_bad_404_template ( self ) :
" Errors found when rendering 404 error templates are re-raised "
try :
2013-10-19 20:31:38 +08:00
self . client . get ( " /no_such_view/ " )
2012-05-31 23:21:13 +08:00
except TemplateSyntaxError :
pass
2014-11-16 03:58:26 +08:00
else :
self . fail ( " Should get error about syntax error in template " )
2012-05-31 23:21:13 +08:00
2013-11-03 05:34:05 +08:00
2012-05-31 23:21:13 +08:00
# We need two different tests to check URLconf substitution - one to check
# it was changed, and another one (without self.urls) to check it was reverted on
# teardown. This pair of tests relies upon the alphabetical ordering of test execution.
2014-04-05 14:04:46 +08:00
@override_settings ( ROOT_URLCONF = ' test_client_regress.urls ' )
2012-05-31 23:21:13 +08:00
class UrlconfSubstitutionTests ( TestCase ) :
def test_urlconf_was_changed ( self ) :
" TestCase can enforce a custom URLconf on a per-test basis "
url = reverse ( ' arg_view ' , args = [ ' somename ' ] )
self . assertEqual ( url , ' /arg_view/somename/ ' )
2013-11-03 05:34:05 +08:00
2012-05-31 23:21:13 +08:00
# This test needs to run *after* UrlconfSubstitutionTests; the zz prefix in the
# name is to ensure alphabetical ordering.
class zzUrlconfSubstitutionTests ( TestCase ) :
2014-01-14 23:43:27 +08:00
2012-05-31 23:21:13 +08:00
def test_urlconf_was_reverted ( self ) :
2014-01-14 23:43:27 +08:00
""" URLconf is reverted to original value after modification in a TestCase
This will not find a match as the default ROOT_URLCONF is empty .
"""
with self . assertRaises ( NoReverseMatch ) :
reverse ( ' arg_view ' , args = [ ' somename ' ] )
2012-05-31 23:21:13 +08:00
2013-11-03 05:34:05 +08:00
2014-04-05 14:04:46 +08:00
@override_settings ( PASSWORD_HASHERS = ( ' django.contrib.auth.hashers.SHA1PasswordHasher ' , ) ,
ROOT_URLCONF = ' test_client_regress.urls ' , )
2012-05-31 23:21:13 +08:00
class ContextTests ( TestCase ) :
fixtures = [ ' testdata ' ]
def test_single_context ( self ) :
" Context variables can be retrieved from a single context "
2014-01-14 23:43:27 +08:00
response = self . client . get ( " /request_data/ " , data = { ' foo ' : ' whiz ' } )
2012-05-31 23:21:13 +08:00
self . assertEqual ( response . context . __class__ , Context )
2014-10-28 18:02:56 +08:00
self . assertIn ( ' get-foo ' , response . context )
2012-05-31 23:21:13 +08:00
self . assertEqual ( response . context [ ' get-foo ' ] , ' whiz ' )
self . assertEqual ( response . context [ ' request-foo ' ] , ' whiz ' )
self . assertEqual ( response . context [ ' data ' ] , ' sausage ' )
try :
response . context [ ' does-not-exist ' ]
self . fail ( ' Should not be able to retrieve non-existent key ' )
except KeyError as e :
self . assertEqual ( e . args [ 0 ] , ' does-not-exist ' )
def test_inherited_context ( self ) :
" Context variables can be retrieved from a list of contexts "
2014-01-14 23:43:27 +08:00
response = self . client . get ( " /request_data_extended/ " , data = { ' foo ' : ' whiz ' } )
2012-05-31 23:21:13 +08:00
self . assertEqual ( response . context . __class__ , ContextList )
self . assertEqual ( len ( response . context ) , 2 )
2014-10-28 18:02:56 +08:00
self . assertIn ( ' get-foo ' , response . context )
2012-05-31 23:21:13 +08:00
self . assertEqual ( response . context [ ' get-foo ' ] , ' whiz ' )
self . assertEqual ( response . context [ ' request-foo ' ] , ' whiz ' )
self . assertEqual ( response . context [ ' data ' ] , ' bacon ' )
try :
response . context [ ' does-not-exist ' ]
self . fail ( ' Should not be able to retrieve non-existent key ' )
except KeyError as e :
self . assertEqual ( e . args [ 0 ] , ' does-not-exist ' )
2013-05-14 00:38:29 +08:00
def test_contextlist_keys ( self ) :
c1 = Context ( )
c1 . update ( { ' hello ' : ' world ' , ' goodbye ' : ' john ' } )
c1 . update ( { ' hello ' : ' dolly ' , ' dolly ' : ' parton ' } )
c2 = Context ( )
c2 . update ( { ' goodbye ' : ' world ' , ' python ' : ' rocks ' } )
c2 . update ( { ' goodbye ' : ' dolly ' } )
l = ContextList ( [ c1 , c2 ] )
# None, True and False are builtins of BaseContext, and present
# in every Context without needing to be added.
2014-09-26 20:31:50 +08:00
self . assertEqual ( { ' None ' , ' True ' , ' False ' , ' hello ' , ' goodbye ' ,
2014-09-29 20:13:40 +08:00
' python ' , ' dolly ' } ,
2013-05-14 00:38:29 +08:00
l . keys ( ) )
2012-05-31 23:21:13 +08:00
def test_15368 ( self ) :
2014-12-14 17:17:18 +08:00
with warnings . catch_warnings ( ) :
warnings . filterwarnings ( " ignore " , category = RemovedInDjango20Warning )
# Need to insert a context processor that assumes certain things about
# the request instance. This triggers a bug caused by some ways of
# copying RequestContext.
2014-12-18 06:36:32 +08:00
with self . settings ( TEMPLATES = [ {
' BACKEND ' : ' django.template.backends.django.DjangoTemplates ' ,
' APP_DIRS ' : True ,
' OPTIONS ' : {
' context_processors ' : [
' test_client_regress.context_processors.special ' ,
] ,
} ,
} ] ) :
2014-12-14 17:17:18 +08:00
response = self . client . get ( " /request_context_view/ " )
self . assertContains ( response , ' Path: /request_context_view/ ' )
2012-05-31 23:21:13 +08:00
2013-02-04 23:50:15 +08:00
def test_nested_requests ( self ) :
"""
response . context is not lost when view call another view .
"""
2014-01-14 23:43:27 +08:00
response = self . client . get ( " /nested_view/ " )
2013-02-04 23:50:15 +08:00
self . assertEqual ( response . context . __class__ , Context )
self . assertEqual ( response . context [ ' nested ' ] , ' yes ' )
2012-05-31 23:21:13 +08:00
2014-04-05 14:04:46 +08:00
@override_settings ( PASSWORD_HASHERS = ( ' django.contrib.auth.hashers.SHA1PasswordHasher ' , ) ,
ROOT_URLCONF = ' test_client_regress.urls ' , )
2012-05-31 23:21:13 +08:00
class SessionTests ( TestCase ) :
fixtures = [ ' testdata.json ' ]
def test_session ( self ) :
" The session isn ' t lost if a user logs in "
# The session doesn't exist to start.
2014-01-14 23:43:27 +08:00
response = self . client . get ( ' /check_session/ ' )
2012-05-31 23:21:13 +08:00
self . assertEqual ( response . status_code , 200 )
self . assertEqual ( response . content , b ' NO ' )
# This request sets a session variable.
2014-01-14 23:43:27 +08:00
response = self . client . get ( ' /set_session/ ' )
2012-05-31 23:21:13 +08:00
self . assertEqual ( response . status_code , 200 )
self . assertEqual ( response . content , b ' set_session ' )
# Check that the session has been modified
2014-01-14 23:43:27 +08:00
response = self . client . get ( ' /check_session/ ' )
2012-05-31 23:21:13 +08:00
self . assertEqual ( response . status_code , 200 )
self . assertEqual ( response . content , b ' YES ' )
# Log in
2013-10-27 03:15:03 +08:00
login = self . client . login ( username = ' testclient ' , password = ' password ' )
2012-05-31 23:21:13 +08:00
self . assertTrue ( login , ' Could not log in ' )
# Session should still contain the modified value
2014-01-14 23:43:27 +08:00
response = self . client . get ( ' /check_session/ ' )
2012-05-31 23:21:13 +08:00
self . assertEqual ( response . status_code , 200 )
self . assertEqual ( response . content , b ' YES ' )
2014-05-23 06:45:02 +08:00
def test_session_initiated ( self ) :
session = self . client . session
session [ ' session_var ' ] = ' foo '
session . save ( )
response = self . client . get ( ' /check_session/ ' )
self . assertEqual ( response . content , b ' foo ' )
2012-05-31 23:21:13 +08:00
def test_logout ( self ) :
""" Logout should work whether the user is logged in or not (#9978). """
self . client . logout ( )
2013-10-27 03:15:03 +08:00
login = self . client . login ( username = ' testclient ' , password = ' password ' )
2012-05-31 23:21:13 +08:00
self . assertTrue ( login , ' Could not log in ' )
self . client . logout ( )
self . client . logout ( )
2013-06-05 00:37:44 +08:00
def test_logout_with_user ( self ) :
""" Logout should send user_logged_out signal if user was logged in. """
def listener ( * args , * * kwargs ) :
listener . executed = True
self . assertEqual ( kwargs [ ' sender ' ] , User )
listener . executed = False
user_logged_out . connect ( listener )
self . client . login ( username = ' testclient ' , password = ' password ' )
self . client . logout ( )
user_logged_out . disconnect ( listener )
self . assertTrue ( listener . executed )
@override_settings ( AUTH_USER_MODEL = ' test_client_regress.CustomUser ' )
def test_logout_with_custom_user ( self ) :
""" Logout should send user_logged_out signal if custom user was logged in. """
def listener ( * args , * * kwargs ) :
self . assertEqual ( kwargs [ ' sender ' ] , CustomUser )
2014-06-06 17:19:16 +08:00
listener . executed = True
listener . executed = False
u = CustomUser . custom_objects . create ( email = ' test@test.com ' )
u . set_password ( ' password ' )
u . save ( )
user_logged_out . connect ( listener )
self . client . login ( username = ' test@test.com ' , password = ' password ' )
self . client . logout ( )
user_logged_out . disconnect ( listener )
self . assertTrue ( listener . executed )
@override_settings ( AUTHENTICATION_BACKENDS = (
' django.contrib.auth.backends.ModelBackend ' ,
' test_client_regress.auth_backends.CustomUserBackend ' ) )
def test_logout_with_custom_auth_backend ( self ) :
" Request a logout after logging in with custom authentication backend "
def listener ( * args , * * kwargs ) :
self . assertEqual ( kwargs [ ' sender ' ] , CustomUser )
2013-06-05 00:37:44 +08:00
listener . executed = True
listener . executed = False
u = CustomUser . custom_objects . create ( email = ' test@test.com ' )
u . set_password ( ' password ' )
u . save ( )
user_logged_out . connect ( listener )
self . client . login ( username = ' test@test.com ' , password = ' password ' )
self . client . logout ( )
user_logged_out . disconnect ( listener )
self . assertTrue ( listener . executed )
def test_logout_without_user ( self ) :
""" Logout should send signal even if user not authenticated. """
def listener ( user , * args , * * kwargs ) :
listener . user = user
listener . executed = True
listener . executed = False
user_logged_out . connect ( listener )
self . client . login ( username = ' incorrect ' , password = ' password ' )
self . client . logout ( )
user_logged_out . disconnect ( listener )
self . assertTrue ( listener . executed )
self . assertIsNone ( listener . user )
def test_login_with_user ( self ) :
""" Login should send user_logged_in signal on successful login. """
def listener ( * args , * * kwargs ) :
listener . executed = True
listener . executed = False
user_logged_in . connect ( listener )
self . client . login ( username = ' testclient ' , password = ' password ' )
user_logged_out . disconnect ( listener )
self . assertTrue ( listener . executed )
def test_login_without_signal ( self ) :
""" Login shouldn ' t send signal if user wasn ' t logged in """
def listener ( * args , * * kwargs ) :
listener . executed = True
listener . executed = False
user_logged_in . connect ( listener )
self . client . login ( username = ' incorrect ' , password = ' password ' )
user_logged_in . disconnect ( listener )
self . assertFalse ( listener . executed )
2014-04-05 14:04:46 +08:00
@override_settings ( ROOT_URLCONF = ' test_client_regress.urls ' )
2012-05-31 23:21:13 +08:00
class RequestMethodTests ( TestCase ) :
2014-01-14 23:43:27 +08:00
2012-05-31 23:21:13 +08:00
def test_get ( self ) :
" Request a view via request method GET "
2014-01-14 23:43:27 +08:00
response = self . client . get ( ' /request_methods/ ' )
2012-05-31 23:21:13 +08:00
self . assertEqual ( response . status_code , 200 )
self . assertEqual ( response . content , b ' request method: GET ' )
def test_post ( self ) :
" Request a view via request method POST "
2014-01-14 23:43:27 +08:00
response = self . client . post ( ' /request_methods/ ' )
2012-05-31 23:21:13 +08:00
self . assertEqual ( response . status_code , 200 )
self . assertEqual ( response . content , b ' request method: POST ' )
def test_head ( self ) :
" Request a view via request method HEAD "
2014-01-14 23:43:27 +08:00
response = self . client . head ( ' /request_methods/ ' )
2012-05-31 23:21:13 +08:00
self . assertEqual ( response . status_code , 200 )
# A HEAD request doesn't return any content.
self . assertNotEqual ( response . content , b ' request method: HEAD ' )
self . assertEqual ( response . content , b ' ' )
def test_options ( self ) :
" Request a view via request method OPTIONS "
2014-01-14 23:43:27 +08:00
response = self . client . options ( ' /request_methods/ ' )
2012-05-31 23:21:13 +08:00
self . assertEqual ( response . status_code , 200 )
self . assertEqual ( response . content , b ' request method: OPTIONS ' )
def test_put ( self ) :
" Request a view via request method PUT "
2014-01-14 23:43:27 +08:00
response = self . client . put ( ' /request_methods/ ' )
2012-05-31 23:21:13 +08:00
self . assertEqual ( response . status_code , 200 )
self . assertEqual ( response . content , b ' request method: PUT ' )
def test_delete ( self ) :
" Request a view via request method DELETE "
2014-01-14 23:43:27 +08:00
response = self . client . delete ( ' /request_methods/ ' )
2012-05-31 23:21:13 +08:00
self . assertEqual ( response . status_code , 200 )
self . assertEqual ( response . content , b ' request method: DELETE ' )
2013-02-03 10:22:40 +08:00
def test_patch ( self ) :
" Request a view via request method PATCH "
2014-01-14 23:43:27 +08:00
response = self . client . patch ( ' /request_methods/ ' )
2013-02-03 10:22:40 +08:00
self . assertEqual ( response . status_code , 200 )
self . assertEqual ( response . content , b ' request method: PATCH ' )
2014-04-05 14:04:46 +08:00
@override_settings ( ROOT_URLCONF = ' test_client_regress.urls ' )
2012-05-31 23:21:13 +08:00
class RequestMethodStringDataTests ( TestCase ) :
2014-01-14 23:43:27 +08:00
2012-05-31 23:21:13 +08:00
def test_post ( self ) :
" Request a view with string data via request method POST "
# Regression test for #11371
2012-06-08 00:08:47 +08:00
data = ' { " test " : " json " } '
2014-01-14 23:43:27 +08:00
response = self . client . post ( ' /request_methods/ ' , data = data , content_type = ' application/json ' )
2012-05-31 23:21:13 +08:00
self . assertEqual ( response . status_code , 200 )
self . assertEqual ( response . content , b ' request method: POST ' )
def test_put ( self ) :
" Request a view with string data via request method PUT "
# Regression test for #11371
2012-06-08 00:08:47 +08:00
data = ' { " test " : " json " } '
2014-01-14 23:43:27 +08:00
response = self . client . put ( ' /request_methods/ ' , data = data , content_type = ' application/json ' )
2012-05-31 23:21:13 +08:00
self . assertEqual ( response . status_code , 200 )
self . assertEqual ( response . content , b ' request method: PUT ' )
2013-02-03 10:22:40 +08:00
def test_patch ( self ) :
" Request a view with string data via request method PATCH "
# Regression test for #17797
2013-02-03 12:57:38 +08:00
data = ' { " test " : " json " } '
2014-01-14 23:43:27 +08:00
response = self . client . patch ( ' /request_methods/ ' , data = data , content_type = ' application/json ' )
2013-02-03 10:22:40 +08:00
self . assertEqual ( response . status_code , 200 )
self . assertEqual ( response . content , b ' request method: PATCH ' )
2014-10-21 04:32:43 +08:00
def test_empty_string_data ( self ) :
" Request a view with empty string data via request method GET/POST/HEAD "
# Regression test for #21740
response = self . client . get ( ' /body/ ' , data = ' ' , content_type = ' application/json ' )
self . assertEqual ( response . content , b ' ' )
response = self . client . post ( ' /body/ ' , data = ' ' , content_type = ' application/json ' )
self . assertEqual ( response . content , b ' ' )
response = self . client . head ( ' /body/ ' , data = ' ' , content_type = ' application/json ' )
self . assertEqual ( response . content , b ' ' )
2013-11-03 05:34:05 +08:00
2014-04-05 14:04:46 +08:00
@override_settings ( ROOT_URLCONF = ' test_client_regress.urls ' , )
2012-05-31 23:21:13 +08:00
class QueryStringTests ( TestCase ) :
2014-01-14 23:43:27 +08:00
2012-05-31 23:21:13 +08:00
def test_get_like_requests ( self ) :
# See: https://code.djangoproject.com/ticket/10571.
for method_name in ( ' get ' , ' head ' ) :
# A GET-like request can pass a query string as data
method = getattr ( self . client , method_name )
2014-01-14 23:43:27 +08:00
response = method ( " /request_data/ " , data = { ' foo ' : ' whiz ' } )
2012-05-31 23:21:13 +08:00
self . assertEqual ( response . context [ ' get-foo ' ] , ' whiz ' )
self . assertEqual ( response . context [ ' request-foo ' ] , ' whiz ' )
# A GET-like request can pass a query string as part of the URL
2014-01-14 23:43:27 +08:00
response = method ( " /request_data/?foo=whiz " )
2012-05-31 23:21:13 +08:00
self . assertEqual ( response . context [ ' get-foo ' ] , ' whiz ' )
self . assertEqual ( response . context [ ' request-foo ' ] , ' whiz ' )
# Data provided in the URL to a GET-like request is overridden by actual form data
2014-01-14 23:43:27 +08:00
response = method ( " /request_data/?foo=whiz " , data = { ' foo ' : ' bang ' } )
2012-05-31 23:21:13 +08:00
self . assertEqual ( response . context [ ' get-foo ' ] , ' bang ' )
self . assertEqual ( response . context [ ' request-foo ' ] , ' bang ' )
2014-01-14 23:43:27 +08:00
response = method ( " /request_data/?foo=whiz " , data = { ' bar ' : ' bang ' } )
2012-05-31 23:21:13 +08:00
self . assertEqual ( response . context [ ' get-foo ' ] , None )
self . assertEqual ( response . context [ ' get-bar ' ] , ' bang ' )
self . assertEqual ( response . context [ ' request-foo ' ] , None )
self . assertEqual ( response . context [ ' request-bar ' ] , ' bang ' )
def test_post_like_requests ( self ) :
# A POST-like request can pass a query string as data
2014-01-14 23:43:27 +08:00
response = self . client . post ( " /request_data/ " , data = { ' foo ' : ' whiz ' } )
2012-05-31 23:21:13 +08:00
self . assertEqual ( response . context [ ' get-foo ' ] , None )
self . assertEqual ( response . context [ ' post-foo ' ] , ' whiz ' )
# A POST-like request can pass a query string as part of the URL
2014-01-14 23:43:27 +08:00
response = self . client . post ( " /request_data/?foo=whiz " )
2012-05-31 23:21:13 +08:00
self . assertEqual ( response . context [ ' get-foo ' ] , ' whiz ' )
self . assertEqual ( response . context [ ' post-foo ' ] , None )
self . assertEqual ( response . context [ ' request-foo ' ] , ' whiz ' )
# POST data provided in the URL augments actual form data
2014-01-14 23:43:27 +08:00
response = self . client . post ( " /request_data/?foo=whiz " , data = { ' foo ' : ' bang ' } )
2012-05-31 23:21:13 +08:00
self . assertEqual ( response . context [ ' get-foo ' ] , ' whiz ' )
self . assertEqual ( response . context [ ' post-foo ' ] , ' bang ' )
self . assertEqual ( response . context [ ' request-foo ' ] , ' bang ' )
2014-01-14 23:43:27 +08:00
response = self . client . post ( " /request_data/?foo=whiz " , data = { ' bar ' : ' bang ' } )
2012-05-31 23:21:13 +08:00
self . assertEqual ( response . context [ ' get-foo ' ] , ' whiz ' )
self . assertEqual ( response . context [ ' get-bar ' ] , None )
self . assertEqual ( response . context [ ' post-foo ' ] , None )
self . assertEqual ( response . context [ ' post-bar ' ] , ' bang ' )
self . assertEqual ( response . context [ ' request-foo ' ] , ' whiz ' )
self . assertEqual ( response . context [ ' request-bar ' ] , ' bang ' )
2013-11-03 05:34:05 +08:00
2014-04-05 14:04:46 +08:00
@override_settings ( ROOT_URLCONF = ' test_client_regress.urls ' )
2012-05-31 23:21:13 +08:00
class UnicodePayloadTests ( TestCase ) :
2014-01-14 23:43:27 +08:00
2012-05-31 23:21:13 +08:00
def test_simple_unicode_payload ( self ) :
" A simple ASCII-only unicode JSON document can be POSTed "
# Regression test for #10571
2012-06-08 00:08:47 +08:00
json = ' { " english " : " mountain pass " } '
2014-01-14 23:43:27 +08:00
response = self . client . post ( " /parse_unicode_json/ " , json ,
2012-05-31 23:21:13 +08:00
content_type = " application/json " )
2012-08-15 16:57:05 +08:00
self . assertEqual ( response . content , json . encode ( ) )
2012-05-31 23:21:13 +08:00
def test_unicode_payload_utf8 ( self ) :
" A non-ASCII unicode data encoded as UTF-8 can be POSTed "
# Regression test for #10571
2012-06-08 00:08:47 +08:00
json = ' { " dog " : " собака " } '
2014-01-14 23:43:27 +08:00
response = self . client . post ( " /parse_unicode_json/ " , json ,
2012-05-31 23:21:13 +08:00
content_type = " application/json; charset=utf-8 " )
self . assertEqual ( response . content , json . encode ( ' utf-8 ' ) )
def test_unicode_payload_utf16 ( self ) :
" A non-ASCII unicode data encoded as UTF-16 can be POSTed "
# Regression test for #10571
2012-06-08 00:08:47 +08:00
json = ' { " dog " : " собака " } '
2014-01-14 23:43:27 +08:00
response = self . client . post ( " /parse_unicode_json/ " , json ,
2012-05-31 23:21:13 +08:00
content_type = " application/json; charset=utf-16 " )
self . assertEqual ( response . content , json . encode ( ' utf-16 ' ) )
def test_unicode_payload_non_utf ( self ) :
" A non-ASCII unicode data as a non-UTF based encoding can be POSTed "
2013-11-03 05:02:56 +08:00
# Regression test for #10571
2012-06-08 00:08:47 +08:00
json = ' { " dog " : " собака " } '
2014-01-14 23:43:27 +08:00
response = self . client . post ( " /parse_unicode_json/ " , json ,
2012-05-31 23:21:13 +08:00
content_type = " application/json; charset=koi8-r " )
self . assertEqual ( response . content , json . encode ( ' koi8-r ' ) )
2013-11-03 05:34:05 +08:00
2012-05-31 23:21:13 +08:00
class DummyFile ( object ) :
def __init__ ( self , filename ) :
self . name = filename
2013-10-22 18:21:07 +08:00
2012-05-31 23:21:13 +08:00
def read ( self ) :
2012-08-15 16:57:05 +08:00
return b ' TEST_FILE_CONTENT '
2012-05-31 23:21:13 +08:00
2013-11-03 05:34:05 +08:00
2012-05-31 23:21:13 +08:00
class UploadedFileEncodingTest ( TestCase ) :
def test_file_encoding ( self ) :
encoded_file = encode_file ( ' TEST_BOUNDARY ' , ' TEST_KEY ' , DummyFile ( ' test_name.bin ' ) )
self . assertEqual ( b ' --TEST_BOUNDARY ' , encoded_file [ 0 ] )
self . assertEqual ( b ' Content-Disposition: form-data; name= " TEST_KEY " ; filename= " test_name.bin " ' , encoded_file [ 1 ] )
self . assertEqual ( b ' TEST_FILE_CONTENT ' , encoded_file [ - 1 ] )
def test_guesses_content_type_on_file_encoding ( self ) :
self . assertEqual ( b ' Content-Type: application/octet-stream ' ,
encode_file ( ' IGNORE ' , ' IGNORE ' , DummyFile ( " file.bin " ) ) [ 2 ] )
self . assertEqual ( b ' Content-Type: text/plain ' ,
encode_file ( ' IGNORE ' , ' IGNORE ' , DummyFile ( " file.txt " ) ) [ 2 ] )
self . assertIn ( encode_file ( ' IGNORE ' , ' IGNORE ' , DummyFile ( " file.zip " ) ) [ 2 ] , (
2013-10-20 07:33:10 +08:00
b ' Content-Type: application/x-compress ' ,
b ' Content-Type: application/x-zip ' ,
b ' Content-Type: application/x-zip-compressed ' ,
b ' Content-Type: application/zip ' , ) )
2012-05-31 23:21:13 +08:00
self . assertEqual ( b ' Content-Type: application/octet-stream ' ,
encode_file ( ' IGNORE ' , ' IGNORE ' , DummyFile ( " file.unknown " ) ) [ 2 ] )
2013-11-03 05:34:05 +08:00
2014-04-05 14:04:46 +08:00
@override_settings ( ROOT_URLCONF = ' test_client_regress.urls ' , )
2012-05-31 23:21:13 +08:00
class RequestHeadersTest ( TestCase ) :
def test_client_headers ( self ) :
" A test client can receive custom headers "
2014-01-14 23:43:27 +08:00
response = self . client . get ( " /check_headers/ " , HTTP_X_ARG_CHECK = ' Testing 123 ' )
2012-05-31 23:21:13 +08:00
self . assertEqual ( response . content , b " HTTP_X_ARG_CHECK: Testing 123 " )
self . assertEqual ( response . status_code , 200 )
def test_client_headers_redirect ( self ) :
" Test client headers are preserved through redirects "
2014-01-14 23:43:27 +08:00
response = self . client . get ( " /check_headers_redirect/ " , follow = True , HTTP_X_ARG_CHECK = ' Testing 123 ' )
2012-05-31 23:21:13 +08:00
self . assertEqual ( response . content , b " HTTP_X_ARG_CHECK: Testing 123 " )
2014-01-14 23:43:27 +08:00
self . assertRedirects ( response , ' /check_headers/ ' ,
2012-05-31 23:21:13 +08:00
status_code = 301 , target_status_code = 200 )
2014-04-05 14:04:46 +08:00
@override_settings ( ROOT_URLCONF = ' test_client_regress.urls ' )
2012-05-31 23:21:13 +08:00
class ReadLimitedStreamTest ( TestCase ) :
"""
Tests that ensure that HttpRequest . body , HttpRequest . read ( ) and
HttpRequest . read ( BUFFER ) have proper LimitedStream behavior .
Refs #14753, #15785
"""
def test_body_from_empty_request ( self ) :
""" HttpRequest.body on a test client GET request should return
the empty string . """
2014-01-14 23:43:27 +08:00
self . assertEqual ( self . client . get ( " /body/ " ) . content , b ' ' )
2012-05-31 23:21:13 +08:00
def test_read_from_empty_request ( self ) :
""" HttpRequest.read() on a test client GET request should return the
empty string . """
2014-01-14 23:43:27 +08:00
self . assertEqual ( self . client . get ( " /read_all/ " ) . content , b ' ' )
2012-05-31 23:21:13 +08:00
def test_read_numbytes_from_empty_request ( self ) :
""" HttpRequest.read(LARGE_BUFFER) on a test client GET request should
return the empty string . """
2014-01-14 23:43:27 +08:00
self . assertEqual ( self . client . get ( " /read_buffer/ " ) . content , b ' ' )
2012-05-31 23:21:13 +08:00
def test_read_from_nonempty_request ( self ) :
""" HttpRequest.read() on a test client PUT request with some payload
should return that payload . """
payload = b ' foobar '
2013-11-03 05:34:05 +08:00
self . assertEqual ( self . client . put (
2014-01-14 23:43:27 +08:00
" /read_all/ " ,
2013-11-03 05:34:05 +08:00
data = payload ,
content_type = ' text/plain ' ) . content , payload )
2012-05-31 23:21:13 +08:00
def test_read_numbytes_from_nonempty_request ( self ) :
""" HttpRequest.read(LARGE_BUFFER) on a test client PUT request with
some payload should return that payload . """
payload = b ' foobar '
2013-11-03 05:34:05 +08:00
self . assertEqual (
2014-01-14 23:43:27 +08:00
self . client . put ( " /read_buffer/ " ,
2013-11-03 05:34:05 +08:00
data = payload ,
content_type = ' text/plain ' ) . content , payload )
2012-05-31 23:21:13 +08:00
2014-04-05 14:04:46 +08:00
@override_settings ( ROOT_URLCONF = ' test_client_regress.urls ' )
2012-05-31 23:21:13 +08:00
class RequestFactoryStateTest ( TestCase ) :
""" Regression tests for #15929. """
# These tests are checking that certain middleware don't change certain
# global state. Alternatively, from the point of view of a test, they are
# ensuring test isolation behavior. So, unusually, it doesn't make sense to
# run the tests individually, and if any are failing it is confusing to run
# them with any other set of tests.
def common_test_that_should_always_pass ( self ) :
2012-05-31 23:41:31 +08:00
request = RequestFactory ( ) . get ( ' / ' )
2012-05-31 23:21:13 +08:00
request . session = { }
self . assertFalse ( hasattr ( request , ' user ' ) )
def test_request ( self ) :
self . common_test_that_should_always_pass ( )
def test_request_after_client ( self ) :
# apart from the next line the three tests are identical
self . client . get ( ' / ' )
self . common_test_that_should_always_pass ( )
def test_request_after_client_2 ( self ) :
# This test is executed after the previous one
self . common_test_that_should_always_pass ( )
2014-04-05 14:04:46 +08:00
@override_settings ( ROOT_URLCONF = ' test_client_regress.urls ' )
2012-05-31 23:21:13 +08:00
class RequestFactoryEnvironmentTests ( TestCase ) :
"""
Regression tests for #8551 and #17067: ensure that environment variables
are set correctly in RequestFactory .
"""
def test_should_set_correct_env_variables ( self ) :
2012-05-31 23:41:31 +08:00
request = RequestFactory ( ) . get ( ' /path/ ' )
2012-05-31 23:21:13 +08:00
self . assertEqual ( request . META . get ( ' REMOTE_ADDR ' ) , ' 127.0.0.1 ' )
self . assertEqual ( request . META . get ( ' SERVER_NAME ' ) , ' testserver ' )
self . assertEqual ( request . META . get ( ' SERVER_PORT ' ) , ' 80 ' )
self . assertEqual ( request . META . get ( ' SERVER_PROTOCOL ' ) , ' HTTP/1.1 ' )
self . assertEqual ( request . META . get ( ' SCRIPT_NAME ' ) +
request . META . get ( ' PATH_INFO ' ) , ' /path/ ' )