Use pytest for testing & modernize assertions

docs/contributing.rst

@@ -13,7 +13,7 @@ We require features to be backed by a unit test.
 This way, we can test *django-polymorphic* against new Django versions.
 To run the included test suite, execute::
 
-    ./runtests.py
+    py.test
 
 To test support for multiple Python and Django versions, run tox from the repository root::
 

manage.py

@@ -0,0 +1,9 @@
+# This helps pytest-django locate the project.
+import os
+import sys
+
+if __name__ == "__main__":
+    os.environ.setdefault("DJANGO_SETTINGS_MODULE", "polymorphic_test_settings")
+    from django.core.management import execute_from_command_line
+
+    execute_from_command_line(sys.argv)

polymorphic/tests/__init__.py

@@ -0,0 +1,6 @@
+import django
+
+if django.VERSION < (4, 2):  # TODO: remove when dropping support for Django < 4.2
+    from django.test.testcases import TransactionTestCase
+
+    TransactionTestCase.assertQuerySetEqual = TransactionTestCase.assertQuerysetEqual

polymorphic/tests/admintestcase.py

@@ -94,7 +94,7 @@ def admin_get_add(self, model, qs=""):
         admin_instance = self.get_admin_instance(model)
         request = self.create_admin_request("get", self.get_add_url(model) + qs)
         response = admin_instance.add_view(request)
-        self.assertEqual(response.status_code, 200)
+        assert response.status_code == 200
         return response
 
     def admin_post_add(self, model, formdata, qs=""):
@@ -114,7 +114,7 @@ def admin_get_changelist(self, model):
         admin_instance = self.get_admin_instance(model)
         request = self.create_admin_request("get", self.get_changelist_url(model))
         response = admin_instance.changelist_view(request)
-        self.assertEqual(response.status_code, 200)
+        assert response.status_code == 200
         return response
 
     def admin_get_change(self, model, object_id, query=None, **extra):
@@ -126,7 +126,7 @@ def admin_get_change(self, model, object_id, query=None, **extra):
             "get", self.get_change_url(model, object_id), data=query, **extra
         )
         response = admin_instance.change_view(request, str(object_id))
-        self.assertEqual(response.status_code, 200)
+        assert response.status_code == 200
         return response
 
     def admin_post_change(self, model, object_id, formdata, **extra):
@@ -150,7 +150,7 @@ def admin_get_history(self, model, object_id, query=None, **extra):
             "get", self.get_history_url(model, object_id), data=query, **extra
         )
         response = admin_instance.history_view(request, str(object_id))
-        self.assertEqual(response.status_code, 200)
+        assert response.status_code == 200
         return response
 
     def admin_get_delete(self, model, object_id, query=None, **extra):
@@ -162,7 +162,7 @@ def admin_get_delete(self, model, object_id, query=None, **extra):
             "get", self.get_delete_url(model, object_id), data=query, **extra
         )
         response = admin_instance.delete_view(request, str(object_id))
-        self.assertEqual(response.status_code, 200)
+        assert response.status_code == 200
         return response
 
     def admin_post_delete(self, model, object_id, **extra):
@@ -175,7 +175,7 @@ def admin_post_delete(self, model, object_id, **extra):
         admin_instance = self.get_admin_instance(model)
         request = self.create_admin_request("post", self.get_delete_url(model, object_id), **extra)
         response = admin_instance.delete_view(request, str(object_id))
-        self.assertEqual(response.status_code, 302, f"Form errors in calling {request.path}")
+        assert response.status_code == 302, f"Form errors in calling {request.path}"
         return response
 
     def create_admin_request(self, method, url, data=None, **extra):
@@ -209,7 +209,7 @@ def assertFormSuccess(self, request_url, response):
         """
         Assert that the response was a redirect, not a form error.
         """
-        self.assertIn(response.status_code, [200, 302])
+        assert response.status_code in [200, 302]
         if response.status_code != 302:
             context_data = response.context_data
             if "errors" in context_data:
@@ -219,12 +219,9 @@ def assertFormSuccess(self, request_url, response):
             else:
                 raise KeyError("Unknown field for errors in the TemplateResponse!")
 
