def test_get_messages_from_a_location(self):
location_name = 'Abim'
text = "NECOC.%s. fire baba fire" % location_name
district = Location(**dict(name=location_name, parent=None, type='district')).save()
message_attr = self.message.copy()
message_attr['text'] = text
message = RapidProMessage(**message_attr).save()
message1_attr = message_attr.copy()
message1_attr["text"] = " message without location"
message1 = RapidProMessage(**message1_attr).save()
location_messages = RapidProMessage.from_(district)
self.assertEqual(1, location_messages.count())
self.assertIn(message, location_messages)
self.assertNotIn(message1, location_messages)
def test_get_messages_from_a_location(self):
location_name = 'Abim'
text = "NECOC.%s. fire baba fire" % location_name
district = Location(
**dict(name=location_name, parent=None, type='district')).save()
message_attr = self.message.copy()
message_attr['text'] = text
message = RapidProMessage(**message_attr).save()
message1_attr = message_attr.copy()
message1_attr["text"] = " message without location"
message1 = RapidProMessage(**message1_attr).save()
location_messages = RapidProMessage.from_(district)
self.assertEqual(1, location_messages.count())
self.assertIn(message, location_messages)
self.assertNotIn(message1, location_messages)
def test_get_messages_from_children_are_also_added(self):
location_name = 'Abim'
text = "NECOC.%s. fire baba fire" % location_name
district = Location(**dict(name=location_name, parent=None, type='district')).save()
message_attr = self.message.copy()
message_attr['text'] = text
message = RapidProMessage(**message_attr).save()
message1_attr = message_attr.copy()
location_name = 'Wakiso'
text = "NECOC.%s. fire baba fire" % location_name
district_son = Location(**dict(name=location_name, parent=district, type='village')).save()
message1_attr["text"] = text
message1 = RapidProMessage(**message1_attr).save()
location_messages = RapidProMessage.from_(district)
self.assertEqual(2, location_messages.count())
self.assertIn(message, location_messages)
self.assertIn(message1, location_messages)
def test_get_messages_from_children_are_also_added(self):
location_name = 'Abim'
text = "NECOC.%s. fire baba fire" % location_name
district = Location(
**dict(name=location_name, parent=None, type='district')).save()
message_attr = self.message.copy()
message_attr['text'] = text
message = RapidProMessage(**message_attr).save()
message1_attr = message_attr.copy()
location_name = 'Wakiso'
text = "NECOC.%s. fire baba fire" % location_name
district_son = Location(**dict(
name=location_name, parent=district, type='village')).save()
message1_attr["text"] = text
message1 = RapidProMessage(**message1_attr).save()
location_messages = RapidProMessage.from_(district)
self.assertEqual(2, location_messages.count())
self.assertIn(message, location_messages)
self.assertIn(message1, location_messages)
