def test_urgent_stockout_alert(self):
urgent_stockout_alert = UrgentStockoutAlert(TEST_DOMAIN)
urgent_stockout_alert.send()
self.assertEqual(SMS.objects.count(), 0)
create_stock_report(self.loc1, {'tp2': 0})
create_stock_report(self.loc2, {'tp2': 0})
create_stock_report(self.loc3, {'tp2': 0})
urgent_stockout_alert.send()
self.assertEqual(SMS.objects.count(), 1)
self.assertEqual(
SMS.objects.all().first().text, URGENT_STOCKOUT % {
'location': self.district.name,
'products': "Test Product2",
})
create_stock_report(self.loc1, {'tp': 0})
create_stock_report(self.loc2, {'tp': 0})
create_stock_report(self.loc3, {'tp': 0})
now = datetime.utcnow()
urgent_stockout_alert.send()
smses = SMS.objects.filter(date__gte=now)
self.assertEqual(smses.count(), 1)
self.assertEqual(
smses.first().text, URGENT_STOCKOUT % {
'location': self.district.name,
'products': "Test Product, Test Product2",
})
def test_product_type_filter(self):
"""
Notifications will not be sent if the stockout is a product type does
not interest the user.
"""
self.user.get_domain_membership(self.TEST_DOMAIN).program_id = self.program.get_id
self.user.save()
program = Program(domain=self.TEST_DOMAIN, name='Test Program 2')
program.save()
bootstrap_web_user(
username='******', domain=self.TEST_DOMAIN, phone_number='+44445', location=self.district,
password='******', email='*****@*****.**', user_data={}, program_id=program.get_id
)
assign_products_to_location(self.facility, [self.product])
assign_products_to_location(self.other_facility, [self.product])
assign_products_to_location(self.last_facility, [self.product])
create_stock_report(self.facility, {'tp': 0})
create_stock_report(self.other_facility, {'tp': 0})
create_stock_report(self.last_facility, {'tp': 0})
generated = list(UrgentStockoutAlert(self.TEST_DOMAIN).get_notifications())
self.assertEqual(len(generated), 1)
self.assertEqual(generated[0].user.get_id, self.user.get_id)
def test_urgent_stockout_alert(self):
urgent_stockout_alert = UrgentStockoutAlert(TEST_DOMAIN)
urgent_stockout_alert.send()
self.assertEqual(SMS.objects.count(), 0)
create_stock_report(self.loc1, {'tp2': 0})
create_stock_report(self.loc2, {'tp2': 0})
create_stock_report(self.loc3, {'tp2': 0})
urgent_stockout_alert.send()
self.assertEqual(SMS.objects.count(), 1)
self.assertEqual(SMS.objects.all().first().text, URGENT_STOCKOUT % {
'location': self.district.name,
'products': "Test Product2",
})
create_stock_report(self.loc1, {'tp': 0})
create_stock_report(self.loc2, {'tp': 0})
create_stock_report(self.loc3, {'tp': 0})
now = datetime.utcnow()
urgent_stockout_alert.send()
smses = SMS.objects.filter(date__gte=now)
self.assertEqual(smses.count(), 1)
self.assertEqual(smses.first().text, URGENT_STOCKOUT % {
'location': self.district.name,
'products': "Test Product, Test Product2",
})
def test_get_products_function(self):
urgent_stockout_alert = UrgentStockoutAlert(TEST_DOMAIN)
self.assertEqual(
len(
urgent_stockout_alert.get_sql_products_list(
self.district.sql_location)), 0)
create_stock_report(self.loc1, {'tp': 0})
self.assertEqual(
len(
urgent_stockout_alert.get_sql_products_list(
self.district.sql_location)), 0)
create_stock_report(self.loc2, {'tp': 0})
self.assertEqual(
len(
urgent_stockout_alert.get_sql_products_list(
self.district.sql_location)), 0)
create_stock_report(self.loc3, {'tp': 0})
self.assertEqual(
len(
urgent_stockout_alert.get_sql_products_list(
self.district.sql_location)), 1)
create_stock_report(self.loc1, {'tp2': 0})
create_stock_report(self.loc2, {'tp2': 0})
create_stock_report(self.loc3, {'tp2': 0})
self.assertEqual(
len(
urgent_stockout_alert.get_sql_products_list(
self.district.sql_location)), 2)
def test_minority_facility_stockout(self):
"""No notification because < 50% of the facilities don't send report."""
assign_products_to_location(self.facility, [self.product])
assign_products_to_location(self.other_facility, [self.product])
assign_products_to_location(self.last_facility, [self.product])
create_stock_report(self.other_facility, {'tp': 10})
create_stock_report(self.last_facility, {'tp': 10})
generated = list(UrgentStockoutAlert(self.TEST_DOMAIN).get_notifications())
self.assertEqual(len(generated), 0)
def test_partial_stock_coverage(self):
"""
Handle the case when not all facilities are expected to have stocked a
given product. i.e. if only one facility is expected to have a certain
product and it is stocked out then that is an urgent stockout.
"""
assign_products_to_location(self.facility, [self.product])
create_stock_report(self.facility, {'tp': 0})
generated = list(UrgentStockoutAlert(self.TEST_DOMAIN).get_notifications())
self.assertEqual(len(generated), 1)
self.assertEqual(generated[0].user.get_id, self.user.get_id)
def test_majority_facility_stockout(self):
"""Send a notification because > 50% of the facilities are stocked out of a product."""
assign_products_to_location(self.facility, [self.product])
assign_products_to_location(self.other_facility, [self.product])
assign_products_to_location(self.last_facility, [self.product])
create_stock_report(self.facility, {'tp': 0})
create_stock_report(self.other_facility, {'tp': 0})
create_stock_report(self.last_facility, {'tp': 10})
generated = list(UrgentStockoutAlert(self.TEST_DOMAIN).get_notifications())
self.assertEqual(len(generated), 1)
self.assertEqual(generated[0].user.get_id, self.user.get_id)
def test_country_user(self):
"""Country as well as region users should get notifications."""
other_user = bootstrap_web_user(
username='******', domain=self.TEST_DOMAIN, phone_number='+44445', location=self.country,
password='******', email='*****@*****.**', user_data={}
)
assign_products_to_location(self.facility, [self.product])
assign_products_to_location(self.other_facility, [self.product])
assign_products_to_location(self.last_facility, [self.product])
create_stock_report(self.facility, {'tp': 0})
create_stock_report(self.other_facility, {'tp': 0})
create_stock_report(self.last_facility, {'tp': 0})
generated = list(UrgentStockoutAlert(self.TEST_DOMAIN).get_notifications())
self.assertEqual(len(generated), 2)
self.assertEqual({generated[0].user.get_id, generated[1].user.get_id},
{self.user.get_id, other_user.get_id})
def test_get_products_function(self):
urgent_stockout_alert = UrgentStockoutAlert(TEST_DOMAIN)
self.assertEqual(len(urgent_stockout_alert.get_sql_products_list(self.district.sql_location)), 0)
create_stock_report(self.loc1, {'tp': 0})
self.assertEqual(len(urgent_stockout_alert.get_sql_products_list(self.district.sql_location)), 0)
create_stock_report(self.loc2, {'tp': 0})
self.assertEqual(len(urgent_stockout_alert.get_sql_products_list(self.district.sql_location)), 0)
create_stock_report(self.loc3, {'tp': 0})
self.assertEqual(len(urgent_stockout_alert.get_sql_products_list(self.district.sql_location)), 1)
create_stock_report(self.loc1, {'tp2': 0})
create_stock_report(self.loc2, {'tp2': 0})
create_stock_report(self.loc3, {'tp2': 0})
self.assertEqual(len(urgent_stockout_alert.get_sql_products_list(self.district.sql_location)), 2)
def urgent_stockout():
domains = EWSGhanaConfig.get_all_enabled_domains()
for domain in domains:
UrgentStockoutAlert(domain).send()
