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
