Documents: Add default filtering of stubs

Add filter(is_stub) to the default Document model manager.

Now only the Passthrough manager can access document stubs.

Remove the explicit filtering of stubs from code that obtains
the queryset from the default document manager.

Signed-off-by: Roberto Rosario <roberto.rosario.gonzalez@gmail.com>
This commit is contained in:
Roberto Rosario
2018-12-05 03:05:39 -04:00
parent f6a675c9db
commit 55cd928069
5 changed files with 14 additions and 18 deletions

View File

@@ -242,34 +242,34 @@ class DocumentVersionTestCase(GenericDocumentTestCase):
@override_settings(OCR_AUTO_OCR=False)
class DocumentManagerTestCase(BaseTestCase):
class DocumentPassthroughManagerTestCase(BaseTestCase):
def setUp(self):
super(DocumentManagerTestCase, self).setUp()
super(DocumentPassthroughManagerTestCase, self).setUp()
self.document_type = DocumentType.objects.create(
label=TEST_DOCUMENT_TYPE_LABEL
)
def tearDown(self):
self.document_type.delete()
super(DocumentManagerTestCase, self).tearDown()
super(DocumentPassthroughManagerTestCase, self).tearDown()
def test_document_stubs_deletion(self):
document_stub = Document.objects.create(
document_type=self.document_type
)
Document.objects.delete_stubs()
Document.passthrough.delete_stubs()
self.assertEqual(Document.objects.count(), 1)
self.assertEqual(Document.passthrough.count(), 1)
document_stub.date_added = document_stub.date_added - timedelta(
seconds=STUB_EXPIRATION_INTERVAL + 1
)
document_stub.save()
Document.objects.delete_stubs()
Document.passthrough.delete_stubs()
self.assertEqual(Document.objects.count(), 0)
self.assertEqual(Document.passthrough.count(), 0)
class DuplicatedDocumentsTestCase(GenericDocumentTestCase):