1. import datetime
    
  2. 
    
  3. from django.core.exceptions import ImproperlyConfigured, ObjectDoesNotExist
    
  4. from django.test import TestCase, override_settings
    
  5. from django.test.client import RequestFactory
    
  6. from django.views.generic.base import View
    
  7. from django.views.generic.detail import SingleObjectTemplateResponseMixin
    
  8. from django.views.generic.edit import ModelFormMixin
    
  9. 
    
  10. from .models import Artist, Author, Book, Page
    
  11. 
    
  12. 
    
  13. @override_settings(ROOT_URLCONF="generic_views.urls")
    
  14. class DetailViewTest(TestCase):
    
  15.     @classmethod
    
  16.     def setUpTestData(cls):
    
  17.         cls.artist1 = Artist.objects.create(name="Rene Magritte")
    
  18.         cls.author1 = Author.objects.create(
    
  19.             name="Roberto BolaƱo", slug="roberto-bolano"
    
  20.         )
    
  21.         cls.author2 = Author.objects.create(
    
  22.             name="Scott Rosenberg", slug="scott-rosenberg"
    
  23.         )
    
  24.         cls.book1 = Book.objects.create(
    
  25.             name="2066", slug="2066", pages=800, pubdate=datetime.date(2008, 10, 1)
    
  26.         )
    
  27.         cls.book1.authors.add(cls.author1)
    
  28.         cls.book2 = Book.objects.create(
    
  29.             name="Dreaming in Code",
    
  30.             slug="dreaming-in-code",
    
  31.             pages=300,
    
  32.             pubdate=datetime.date(2006, 5, 1),
    
  33.         )
    
  34.         cls.page1 = Page.objects.create(
    
  35.             content="I was once bitten by a moose.",
    
  36.             template="generic_views/page_template.html",
    
  37.         )
    
  38. 
    
  39.     def test_simple_object(self):
    
  40.         res = self.client.get("/detail/obj/")
    
  41.         self.assertEqual(res.status_code, 200)
    
  42.         self.assertEqual(res.context["object"], {"foo": "bar"})
    
  43.         self.assertIsInstance(res.context["view"], View)
    
  44.         self.assertTemplateUsed(res, "generic_views/detail.html")
    
  45. 
    
  46.     def test_detail_by_pk(self):
    
  47.         res = self.client.get("/detail/author/%s/" % self.author1.pk)
    
  48.         self.assertEqual(res.status_code, 200)
    
  49.         self.assertEqual(res.context["object"], self.author1)
    
  50.         self.assertEqual(res.context["author"], self.author1)
    
  51.         self.assertTemplateUsed(res, "generic_views/author_detail.html")
    
  52. 
    
  53.     def test_detail_missing_object(self):
    
  54.         res = self.client.get("/detail/author/500/")
    
  55.         self.assertEqual(res.status_code, 404)
    
  56. 
    
  57.     def test_detail_object_does_not_exist(self):
    
  58.         with self.assertRaises(ObjectDoesNotExist):
    
  59.             self.client.get("/detail/doesnotexist/1/")
    
  60. 
    
  61.     def test_detail_by_custom_pk(self):
    
  62.         res = self.client.get("/detail/author/bycustompk/%s/" % self.author1.pk)
    
  63.         self.assertEqual(res.status_code, 200)
    
  64.         self.assertEqual(res.context["object"], self.author1)
    
  65.         self.assertEqual(res.context["author"], self.author1)
    
  66.         self.assertTemplateUsed(res, "generic_views/author_detail.html")
    
  67. 
    
  68.     def test_detail_by_slug(self):
    
  69.         res = self.client.get("/detail/author/byslug/scott-rosenberg/")
    
  70.         self.assertEqual(res.status_code, 200)
    
  71.         self.assertEqual(
    
  72.             res.context["object"], Author.objects.get(slug="scott-rosenberg")
    
  73.         )
    
  74.         self.assertEqual(
    
  75.             res.context["author"], Author.objects.get(slug="scott-rosenberg")
    
  76.         )
    
  77.         self.assertTemplateUsed(res, "generic_views/author_detail.html")
    
  78. 
    
  79.     def test_detail_by_custom_slug(self):
    
  80.         res = self.client.get("/detail/author/bycustomslug/scott-rosenberg/")
    
  81.         self.assertEqual(res.status_code, 200)
    
  82.         self.assertEqual(
    
  83.             res.context["object"], Author.objects.get(slug="scott-rosenberg")
    
  84.         )
    
  85.         self.assertEqual(
    
  86.             res.context["author"], Author.objects.get(slug="scott-rosenberg")
    
  87.         )
    
  88.         self.assertTemplateUsed(res, "generic_views/author_detail.html")
    
  89. 
    
  90.     def test_detail_by_pk_ignore_slug(self):
    
  91.         res = self.client.get(
    
  92.             "/detail/author/bypkignoreslug/%s-roberto-bolano/" % self.author1.pk
    
  93.         )
    
  94.         self.assertEqual(res.status_code, 200)
    
  95.         self.assertEqual(res.context["object"], self.author1)
    
  96.         self.assertEqual(res.context["author"], self.author1)
    
  97.         self.assertTemplateUsed(res, "generic_views/author_detail.html")
    
  98. 
    
  99.     def test_detail_by_pk_ignore_slug_mismatch(self):
    
  100.         res = self.client.get(
    
  101.             "/detail/author/bypkignoreslug/%s-scott-rosenberg/" % self.author1.pk
    
  102.         )
    
  103.         self.assertEqual(res.status_code, 200)
    
  104.         self.assertEqual(res.context["object"], self.author1)
    
  105.         self.assertEqual(res.context["author"], self.author1)
    
  106.         self.assertTemplateUsed(res, "generic_views/author_detail.html")
    
  107. 
    
  108.     def test_detail_by_pk_and_slug(self):
    
  109.         res = self.client.get(
    
  110.             "/detail/author/bypkandslug/%s-roberto-bolano/" % self.author1.pk
    
  111.         )
    
  112.         self.assertEqual(res.status_code, 200)
    
  113.         self.assertEqual(res.context["object"], self.author1)
    
  114.         self.assertEqual(res.context["author"], self.author1)
    
  115.         self.assertTemplateUsed(res, "generic_views/author_detail.html")
    
  116. 
    
  117.     def test_detail_by_pk_and_slug_mismatch_404(self):
    
  118.         res = self.client.get(
    
  119.             "/detail/author/bypkandslug/%s-scott-rosenberg/" % self.author1.pk
    
  120.         )
    
  121.         self.assertEqual(res.status_code, 404)
    
  122. 
    
  123.     def test_verbose_name(self):
    
  124.         res = self.client.get("/detail/artist/%s/" % self.artist1.pk)
    
  125.         self.assertEqual(res.status_code, 200)
    
  126.         self.assertEqual(res.context["object"], self.artist1)
    
  127.         self.assertEqual(res.context["artist"], self.artist1)
    
  128.         self.assertTemplateUsed(res, "generic_views/artist_detail.html")
    
  129. 
    
  130.     def test_template_name(self):
    
  131.         res = self.client.get("/detail/author/%s/template_name/" % self.author1.pk)
    
  132.         self.assertEqual(res.status_code, 200)
    
  133.         self.assertEqual(res.context["object"], self.author1)
    
  134.         self.assertEqual(res.context["author"], self.author1)
    
  135.         self.assertTemplateUsed(res, "generic_views/about.html")
    
  136. 
    
  137.     def test_template_name_suffix(self):
    
  138.         res = self.client.get(
    
  139.             "/detail/author/%s/template_name_suffix/" % self.author1.pk
    
  140.         )
    
  141.         self.assertEqual(res.status_code, 200)
    
  142.         self.assertEqual(res.context["object"], self.author1)
    
  143.         self.assertEqual(res.context["author"], self.author1)
    
  144.         self.assertTemplateUsed(res, "generic_views/author_view.html")
    
  145. 
    
  146.     def test_template_name_field(self):
    
  147.         res = self.client.get("/detail/page/%s/field/" % self.page1.pk)
    
  148.         self.assertEqual(res.status_code, 200)
    
  149.         self.assertEqual(res.context["object"], self.page1)
    
  150.         self.assertEqual(res.context["page"], self.page1)
    
  151.         self.assertTemplateUsed(res, "generic_views/page_template.html")
    
  152. 
    
  153.     def test_context_object_name(self):
    
  154.         res = self.client.get(
    
  155.             "/detail/author/%s/context_object_name/" % self.author1.pk
    
  156.         )
    
  157.         self.assertEqual(res.status_code, 200)
    
  158.         self.assertEqual(res.context["object"], self.author1)
    
  159.         self.assertEqual(res.context["thingy"], self.author1)
    
  160.         self.assertNotIn("author", res.context)
    
  161.         self.assertTemplateUsed(res, "generic_views/author_detail.html")
    
  162. 
    
  163.     def test_duplicated_context_object_name(self):
    
  164.         res = self.client.get(
    
  165.             "/detail/author/%s/dupe_context_object_name/" % self.author1.pk
    
  166.         )
    
  167.         self.assertEqual(res.status_code, 200)
    
  168.         self.assertEqual(res.context["object"], self.author1)
    
  169.         self.assertNotIn("author", res.context)
    
  170.         self.assertTemplateUsed(res, "generic_views/author_detail.html")
    
  171. 
    
  172.     def test_custom_detail(self):
    
  173.         """
    
  174.         AuthorCustomDetail overrides get() and ensures that
    
  175.         SingleObjectMixin.get_context_object_name() always uses the obj
    
  176.         parameter instead of self.object.
    
  177.         """
    
  178.         res = self.client.get("/detail/author/%s/custom_detail/" % self.author1.pk)
    
  179.         self.assertEqual(res.status_code, 200)
    
  180.         self.assertEqual(res.context["custom_author"], self.author1)
    
  181.         self.assertNotIn("author", res.context)
    
  182.         self.assertNotIn("object", res.context)
    
  183.         self.assertTemplateUsed(res, "generic_views/author_detail.html")
    
  184. 
    
  185.     def test_deferred_queryset_template_name(self):
    
  186.         class FormContext(SingleObjectTemplateResponseMixin):
    
  187.             request = RequestFactory().get("/")
    
  188.             model = Author
    
  189.             object = Author.objects.defer("name").get(pk=self.author1.pk)
    
  190. 
    
  191.         self.assertEqual(
    
  192.             FormContext().get_template_names()[0], "generic_views/author_detail.html"
    
  193.         )
    
  194. 
    
  195.     def test_deferred_queryset_context_object_name(self):
    
  196.         class FormContext(ModelFormMixin):
    
  197.             request = RequestFactory().get("/")
    
  198.             model = Author
    
  199.             object = Author.objects.defer("name").get(pk=self.author1.pk)
    
  200.             fields = ("name",)
    
  201. 
    
  202.         form_context_data = FormContext().get_context_data()
    
  203.         self.assertEqual(form_context_data["object"], self.author1)
    
  204.         self.assertEqual(form_context_data["author"], self.author1)
    
  205. 
    
  206.     def test_invalid_url(self):
    
  207.         with self.assertRaises(AttributeError):
    
  208.             self.client.get("/detail/author/invalid/url/")
    
  209. 
    
  210.     def test_invalid_queryset(self):
    
  211.         msg = (
    
  212.             "AuthorDetail is missing a QuerySet. Define AuthorDetail.model, "
    
  213.             "AuthorDetail.queryset, or override AuthorDetail.get_queryset()."
    
  214.         )
    
  215.         with self.assertRaisesMessage(ImproperlyConfigured, msg):
    
  216.             self.client.get("/detail/author/invalid/qs/")
    
  217. 
    
  218.     def test_non_model_object_with_meta(self):
    
  219.         res = self.client.get("/detail/nonmodel/1/")
    
  220.         self.assertEqual(res.status_code, 200)
    
  221.         self.assertEqual(res.context["object"].id, "non_model_1")