def setUpTestData(cls): call_command('update_tpm_permissions', verbosity=0) call_command('update_notifications') # clearing groups cache GroupWrapper.invalidate_instances() cls.pme_user = PMEUserFactory() cls.unicef_user = UserFactory() cls.tpm_partner = TPMPartnerFactory() cls.tpm_user = TPMUserFactory(tpm_partner=cls.tpm_partner)
def test_list(self): partner = TPMPartnerFactory() attachments_num = partner.attachments.count() AttachmentFactory(content_object=partner) response = self.forced_auth_req('get', reverse('tpm:partner-attachments-list', args=[partner.id]), user=self.pme_user) self.assertEqual(response.status_code, status.HTTP_200_OK) self.assertEqual(len(response.data['results']), attachments_num + 1)
def _test_partner(self, expected_status=status.HTTP_201_CREATED, **kwargs): partner = TPMPartnerFactory(**kwargs) response = self.forced_auth_req( 'post', reverse('tpm:visits-list'), user=self.pme_user, data={'tpm_partner': partner.id}, ) self.assertEqual(response.status_code, expected_status) if expected_status == status.HTTP_400_BAD_REQUEST: self.assertIn('tpm_partner', response.data)
def setUpTestData(cls): super().setUpTestData() cls.visit = TPMVisitFactory( status='draft', tpm_partner=cls.tpm_user.tpmpartners_tpmpartnerstaffmember. tpm_partner, tpm_partner_focal_points=[ cls.tpm_user.tpmpartners_tpmpartnerstaffmember ], tpm_activities__count=1) cls.activity = cls.visit.tpm_activities.first() partner = TPMPartnerFactory() cls.attachment = AttachmentFactory(content_object=partner)
def test_add(self): partner = TPMPartnerFactory() response = self.forced_auth_req( 'post', reverse('tpm:partner-attachments-list', args=[partner.id]), user=self.pme_user, request_format='multipart', data={ 'file_type': AttachmentFileTypeFactory(code='tpm_partner').id, 'file': SimpleUploadedFile('hello_world.txt', u'hello world!'.encode('utf-8')), }) self.assertEqual(response.status_code, status.HTTP_201_CREATED)
def test_activation(self): partner = TPMPartnerFactory(countries=[]) # partner is deactivated yet, so wouldn't appear in list self._test_list_view(self.pme_user, [self.tpm_partner, self.second_tpm_partner]) activate_response = self.forced_auth_req('post', reverse( 'tpm:partners-activate', args=(partner.id, )), user=self.pme_user) self.assertEqual(activate_response.status_code, status.HTTP_200_OK) self._test_list_view( self.pme_user, [self.tpm_partner, self.second_tpm_partner, partner])
def test_not_editable_by_tpm(self): partner = TPMPartnerFactory() response = self.forced_auth_req( 'post', reverse('tpm:partner-attachments-list', args=[partner.id]), user=TPMUserFactory(tpm_partner=partner), request_format='multipart', data={ 'file_type': AttachmentFileTypeFactory(code='tpm_partner').id, 'file': SimpleUploadedFile('hello_world.txt', 'hello world!'.encode('utf-8')), }) self.assertEqual(response.status_code, status.HTTP_403_FORBIDDEN)
def setUpTestData(cls): super(TestTPMPartnerViewSet, cls).setUpTestData() cls.second_tpm_partner = TPMPartnerFactory()
def setUpTestData(cls): cls.firm = TPMPartnerFactory() call_command('update_notifications')
def setUpTestData(cls): super().setUpTestData() cls.second_tpm_partner = TPMPartnerFactory()