def test_no_rd_match(self, mock):
"""If there's no RECAPDocument to match on, do we fail gracefully?"""
RECAPDocument.objects.all().delete()
with self.assertRaises(RECAPDocument.DoesNotExist):
process_recap_attachment(self.pq.pk)
self.pq.refresh_from_db()
# This doesn't do the celery retries, unfortunately. If we get that
# working, the correct status is self.pq.PROCESSING_FAILED.
self.assertEqual(self.pq.status, self.pq.QUEUED_FOR_RETRY)
def test_attachments_get_created(self, mock):
"""Do attachments get created if we have a RECAPDocument to match on?"""
process_recap_attachment(self.pq.pk)
num_attachments_to_create = 3
self.assertEqual(
RECAPDocument.objects.filter(
document_type=RECAPDocument.ATTACHMENT).count(),
num_attachments_to_create,
)
self.pq.refresh_from_db()
self.assertEqual(self.pq.status, ProcessingQueue.PROCESSING_SUCCESSFUL)
def test_debug_does_not_create_recap_documents(self, mock):
"""If debug is passed, do we avoid creating recap documents?"""
d = Docket.objects.create(source=0, court_id='scotus',
pacer_case_id='asdf')
de = DocketEntry.objects.create(docket=d, entry_number=1)
rd = RECAPDocument.objects.create(
docket_entry=de,
document_number='1',
pacer_doc_id='04505578698',
document_type=RECAPDocument.PACER_DOCUMENT,
)
pq = ProcessingQueue.objects.create(
court_id='scotus',
uploader=self.user,
upload_type=ProcessingQueue.ATTACHMENT_PAGE,
filepath_local=self.att,
debug=True,
)
process_recap_attachment(pq.pk)
self.assertEqual(Docket.objects.count(), 1)
self.assertEqual(DocketEntry.objects.count(), 1)
self.assertEqual(RECAPDocument.objects.count(), 1)
mock.assert_not_called()
