def test_otg_creation(self): ct = ContentType.objects.get_for_model(OutgoingTransmittal) cat_template = CategoryTemplateFactory( metadata_model=ct) dest_cat = CategoryFactory(category_template=cat_template) ctr1 = EntityFactory(type='contractor') ctr2 = EntityFactory(type='contractor') # We have to link third parties to categories self.category.third_parties.add(ctr1, ctr2) dest_cat.third_parties.add(ctr1, ctr2) recipients_ids = [ctr1.pk, ctr2.pk] document_ids = [doc.pk for doc in self.docs] do_create_transmittal.delay( self.user, self.category.id, dest_cat.id, document_ids, self.contract.number, recipients_ids ) # We have 2 docs and 2 recipients, so we expect 2 outgoing trs # and 2 notifications self.assertEqual(OutgoingTransmittal.objects.count(), 2) self.assertEqual(Notification.objects.count(), 2) # Check each doc revision links to both transmittals for rev in self.revisions: self.assertEqual(rev.transmittals.count(), 2)
def test_otg_creation(self): ct = ContentType.objects.get_for_model(OutgoingTransmittal) cat_template = CategoryTemplateFactory( metadata_model=ct) dest_cat = CategoryFactory(category_template=cat_template) ctr1 = EntityFactory(type='contractor') ctr2 = EntityFactory(type='contractor') # We have to link third parties to categories self.category.third_parties.add(ctr1, ctr2) dest_cat.third_parties.add(ctr1, ctr2) recipients_ids = [ctr1.pk, ctr2.pk] document_ids = [doc.pk for doc in self.docs] do_create_transmittal.delay( self.user.id, self.category.id, dest_cat.id, document_ids, self.contract.number, recipients_ids ) # We have 2 docs and 2 recipients, so we expect 2 outgoing trs # and 2 notifications self.assertEqual(OutgoingTransmittal.objects.count(), 2) self.assertEqual(Notification.objects.count(), 2) # Check each doc revision links to both transmittals for rev in self.revisions: self.assertEqual(rev.transmittals.count(), 2)
def start_job(self, contenttype, document_ids): from_category_id = self.category.id to_category_id = self.request.POST.get('destination_category') recipients_ids = self.request.POST.getlist('recipients') contract_number = self.request.POST.get('contract_number') job = do_create_transmittal.delay(self.request.user.id, from_category_id, to_category_id, document_ids, contract_number, recipients_ids) return job
def start_job(self, contenttype, document_ids): from_category_id = self.category.id to_category_id = self.request.POST.get('destination_category') recipients_ids = self.request.POST.getlist('recipients') contract_number = self.request.POST.get('contract_number') job = do_create_transmittal.delay( self.request.user.id, from_category_id, to_category_id, document_ids, contract_number, recipients_ids) return job