def test_settings_view(self) -> None:
self.set_tenant(0)
user = self.users[0]
factory = TenantsAPIRequestFactory(force_authenticate=user)
request = factory.post('', data=dict(name='testing settings'))
response = CampaignsViewSet.as_view({'post': 'create'})(request)
self.assertEqual(response.status_code, status.HTTP_201_CREATED, str(response.data))
campaign_data = response.data
campaign = Campaign.objects.get(id=campaign_data['id'])
self.assertTrue(campaign.settings)
request = factory.get('')
response = CampaignSettingsViewSet.as_view({'get': 'retrieve_single'})(request, campaign=campaign_data['id'])
self.assertEqual(response.status_code, status.HTTP_200_OK, str(response.data))
response.render() # render just in case
settings_data = response.data
self.assertEqual(campaign.settings.step_max_number, settings_data['step_max_number'])
email_send_delay = datetime.timedelta(**dict(zip(
['hours', 'minutes', 'seconds'],
map(int, settings_data['email_send_delay'].split(':', 2))
)))
self.assertEqual(campaign.settings.email_send_delay, email_send_delay)
def test_settings_view(self) -> None:
self.set_tenant(0)
factory = TenantsAPIRequestFactory(force_authenticate=self.user)
request = factory.get('')
response = NoticeSettingsViewSet.as_view({'get': 'list'})(request)
contact_data = response.data
self.assertEqual(response.status_code, status.HTTP_200_OK,
str(contact_data))
self.assertEqual(2, len(contact_data))
def test_campaign_with_steps_creation_responses(self) -> None:
self.set_tenant(0)
user = self.users[0]
campaign = Campaign.objects.create(name='this gonna be great', owner=user)
factory = TenantsAPIRequestFactory(force_authenticate=user)
request = factory.get('')
response = CampaignsViewSet.as_view({'get': 'list'})(request)
self.assertEqual(response.status_code, status.HTTP_200_OK, str(response.data))
campaigns_data = response.data
self.assertEqual(1, len(campaigns_data))
campaign_data = campaigns_data[0]
self.assertEqual(campaign.id, campaign_data['id'])
self.assertEqual(campaign.name, campaign_data['name'])
self.assertEqual(user.id, campaign_data['owner'])
contact = Contact.objects.create(email='*****@*****.**')
Participation.objects.create(campaign=campaign, contact=contact)
request = factory.get('')
response = CampaignsViewSet.as_view({'get': 'retrieve'})(request, pk=campaign_data['id'])
self.assertEqual(response.status_code, status.HTTP_200_OK, str(response.data))
campaign_data = response.data
self.assertEqual(1, campaign_data['contacts_count'])
step = Step.objects.create(campaign=campaign, start=datetime.time(9, 45), end=datetime.time(18, 30))
request = factory.get('')
response = CampaignsViewSet.as_view({'get': 'retrieve'})(request, pk=campaign_data['id'])
self.assertEqual(response.status_code, status.HTTP_200_OK, str(response.data))
campaign_data = response.data
steps_data = campaign_data['steps']
self.assertEqual(1, len(steps_data))
step_data = steps_data[0]
self.assertEqual(step.pk, step_data)
def test_contact_view(self) -> None:
self.set_tenant(0)
contact = Contact.objects.create(email='*****@*****.**',
first_name='James',
last_name='Bond')
factory = TenantsAPIRequestFactory(force_authenticate=self.user)
request = factory.get('')
response = ContactViewSet.as_view({'get': 'list'})(request)
contact_data = response.data
self.assertEqual(response.status_code, status.HTTP_200_OK,
str(contact_data))
self.assertEqual(1, len(contact_data))
self.assertEqual(contact.id, contact_data[0]['id'])
self.assertEqual(contact.first_name, contact_data[0]['first_name'])
self.assertEqual(contact.last_name, contact_data[0]['last_name'])
def test_unable_to_set_different_tenant_user_as_campaign_owner(self) -> None:
self.set_tenant(0)
user = self.users[0]
campaign = Campaign.objects.create(name='this gonna be great', owner=user)
factory = TenantsAPIRequestFactory(force_authenticate=user)
request = factory.get('')
response = CampaignsViewSet.as_view({'get': 'retrieve'})(request, pk=campaign.pk)
self.assertEqual(response.status_code, status.HTTP_200_OK, str(response.data))
other_user = self.users[2]
self.assertNotEqual(other_user.profile.tenant, user.profile.tenant)
factory = TenantsAPIRequestFactory(force_authenticate=user)
request = factory.patch('', data=dict(owner=other_user.pk))
response = CampaignsViewSet.as_view({'patch': 'partial_update'})(request, pk=campaign.pk)
self.assertEqual(status.HTTP_400_BAD_REQUEST, response.status_code, str(response.data))
self.assertDictEqual({'owner': [
'Invalid pk "%s" - object does not exist.' % other_user.pk
]}, response.data)
def test_sample_emails_preview(self) -> None:
self.set_tenant(0)
user = self.users[0]
kwargs = _generate_email_account_kwargs()
provider = EmailAccount.objects.create(user=user, email='*****@*****.**', **kwargs)
first_contact = Contact.objects.create(email='*****@*****.**', first_name='First', last_name='Smith')
second_contact = Contact.objects.create(email='*****@*****.**', first_name='Second', last_name='Smith')
campaign = Campaign.objects.create(name='cool campaign', owner=user, provider=provider, )
step = Step.objects.create(campaign=campaign, start=datetime.time(9, 45), end=datetime.time(18, 30))
EmailStage.objects.create(
step=step,
name='first email for second campaign',
subject='My dear {{ first_name|default:"cousin" }}!',
html_content='Please, send me some money, <b>{{ first_name|default:"dude" }}</b>!')
Participation.objects.bulk_create([
Participation(campaign=campaign, contact=first_contact),
Participation(campaign=campaign, contact=second_contact),
])
factory = TenantsAPIRequestFactory(force_authenticate=user)
request = factory.get('', data=dict(contact=second_contact.id))
response = CampaignsViewSet.as_view({'get': 'preview'})(request, pk=campaign.id)
self.assertEqual(status.HTTP_200_OK, response.status_code, str(response.data))
emails_list = response.data
self.assertEqual(1, len(emails_list))
email_data = emails_list[0]
self.assertEqual(second_contact.id, email_data['contact'])
self.assertEqual(step.id, email_data['step'])
self.assertListEqual([second_contact.email], email_data['to'])
self.assertEqual('Please, send me some money, <b>Second</b>!',
email_data['html_message'])
self.assertEqual('My dear Second!', email_data['subject'])