diff --git a/wagtail/documents/models.py b/wagtail/documents/models.py index cf611eb56..1be6acd63 100644 --- a/wagtail/documents/models.py +++ b/wagtail/documents/models.py @@ -14,7 +14,7 @@ from wagtail.admin.models import get_object_usage from wagtail.core.models import CollectionMember from wagtail.search import index from wagtail.search.queryset import SearchableQuerySetMixin -from wagtail.utils.deprecation import RemovedInWagtail29Warning +from wagtail.utils.deprecation import RemovedInWagtail210Warning class DocumentQuerySet(SearchableQuerySetMixin, models.QuerySet): @@ -24,7 +24,7 @@ class DocumentQuerySet(SearchableQuerySetMixin, models.QuerySet): def get_document_model(): warnings.warn("wagtail.documents.models.get_document_model " "has been moved to wagtail.documents.get_document_model", - RemovedInWagtail29Warning) + RemovedInWagtail210Warning) from wagtail.documents import get_document_model return get_document_model() diff --git a/wagtail/documents/tests/test_models.py b/wagtail/documents/tests/test_models.py index 2f5b27aee..c4440f067 100644 --- a/wagtail/documents/tests/test_models.py +++ b/wagtail/documents/tests/test_models.py @@ -14,7 +14,7 @@ from wagtail.documents import get_document_model, get_document_model_string, mod from wagtail.images.tests.utils import get_test_image_file from wagtail.tests.testapp.models import CustomDocument from wagtail.tests.utils import WagtailTestUtils -from wagtail.utils.deprecation import RemovedInWagtail29Warning +from wagtail.utils.deprecation import RemovedInWagtail210Warning class TestDocumentQuerySet(TestCase): @@ -212,4 +212,4 @@ class TestGetDocumentModel(WagtailTestUtils, TestCase): self.assertIs(Document, get_document_model()) self.assertEqual(len(ws), 1) - self.assertIs(ws[0].category, RemovedInWagtail29Warning) + self.assertIs(ws[0].category, RemovedInWagtail210Warning)