def test_engagement(self): purchase_order = PurchaseOrderFactory.build(order_number='two') instance = EngagementFactory.build(agreement=purchase_order) self.assertIn(' two,', str(instance)) purchase_order = PurchaseOrderFactory.build(order_number='tv\xe5') instance = EngagementFactory.build(agreement=purchase_order) self.assertIn(' tv\xe5,', str(instance))
def setUpTestData(cls): cls.code_1 = "test_code_1" cls.file_type_1 = AttachmentFileTypeFactory(code=cls.code_1) cls.code_2 = "test_code_2" cls.file_type_2 = AttachmentFileTypeFactory(code=cls.code_2) cls.unicef_staff = UserFactory(is_staff=True) cls.user = UserFactory() cls.url = reverse("attachments:list") cls.attachment_1 = AttachmentFactory( file_type=cls.file_type_1, code=cls.code_1, file="sample1.pdf", content_object=cls.file_type_1, uploaded_by=cls.unicef_staff ) cls.attachment_2 = AttachmentFactory( file_type=cls.file_type_2, code=cls.code_2, file="sample2.pdf", content_object=cls.file_type_2, uploaded_by=cls.user ) cls.partner = PartnerFactory( partner_type=PartnerType.UN_AGENCY, vendor_number="V123", ) cls.agreement = AgreementFactory(partner=cls.partner) cls.assessment = AssessmentFactory(partner=cls.partner) cls.amendment = AgreementAmendmentFactory(agreement=cls.agreement) cls.intervention = InterventionFactory(agreement=cls.agreement) cls.result_link = InterventionResultLinkFactory( intervention=cls.intervention ) cls.intervention_amendment = InterventionAmendmentFactory( intervention=cls.intervention ) cls.intervention_attachment = InterventionAttachmentFactory( intervention=cls.intervention ) cls.tpm_partner = SimpleTPMPartnerFactory(vendor_number="V432") cls.tpm_visit = TPMVisitFactory(tpm_partner=cls.tpm_partner) cls.tpm_activity = TPMActivityFactory( partner=cls.partner, intervention=cls.intervention, tpm_visit=cls.tpm_visit ) cls.engagement = EngagementFactory(partner=cls.partner) cls.default_partner_response = [{ "partner": "", "partner_type": "", "vendor_number": "", "pd_ssfa_number": "", "agreement_reference_number": "", "source": "", }] * 2
def test_get(self): engagement = EngagementFactory() response = self.forced_auth_req( "get", "/api/audit/engagements/partners/", user=self.unicef_user, ) self.assertEqual(response.status_code, status.HTTP_200_OK) self.assertEqual(len(response.data), 1) self.assertEqual(response.data[0]["id"], engagement.partner.pk)
def test_signal(self): auditor_firm = AuditPartnerFactory() staff_member = auditor_firm.staff_members.first() staff_member.user.profile.countries_available = [] engagement = EngagementFactory(staff_members=[], agreement__auditor_firm=auditor_firm) engagement.staff_members.add(staff_member) self.assertSequenceEqual( staff_member.user.profile.countries_available.all(), [Country.objects.get(schema_name=connection.schema_name)]) self.assertEqual(len(mail.outbox), 1) mail.outbox = [] engagement = EngagementFactory(staff_members=[], agreement__auditor_firm=auditor_firm) engagement.staff_members.add(staff_member) self.assertEqual(len(mail.outbox), 1) mail.outbox = []
def test_get_object_url(self): """Check that engagement pk is part of url""" engagement = EngagementFactory() url = engagement.get_object_url() self.assertIn(str(engagement.pk), url)
def setUpTestData(cls): cls.user = UserFactory() cls.engagement = EngagementFactory() cls.attachment = AttachmentFactory(content_object=cls.engagement)