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()