-            self.assertEqual(
-                response.status_code,
-                302,
-                "Form errors in calling {}:\n{}".format(request_url, errors.as_text()),
+            assert response.status_code == 302, "Form errors in calling {}:\n{}".format(
+                request_url, errors.as_text()
             )
-        self.assertTrue(
-            "/login/?next=" not in response["Location"],
-            f"Received login response for {request_url}",
-        )
+        assert (
+            "/login/?next=" not in response["Location"]
+        ), f"Received login response for {request_url}"

polymorphic/tests/test_admin.py

@@ -1,3 +1,4 @@
+import pytest
 from django.contrib import admin
 from django.contrib.contenttypes.models import ContentType
 from django.utils.html import escape
@@ -53,9 +54,9 @@ class Model2ChildAdmin(PolymorphicChildModelAdmin):
         )
 
         d_obj = Model2A.objects.all()[0]
-        self.assertEqual(d_obj.__class__, Model2D)
-        self.assertEqual(d_obj.field1, "A")
-        self.assertEqual(d_obj.field2, "B")
+        assert d_obj.__class__ == Model2D
+        assert d_obj.field1 == "A"
+        assert d_obj.field2 == "B"
 
         # -- list page
         self.admin_get_changelist(Model2A)  # asserts 200
@@ -70,18 +71,18 @@ class Model2ChildAdmin(PolymorphicChildModelAdmin):
         )
 
         d_obj.refresh_from_db()
-        self.assertEqual(d_obj.field1, "A2")
-        self.assertEqual(d_obj.field2, "B2")
-        self.assertEqual(d_obj.field3, "C2")
-        self.assertEqual(d_obj.field4, "D2")
+        assert d_obj.field1 == "A2"
+        assert d_obj.field2 == "B2"
+        assert d_obj.field3 == "C2"
+        assert d_obj.field4 == "D2"
 
         # -- history
         self.admin_get_history(Model2A, d_obj.pk)
 
         # -- delete
         self.admin_get_delete(Model2A, d_obj.pk)
         self.admin_post_delete(Model2A, d_obj.pk)
