mirror of
https://github.com/encode/django-rest-framework.git
synced 2025-06-22 22:43:17 +03:00
Refactor request tests style
This commit is contained in:
parent
bd86104861
commit
9a6352789e
|
@ -42,14 +42,14 @@ class TestContentParsing(TestCase):
|
||||||
Ensure request.data returns empty QueryDict for GET request.
|
Ensure request.data returns empty QueryDict for GET request.
|
||||||
"""
|
"""
|
||||||
request = Request(factory.get('/'))
|
request = Request(factory.get('/'))
|
||||||
self.assertEqual(request.data, {})
|
assert request.data == {}
|
||||||
|
|
||||||
def test_standard_behaviour_determines_no_content_HEAD(self):
|
def test_standard_behaviour_determines_no_content_HEAD(self):
|
||||||
"""
|
"""
|
||||||
Ensure request.data returns empty QueryDict for HEAD request.
|
Ensure request.data returns empty QueryDict for HEAD request.
|
||||||
"""
|
"""
|
||||||
request = Request(factory.head('/'))
|
request = Request(factory.head('/'))
|
||||||
self.assertEqual(request.data, {})
|
assert request.data == {}
|
||||||
|
|
||||||
def test_request_DATA_with_form_content(self):
|
def test_request_DATA_with_form_content(self):
|
||||||
"""
|
"""
|
||||||
|
@ -58,7 +58,7 @@ class TestContentParsing(TestCase):
|
||||||
data = {'qwerty': 'uiop'}
|
data = {'qwerty': 'uiop'}
|
||||||
request = Request(factory.post('/', data))
|
request = Request(factory.post('/', data))
|
||||||
request.parsers = (FormParser(), MultiPartParser())
|
request.parsers = (FormParser(), MultiPartParser())
|
||||||
self.assertEqual(list(request.data.items()), list(data.items()))
|
assert list(request.data.items()) == list(data.items())
|
||||||
|
|
||||||
def test_request_DATA_with_text_content(self):
|
def test_request_DATA_with_text_content(self):
|
||||||
"""
|
"""
|
||||||
|
@ -69,7 +69,7 @@ class TestContentParsing(TestCase):
|
||||||
content_type = 'text/plain'
|
content_type = 'text/plain'
|
||||||
request = Request(factory.post('/', content, content_type=content_type))
|
request = Request(factory.post('/', content, content_type=content_type))
|
||||||
request.parsers = (PlainTextParser(),)
|
request.parsers = (PlainTextParser(),)
|
||||||
self.assertEqual(request.data, content)
|
assert request.data == content
|
||||||
|
|
||||||
def test_request_POST_with_form_content(self):
|
def test_request_POST_with_form_content(self):
|
||||||
"""
|
"""
|
||||||
|
@ -78,7 +78,7 @@ class TestContentParsing(TestCase):
|
||||||
data = {'qwerty': 'uiop'}
|
data = {'qwerty': 'uiop'}
|
||||||
request = Request(factory.post('/', data))
|
request = Request(factory.post('/', data))
|
||||||
request.parsers = (FormParser(), MultiPartParser())
|
request.parsers = (FormParser(), MultiPartParser())
|
||||||
self.assertEqual(list(request.POST.items()), list(data.items()))
|
assert list(request.POST.items()) == list(data.items())
|
||||||
|
|
||||||
def test_request_POST_with_files(self):
|
def test_request_POST_with_files(self):
|
||||||
"""
|
"""
|
||||||
|
@ -87,8 +87,8 @@ class TestContentParsing(TestCase):
|
||||||
upload = SimpleUploadedFile("file.txt", b"file_content")
|
upload = SimpleUploadedFile("file.txt", b"file_content")
|
||||||
request = Request(factory.post('/', {'upload': upload}))
|
request = Request(factory.post('/', {'upload': upload}))
|
||||||
request.parsers = (FormParser(), MultiPartParser())
|
request.parsers = (FormParser(), MultiPartParser())
|
||||||
self.assertEqual(list(request.POST.keys()), [])
|
assert list(request.POST.keys()) == []
|
||||||
self.assertEqual(list(request.FILES.keys()), ['upload'])
|
assert list(request.FILES.keys()) == ['upload']
|
||||||
|
|
||||||
def test_standard_behaviour_determines_form_content_PUT(self):
|
def test_standard_behaviour_determines_form_content_PUT(self):
|
||||||
"""
|
"""
|
||||||
|
@ -97,7 +97,7 @@ class TestContentParsing(TestCase):
|
||||||
data = {'qwerty': 'uiop'}
|
data = {'qwerty': 'uiop'}
|
||||||
request = Request(factory.put('/', data))
|
request = Request(factory.put('/', data))
|
||||||
request.parsers = (FormParser(), MultiPartParser())
|
request.parsers = (FormParser(), MultiPartParser())
|
||||||
self.assertEqual(list(request.data.items()), list(data.items()))
|
assert list(request.data.items()) == list(data.items())
|
||||||
|
|
||||||
def test_standard_behaviour_determines_non_form_content_PUT(self):
|
def test_standard_behaviour_determines_non_form_content_PUT(self):
|
||||||
"""
|
"""
|
||||||
|
@ -108,7 +108,7 @@ class TestContentParsing(TestCase):
|
||||||
content_type = 'text/plain'
|
content_type = 'text/plain'
|
||||||
request = Request(factory.put('/', content, content_type=content_type))
|
request = Request(factory.put('/', content, content_type=content_type))
|
||||||
request.parsers = (PlainTextParser(), )
|
request.parsers = (PlainTextParser(), )
|
||||||
self.assertEqual(request.data, content)
|
assert request.data == content
|
||||||
|
|
||||||
|
|
||||||
class MockView(APIView):
|
class MockView(APIView):
|
||||||
|
@ -142,10 +142,10 @@ class TestContentParsingWithAuthentication(TestCase):
|
||||||
content = {'example': 'example'}
|
content = {'example': 'example'}
|
||||||
|
|
||||||
response = self.client.post('/', content)
|
response = self.client.post('/', content)
|
||||||
self.assertEqual(status.HTTP_200_OK, response.status_code)
|
assert status.HTTP_200_OK == response.status_code
|
||||||
|
|
||||||
response = self.csrf_client.post('/', content)
|
response = self.csrf_client.post('/', content)
|
||||||
self.assertEqual(status.HTTP_200_OK, response.status_code)
|
assert status.HTTP_200_OK == response.status_code
|
||||||
|
|
||||||
|
|
||||||
class TestUserSetter(TestCase):
|
class TestUserSetter(TestCase):
|
||||||
|
@ -162,21 +162,21 @@ class TestUserSetter(TestCase):
|
||||||
|
|
||||||
def test_user_can_be_set(self):
|
def test_user_can_be_set(self):
|
||||||
self.request.user = self.user
|
self.request.user = self.user
|
||||||
self.assertEqual(self.request.user, self.user)
|
assert self.request.user == self.user
|
||||||
|
|
||||||
def test_user_can_login(self):
|
def test_user_can_login(self):
|
||||||
login(self.request, self.user)
|
login(self.request, self.user)
|
||||||
self.assertEqual(self.request.user, self.user)
|
assert self.request.user == self.user
|
||||||
|
|
||||||
def test_user_can_logout(self):
|
def test_user_can_logout(self):
|
||||||
self.request.user = self.user
|
self.request.user = self.user
|
||||||
self.assertFalse(is_anonymous(self.request.user))
|
assert is_anonymous(self.request.user) is False
|
||||||
logout(self.request)
|
logout(self.request)
|
||||||
self.assertTrue(is_anonymous(self.request.user))
|
assert is_anonymous(self.request.user) is True
|
||||||
|
|
||||||
def test_logged_in_user_is_set_on_wrapped_request(self):
|
def test_logged_in_user_is_set_on_wrapped_request(self):
|
||||||
login(self.request, self.user)
|
login(self.request, self.user)
|
||||||
self.assertEqual(self.wrapped_request.user, self.user)
|
assert self.wrapped_request.user == self.user
|
||||||
|
|
||||||
def test_calling_user_fails_when_attribute_error_is_raised(self):
|
def test_calling_user_fails_when_attribute_error_is_raised(self):
|
||||||
"""
|
"""
|
||||||
|
@ -207,15 +207,15 @@ class TestAuthSetter(TestCase):
|
||||||
def test_auth_can_be_set(self):
|
def test_auth_can_be_set(self):
|
||||||
request = Request(factory.get('/'))
|
request = Request(factory.get('/'))
|
||||||
request.auth = 'DUMMY'
|
request.auth = 'DUMMY'
|
||||||
self.assertEqual(request.auth, 'DUMMY')
|
assert request.auth == 'DUMMY'
|
||||||
|
|
||||||
|
|
||||||
class TestSecure(TestCase):
|
class TestSecure(TestCase):
|
||||||
|
|
||||||
def test_default_secure_false(self):
|
def test_default_secure_false(self):
|
||||||
request = Request(factory.get('/', secure=False))
|
request = Request(factory.get('/', secure=False))
|
||||||
self.assertEqual(request.scheme, 'http')
|
assert request.scheme == 'http'
|
||||||
|
|
||||||
def test_default_secure_true(self):
|
def test_default_secure_true(self):
|
||||||
request = Request(factory.get('/', secure=True))
|
request = Request(factory.get('/', secure=True))
|
||||||
self.assertEqual(request.scheme, 'https')
|
assert request.scheme == 'https'
|
||||||
|
|
Loading…
Reference in New Issue
Block a user