def setup_data(setup_es):
"""Sets up data for the tests."""
with freeze_time('2017-01-01 13:00:00'):
company = CompanyFactory(name='Mercury trading', alias='Uranus supplies')
contact = ContactFactory(company=company, first_name='John', last_name='Doe')
order = OrderFactory(
reference='abcd',
primary_market_id=constants.Country.japan.value.id,
uk_region_id=constants.UKRegion.channel_islands.value.id,
assignees=[],
status=OrderStatus.draft,
company=company,
contact=contact,
discount_value=0,
delivery_date=dateutil_parse('2018-01-01').date(),
vat_verified=False,
)
OrderSubscriberFactory(
order=order,
adviser=AdviserFactory(dit_team_id=constants.Team.healthcare_uk.value.id),
)
OrderAssigneeFactory(
order=order,
adviser=AdviserFactory(dit_team_id=constants.Team.tees_valley_lep.value.id),
estimated_time=60,
)
with freeze_time('2017-02-01 13:00:00'):
company = CompanyFactory(name='Venus Ltd', alias='Earth outsourcing')
contact = ContactFactory(company=company, first_name='Jenny', last_name='Cakeman')
order = OrderWithAcceptedQuoteFactory(
reference='efgh',
primary_market_id=constants.Country.france.value.id,
uk_region_id=constants.UKRegion.east_midlands.value.id,
assignees=[],
status=OrderStatus.quote_awaiting_acceptance,
company=company,
contact=contact,
discount_value=0,
delivery_date=dateutil_parse('2018-02-01').date(),
vat_verified=False,
)
OrderSubscriberFactory(
order=order,
adviser=AdviserFactory(dit_team_id=constants.Team.td_events_healthcare.value.id),
)
OrderAssigneeFactory(
order=order,
adviser=AdviserFactory(dit_team_id=constants.Team.food_from_britain.value.id),
estimated_time=120,
)
setup_es.indices.refresh()
def test_non_empty(self):
"""
Test that calling GET returns the list of advisers subscribed to the order.
"""
advisers = AdviserFactory.create_batch(3)
order = OrderFactory()
for adviser in advisers[:2]:
OrderSubscriberFactory(order=order, adviser=adviser)
url = reverse(
'api-v3:omis:order:subscriber-list',
kwargs={'order_pk': order.id},
)
response = self.api_client.get(url)
assert response.status_code == status.HTTP_200_OK
assert response.json() == [{
'id': str(adviser.id),
'first_name': adviser.first_name,
'last_name': adviser.last_name,
'name': adviser.name,
'dit_team': {
'id': str(adviser.dit_team.id),
'name': adviser.dit_team.name,
'uk_region': {
'id': str(adviser.dit_team.uk_region.pk),
'name': adviser.dit_team.uk_region.name,
},
},
} for adviser in advisers[:2]]
def test_change_existing_list(self, allowed_status):
"""
Test that calling PUT with a different list of advisers completely changes
the subscriber list:
- advisers not in the list will be removed
- new advisers will be added
- existing advisers will be kept
"""
previous_advisers = AdviserFactory.create_batch(2)
order = OrderFactory(status=allowed_status)
subscriptions = [
OrderSubscriberFactory(order=order, adviser=adviser)
for adviser in previous_advisers
]
final_advisers = [
AdviserFactory(), # new
previous_advisers[1], # existing
]
url = reverse(
'api-v3:omis:order:subscriber-list',
kwargs={'order_pk': order.id},
)
response = self.api_client.put(
url,
[{'id': adviser.id} for adviser in final_advisers],
)
assert response.status_code == status.HTTP_200_OK
assert {adv['id'] for adv in response.json()} == {str(adv.id) for adv in final_advisers}
# check that the id of the existing subscription didn't change
assert order.subscribers.filter(id=subscriptions[1].id).exists()
def test_notify_on_order_subscriber_added(self):
"""
Test that a notification is sent to the adviser when they get subscribed to an order.
"""
order = OrderFactory(assignees=[])
notify.client.reset_mock()
subscriber = OrderSubscriberFactory(order=order)
assert notify.client.send_email_notification.called
call_args = notify.client.send_email_notification.call_args_list[0][1]
assert call_args['email_address'] == subscriber.adviser.contact_email
assert call_args[
'template_id'] == Template.you_have_been_added_for_adviser.value
def test_notify_on_order_subscriber_deleted(self, mocked_notify_client):
"""
Test that a notification is sent to the adviser when they get removed from an order.
"""
order = OrderFactory(assignees=[])
subscriber = OrderSubscriberFactory(order=order)
mocked_notify_client.reset_mock()
order.subscribers.all().delete()
assert mocked_notify_client.send_email_notification.called
call_args = mocked_notify_client.send_email_notification.call_args_list[
0][1]
assert call_args['email_address'] == subscriber.adviser.contact_email
assert call_args[
'template_id'] == Template.you_have_been_removed_for_adviser.value
def test_remove_all(self, allowed_status):
"""
Test that calling PUT with an empty list, removes all the subscribers.
"""
advisers = AdviserFactory.create_batch(2)
order = OrderFactory(status=allowed_status)
for adviser in advisers:
OrderSubscriberFactory(order=order, adviser=adviser)
url = reverse(
'api-v3:omis:order:subscriber-list',
kwargs={'order_pk': order.id},
)
response = self.api_client.put(url, [])
assert response.status_code == status.HTTP_200_OK
assert response.json() == []
def setup_data(es_with_collector):
"""Sets up data for the tests."""
with freeze_time('2017-01-01 13:00:00'):
company = CompanyFactory(
name='Mercury trading',
trading_names=[],
)
contact = ContactFactory(company=company, first_name='John', last_name='Doe')
order = OrderFactory(
reference='abcd',
primary_market_id=constants.Country.japan.value.id,
uk_region_id=constants.UKRegion.channel_islands.value.id,
assignees=[],
status=OrderStatus.DRAFT,
company=company,
contact=contact,
discount_value=0,
delivery_date=dateutil_parse('2018-01-01').date(),
completed_on=dateutil_parse('2018-05-01T13:00:00Z'),
vat_verified=False,
)
OrderSubscriberFactory(
order=order,
adviser=AdviserFactory(dit_team_id=constants.Team.healthcare_uk.value.id),
)
OrderAssigneeFactory(
order=order,
adviser=AdviserFactory(dit_team_id=constants.Team.tees_valley_lep.value.id),
estimated_time=60,
)
with freeze_time('2017-02-01 13:00:00'):
company = CompanyFactory(
name='Venus Ltd',
trading_names=['Maine Coon', 'Egyptian Mau', '3a'],
)
contact = ContactFactory(company=company, first_name='Jenny', last_name='Cakeman')
order = OrderWithAcceptedQuoteFactory(
reference='efgh',
primary_market_id=constants.Country.france.value.id,
uk_region_id=constants.UKRegion.east_midlands.value.id,
assignees=[],
status=OrderStatus.QUOTE_AWAITING_ACCEPTANCE,
company=company,
contact=contact,
discount_value=0,
delivery_date=dateutil_parse('2018-02-01').date(),
completed_on=dateutil_parse('2018-06-01T13:00:00Z'),
vat_verified=False,
)
OrderSubscriberFactory(
order=order,
adviser=AdviserFactory(dit_team_id=constants.Team.td_events_healthcare.value.id),
)
OrderAssigneeFactory(
order=order,
adviser=AdviserFactory(dit_team_id=constants.Team.food_from_britain.value.id),
estimated_time=120,
)
es_with_collector.flush_and_refresh()