test: verify correct typing of mixins

tests/openedx_learning/apps/authoring/components/test_models.py

@@ -2,6 +2,7 @@
 Tests related to the Component models
 """
 from datetime import datetime, timezone
+from typing import TYPE_CHECKING, assert_type
 
 from freezegun import freeze_time
 
@@ -10,7 +11,7 @@
     get_component,
     get_or_create_component_type,
 )
-from openedx_learning.apps.authoring.components.models import ComponentType
+from openedx_learning.apps.authoring.components.models import Component, ComponentType, ComponentVersion
 from openedx_learning.apps.authoring.publishing.api import (
     LearningPackage,
     create_learning_package,
@@ -19,6 +20,15 @@
 )
 from openedx_learning.lib.test_utils import TestCase
 
+if TYPE_CHECKING:
+    # Test that our mixins on Component.objects and PublishableEntityVersionMixin etc. haven't broken manager typing
+    assert_type(Component.objects.create(), Component)
+    assert_type(Component.objects.get(), Component)
+    assert_type(Component.with_publishing_relations.create(), Component)
+    assert_type(Component.with_publishing_relations.get(), Component)
+    assert_type(ComponentVersion.objects.create(), ComponentVersion)
+    assert_type(ComponentVersion.objects.get(), ComponentVersion)
+
 
 class TestModelVersioningQueries(TestCase):
     """

tests/openedx_learning/apps/authoring/publishing/test_models.py

@@ -0,0 +1,21 @@
+"""
+Tests related to the Publishing model mixins
+"""
+from typing import TYPE_CHECKING, assert_type
+
+from openedx_learning.apps.authoring.publishing.models import PublishableEntityMixin, PublishableEntityVersionMixin
+from openedx_learning.lib.managers import WithRelationsManager
+
+if TYPE_CHECKING:
+    # Test that our mixins provide the right typing for 'objects'
+    class FooEntity(PublishableEntityMixin):
+        pass
+
+    assert_type(FooEntity.objects.create(), FooEntity)
+    assert_type(FooEntity.objects, WithRelationsManager[FooEntity])
+
+    class FooEntityVersion(PublishableEntityVersionMixin):
+        pass
+
+    assert_type(FooEntityVersion.objects.create(), FooEntityVersion)
+    assert_type(FooEntityVersion.objects, WithRelationsManager[FooEntityVersion])