81 lines
2.8 KiB
Python
81 lines
2.8 KiB
Python
from __future__ import absolute_import
|
|
from __future__ import unicode_literals
|
|
|
|
from django.core.exceptions import ImproperlyConfigured
|
|
from django.test import TestCase, override_settings
|
|
from django.test.client import RequestFactory
|
|
|
|
from django_filters.views import FilterView
|
|
from django_filters.filterset import FilterSet, filterset_factory
|
|
|
|
from .models import Book
|
|
|
|
|
|
@override_settings(ROOT_URLCONF='tests.urls')
|
|
class GenericViewTestCase(TestCase):
|
|
|
|
def setUp(self):
|
|
Book.objects.create(
|
|
title="Ender's Game", price='1.00', average_rating=3.0)
|
|
Book.objects.create(
|
|
title="Rainbow Six", price='1.00', average_rating=3.0)
|
|
Book.objects.create(
|
|
title="Snowcrash", price='1.00', average_rating=3.0)
|
|
|
|
|
|
class GenericClassBasedViewTests(GenericViewTestCase):
|
|
base_url = '/books/'
|
|
|
|
def test_view(self):
|
|
response = self.client.get(self.base_url)
|
|
for b in ['Ender's Game', 'Rainbow Six', 'Snowcrash']:
|
|
self.assertContains(response, b)
|
|
|
|
def test_view_filtering_on_title(self):
|
|
response = self.client.get(self.base_url + '?title=Snowcrash')
|
|
for b in ['Ender's Game', 'Rainbow Six']:
|
|
self.assertNotContains(response, b)
|
|
self.assertContains(response, 'Snowcrash')
|
|
|
|
def test_view_with_filterset_not_model(self):
|
|
factory = RequestFactory()
|
|
request = factory.get(self.base_url)
|
|
filterset = filterset_factory(Book)
|
|
view = FilterView.as_view(filterset_class=filterset)
|
|
response = view(request)
|
|
self.assertEqual(response.status_code, 200)
|
|
for b in ['Ender's Game', 'Rainbow Six', 'Snowcrash']:
|
|
self.assertContains(response, b)
|
|
|
|
def test_view_without_filterset_or_model(self):
|
|
factory = RequestFactory()
|
|
request = factory.get(self.base_url)
|
|
view = FilterView.as_view()
|
|
with self.assertRaises(ImproperlyConfigured):
|
|
view(request)
|
|
|
|
def test_view_with_bad_filterset(self):
|
|
class MyFilterSet(FilterSet):
|
|
pass
|
|
|
|
factory = RequestFactory()
|
|
request = factory.get(self.base_url)
|
|
view = FilterView.as_view(filterset_class=MyFilterSet)
|
|
with self.assertRaises(ImproperlyConfigured):
|
|
view(request)
|
|
|
|
|
|
class GenericFunctionalViewTests(GenericViewTestCase):
|
|
base_url = '/books-legacy/'
|
|
|
|
def test_view(self):
|
|
response = self.client.get(self.base_url)
|
|
for b in ['Ender's Game', 'Rainbow Six', 'Snowcrash']:
|
|
self.assertContains(response, b)
|
|
|
|
def test_view_filtering_on_price(self):
|
|
response = self.client.get(self.base_url + '?title=Snowcrash')
|
|
for b in ['Ender's Game', 'Rainbow Six']:
|
|
self.assertNotContains(response, b)
|
|
self.assertContains(response, 'Snowcrash')
|