Move CustomSoftDeleteQuerySet to tests.models

The `CustomSoftDeleteQuerySet` class was the only code left in the `test.managers` module, while it is not in fact a `Manager`.
jazzband · Apr 16, 2024 · 14ac0b4 · 14ac0b4
1 parent 600d59c
commit 14ac0b4
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 7 deletions.
diff --git a/tests/managers.py b/tests/managers.py
diff --git a/tests/models.py b/tests/models.py
@@ -14,6 +14,7 @@
     JoinManager,
     QueryManager,
     SoftDeletableManager,
+    SoftDeletableQuerySet,
 )
 from model_utils.models import (
     SoftDeletableModel,
@@ -24,7 +25,6 @@
 )
 from model_utils.tracker import FieldTracker, ModelTracker
 from tests.fields import MutableField
-from tests.managers import CustomSoftDeleteQuerySet
 
 
 class InheritanceManagerTestRelated(models.Model):
@@ -352,6 +352,11 @@ class SoftDeletable(SoftDeletableModel):
     all_objects: ClassVar[Manager[SoftDeletable]] = models.Manager()
 
 
+class CustomSoftDeleteQuerySet(SoftDeletableQuerySet):
+    def only_read(self):
+        return self.filter(is_read=True)
+
+
 class CustomSoftDelete(SoftDeletableModel):
     is_read = models.BooleanField(default=False)