Added test for lookup

safedelete/tests/test_lookups.py

@@ -0,0 +1,51 @@
+from django.db import models
+from django.test import TestCase
+
+from safedelete import SOFT_DELETE
+from safedelete.models import SafeDeleteModel
+from safedelete.tests.models import Article, Author
+
+
+class PressLookup(SafeDeleteModel):
+    name = models.CharField(max_length=200)
+    article = models.ForeignKey(Article, on_delete=models.CASCADE)
+
+
+class InLookupTest(TestCase):
+    """
+    Test cases for filtering safe delete models with the `in` ORM lookup.
+    """
+
+    def setUp(self):
+        self.authors = (
+            Author.objects.create(),
+            Author.objects.create(),
+            Author.objects.create(),
+        )
+
+        self.articles = (
+            Article.objects.create(author=self.authors[1]),
+            Article.objects.create(author=self.authors[1]),
+            Article.objects.create(author=self.authors[2]),
+        )
+
+        self.press = (
+            PressLookup.objects.create(name='press 0', article=self.articles[0]),
+            PressLookup.objects.create(name='press 1', article=self.articles[1]),
+            PressLookup.objects.create(name='press 2', article=self.articles[2]),
+        )
+
+    def test_filter_with_in(self):
+        """
+        Issue https://github.com/makinacorpus/django-safedelete/issues/101
+
+        Using a SafeDeleteQueryset as the argument to a filters `in` lookup only works if the queryset is
+          evaluated. i.e. Press.objects.filter(article__in=list(articles))
+        """
+        self.articles[0].delete(force_policy=SOFT_DELETE)
+        self.articles[1].delete(force_policy=SOFT_DELETE)
+
+        articles = Article.objects.filter()
+        press = PressLookup.objects.filter(article__in=articles)
+
+        self.assertEqual(press.count(), 1)