More pytest migrations (#393)

* More pytest migrations

Author: asherf

django_prometheus/tests/end2end/testapp/test_migrations.py

@@ -1,6 +1,6 @@
 from unittest.mock import MagicMock
 
-from django.test import SimpleTestCase
+import pytest
 
 from django_prometheus.migrations import ExportMigrationsForDatabase
 from django_prometheus.testutils import assert_metric_equal
@@ -15,7 +15,8 @@ def M(metric_name):
     return "django_migrations_%s" % metric_name
 
 
-class TestMigrations(SimpleTestCase):
+@pytest.mark.django_db()
+class TestMigrations:
     """Test migration counters."""
 
     def test_counters(self):

django_prometheus/tests/end2end/testapp/test_models.py

@@ -1,4 +1,4 @@
-from django.test import TestCase
+import pytest
 from testapp.models import Dog, Lawn
 
 from django_prometheus.testutils import assert_metric_diff, save_registry
@@ -13,7 +13,8 @@ def M(metric_name):
     return "django_model_%s" % metric_name
 
 
-class TestModelMetrics(TestCase):
+@pytest.mark.django_db()
+class TestModelMetrics:
     """Test django_prometheus.models."""
 
     def test_counters(self):