-        self.assertRaises(Model2A.DoesNotExist, lambda: d_obj.refresh_from_db())
+        pytest.raises(Model2A.DoesNotExist, (lambda: d_obj.refresh_from_db()))
 
     def test_admin_inlines(self):
         """
@@ -103,7 +104,7 @@ class InlineParentAdmin(PolymorphicInlineSupportMixin, admin.ModelAdmin):
             inlines = (Inline,)
 
         parent = InlineParent.objects.create(title="FOO")
-        self.assertEqual(parent.inline_children.count(), 0)
+        assert parent.inline_children.count() == 0
 
         # -- get edit page
         response = self.admin_get_change(InlineParent, parent.pk)
@@ -133,9 +134,9 @@ class InlineParentAdmin(PolymorphicInlineSupportMixin, admin.ModelAdmin):
         )
 
         parent.refresh_from_db()
-        self.assertEqual(parent.title, "FOO2")
-        self.assertEqual(parent.inline_children.count(), 1)
+        assert parent.title == "FOO2"
+        assert parent.inline_children.count() == 1
         child = parent.inline_children.all()[0]
-        self.assertEqual(child.__class__, InlineModelB)
-        self.assertEqual(child.field1, "A2")
-        self.assertEqual(child.field2, "B2")
+        assert child.__class__ == InlineModelB
+        assert child.field1 == "A2"
+        assert child.field2 == "B2"

polymorphic/tests/test_contrib.py

@@ -1,4 +1,4 @@
-from unittest import TestCase
+from django.test import TestCase
 
 from polymorphic.contrib.guardian import get_polymorphic_base_content_type
 from polymorphic.tests.models import Model2D, PlainC
@@ -13,15 +13,15 @@ def test_contrib_guardian(self):
         # Regular Django inheritance should return the child model content type.
         obj = PlainC()
         ctype = get_polymorphic_base_content_type(obj)
-        self.assertEqual(ctype.name, "plain c")
+        assert ctype.name == "plain c"
 
         ctype = get_polymorphic_base_content_type(PlainC)
-        self.assertEqual(ctype.name, "plain c")
+        assert ctype.name == "plain c"
 
         # Polymorphic inheritance should return the parent model content type.
         obj = Model2D()
         ctype = get_polymorphic_base_content_type(obj)
-        self.assertEqual(ctype.name, "model2a")
+        assert ctype.name == "model2a"
 
         ctype = get_polymorphic_base_content_type(Model2D)
-        self.assertEqual(ctype.name, "model2a")
+        assert ctype.name == "model2a"

polymorphic/tests/test_multidb.py

@@ -26,13 +26,13 @@ def test_save_to_non_default_database(self):
         Model2B.objects.create(field1="B1", field2="B2")
         Model2D(field1="D1", field2="D2", field3="D3", field4="D4").save()
 
-        self.assertQuerysetEqual(
+        self.assertQuerySetEqual(
             Model2A.objects.order_by("id"),
             [Model2B, Model2D],
             transform=lambda o: o.__class__,
         )
 
-        self.assertQuerysetEqual(
+        self.assertQuerySetEqual(
             Model2A.objects.db_manager("secondary").order_by("id"),
             [Model2A, Model2C],
             transform=lambda o: o.__class__,
@@ -44,26 +44,26 @@ def test_instance_of_filter_on_non_default_database(self):
         ModelY.objects.db_manager("secondary").create(field_b="Y", field_y="Y")
 
         objects = Base.objects.db_manager("secondary").filter(instance_of=Base)
-        self.assertQuerysetEqual(
+        self.assertQuerySetEqual(
             objects,
             [Base, ModelX, ModelY],
             transform=lambda o: o.__class__,
             ordered=False,
         )
 
-        self.assertQuerysetEqual(
+        self.assertQuerySetEqual(
             Base.objects.db_manager("secondary").filter(instance_of=ModelX),
             [ModelX],
             transform=lambda o: o.__class__,
         )
 
-        self.assertQuerysetEqual(
+        self.assertQuerySetEqual(
             Base.objects.db_manager("secondary").filter(instance_of=ModelY),
             [ModelY],
             transform=lambda o: o.__class__,
         )
 
-        self.assertQuerysetEqual(
+        self.assertQuerySetEqual(
             Base.objects.db_manager("secondary").filter(
                 Q(instance_of=ModelX) | Q(instance_of=ModelY)
             ),
@@ -78,7 +78,7 @@ def func():
             entry = BlogEntry.objects.db_manager("secondary").create(blog=blog, text="Text")
             ContentType.objects.clear_cache()
             entry = BlogEntry.objects.db_manager("secondary").get(pk=entry.id)
-            self.assertEqual(blog, entry.blog)
+            assert blog == entry.blog
 
         # Ensure no queries are made using the default database.
         self.assertNumQueries(0, func)
@@ -89,7 +89,7 @@ def func():
             entry = BlogEntry.objects.db_manager("secondary").create(blog=blog, text="Text")
             ContentType.objects.clear_cache()
             blog = BlogA.objects.db_manager("secondary").get(pk=blog.id)
-            self.assertEqual(entry, blog.blogentry_set.using("secondary").get())
+            assert entry == blog.blogentry_set.using("secondary").get()
 
         # Ensure no queries are made using the default database.
         self.assertNumQueries(0, func)
@@ -102,7 +102,7 @@ def func():
             )
             ContentType.objects.clear_cache()
             m2a = Model2A.objects.db_manager("secondary").get(pk=m2a.id)
-            self.assertEqual(one2one, m2a.one2onerelatingmodel)
+            assert one2one == m2a.one2onerelatingmodel
 
         # Ensure no queries are made using the default database.
         self.assertNumQueries(0, func)
@@ -114,7 +114,7 @@ def func():
             rm.many2many.add(m2a)
             ContentType.objects.clear_cache()
             m2a = Model2A.objects.db_manager("secondary").get(pk=m2a.id)
-            self.assertEqual(rm, m2a.relatingmodel_set.using("secondary").get())
+            assert rm == m2a.relatingmodel_set.using("secondary").get()
 
         # Ensure no queries are made using the default database.
         self.assertNumQueries(0, func)