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)