django-rest-framework/djangorestframework/tests/methods.py

53 lines
2.5 KiB
Python

from django.test import TestCase
from djangorestframework.compat import RequestFactory
from djangorestframework.methods import MethodMixin, StandardMethodMixin, OverloadedPOSTMethodMixin
class TestMethodMixins(TestCase):
def setUp(self):
self.req = RequestFactory()
# Interface tests
def test_method_mixin_interface(self):
"""Ensure the base ContentMixin interface is as expected."""
self.assertRaises(NotImplementedError, MethodMixin().determine_method, None)
def test_standard_method_mixin_interface(self):
"""Ensure the StandardMethodMixin interface is as expected."""
self.assertTrue(issubclass(StandardMethodMixin, MethodMixin))
getattr(StandardMethodMixin, 'determine_method')
def test_overloaded_method_mixin_interface(self):
"""Ensure the OverloadedPOSTMethodMixin interface is as expected."""
self.assertTrue(issubclass(OverloadedPOSTMethodMixin, MethodMixin))
getattr(OverloadedPOSTMethodMixin, 'METHOD_PARAM')
getattr(OverloadedPOSTMethodMixin, 'determine_method')
# Behavioural tests
def test_standard_behaviour_determines_GET(self):
"""GET requests identified as GET method with StandardMethodMixin"""
request = self.req.get('/')
self.assertEqual(StandardMethodMixin().determine_method(request), 'GET')
def test_standard_behaviour_determines_POST(self):
"""POST requests identified as POST method with StandardMethodMixin"""
request = self.req.post('/')
self.assertEqual(StandardMethodMixin().determine_method(request), 'POST')
def test_overloaded_POST_behaviour_determines_GET(self):
"""GET requests identified as GET method with OverloadedPOSTMethodMixin"""
request = self.req.get('/')
self.assertEqual(OverloadedPOSTMethodMixin().determine_method(request), 'GET')
def test_overloaded_POST_behaviour_determines_POST(self):
"""POST requests identified as POST method with OverloadedPOSTMethodMixin"""
request = self.req.post('/')
self.assertEqual(OverloadedPOSTMethodMixin().determine_method(request), 'POST')
def test_overloaded_POST_behaviour_determines_overloaded_method(self):
"""POST requests can be overloaded to another method by setting a reserved form field with OverloadedPOSTMethodMixin"""
request = self.req.post('/', {OverloadedPOSTMethodMixin.METHOD_PARAM: 'DELETE'})
self.assertEqual(OverloadedPOSTMethodMixin().determine_method(request), 'DELETE')