def test_get_messages_given_from_date_and_to_date(self):
location_name = 'Abim'
district = Location(
**dict(name=location_name, parent=None, type='district')).save()
message_attr = self.message.copy()
message_attr['location'] = district
message_attr['received_at'] = datetime.datetime(
2014, 12, 17, 16, 0, 49, 807000)
message = RapidProMessageBase(**message_attr).save()
self.message['location'] = district
message1 = RapidProMessageBase(**self.message).save()
location_messages = RapidProMessageBase.from_(
district, **{
'from': '2014-09-17',
'to': '2014-10-17'
})
self.assertEqual(1, location_messages.count())
self.assertIn(message1, location_messages)
self.assertNotIn(message, location_messages)
location_messages = RapidProMessageBase.from_(
district, **{
'from': None,
'to': None
})
self.assertEqual(2, location_messages.count())
def test_messages_by_from_and_to_date_and_location_when_initial_queryset_is_empty(
self):
location_name = 'Abim'
district = Location(
**dict(name=location_name, parent=None, type='district')).save()
message_attr = self.message.copy()
message_attr['location'] = district
message_attr['received_at'] = datetime.datetime(
2014, 12, 17, 16, 0, 49, 807000)
message = RapidProMessageBase(**message_attr).save()
self.message['location'] = district
message1 = RapidProMessageBase(**self.message).save()
location_messages = RapidProMessageBase.from_(
district, **{
'from': '2014-09-17',
'to': '2014-10-17'
})
self.assertEqual(1, location_messages.count())
self.assertIn(message1, location_messages)
self.assertNotIn(message, location_messages)
empty_queryset = RapidProMessageBase.objects().none()
location_messages = RapidProMessageBase.from_(
district, empty_queryset, **{
'from': '2014-09-17',
'to': '2014-10-17'
})
self.assertEqual(0, location_messages.count())
def test_get_messages_given_from_date_and_to_date(self):
location_name = 'Abim'
district = Location(**dict(name=location_name, parent=None, type='district')).save()
message_attr = self.message.copy()
message_attr['location'] = district
message_attr['received_at'] = datetime.datetime(2014, 12, 17, 16, 0, 49, 807000)
message = RapidProMessageBase(**message_attr).save()
self.message['location'] = district
message1 = RapidProMessageBase(**self.message).save()
location_messages = RapidProMessageBase.from_(district, **{'from': '2014-09-17', 'to': '2014-10-17'})
self.assertEqual(1, location_messages.count())
self.assertIn(message1, location_messages)
self.assertNotIn(message, location_messages)
location_messages = RapidProMessageBase.from_(district, **{'from': None, 'to': None})
self.assertEqual(2, location_messages.count())
def test_messages_by_from_and_to_date_and_location_when_initial_queryset_is_empty(self):
location_name = 'Abim'
district = Location(**dict(name=location_name, parent=None, type='district')).save()
message_attr = self.message.copy()
message_attr['location'] = district
message_attr['received_at'] = datetime.datetime(2014, 12, 17, 16, 0, 49, 807000)
message = RapidProMessageBase(**message_attr).save()
self.message['location'] = district
message1 = RapidProMessageBase(**self.message).save()
location_messages = RapidProMessageBase.from_(district, **{'from': '2014-09-17', 'to': '2014-10-17'})
self.assertEqual(1, location_messages.count())
self.assertIn(message1, location_messages)
self.assertNotIn(message, location_messages)
empty_queryset = RapidProMessageBase.objects().none()
location_messages = RapidProMessageBase.from_(district, empty_queryset, **{'from': '2014-09-17', 'to': '2014-10-17'})
self.assertEqual(0, location_messages.count())
def test_get_messages_from_a_location(self):
location_name = 'Abim'
district = Location(**dict(name=location_name, parent=None, type='district')).save()
message_attr = self.message.copy()
message_attr['location'] = district
message = RapidProMessageBase(**message_attr).save()
message1 = RapidProMessageBase(**self.message).save()
location_messages = RapidProMessageBase.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'
district = Location(
**dict(name=location_name, parent=None, type='district')).save()
message_attr = self.message.copy()
message_attr['location'] = district
message = RapidProMessageBase(**message_attr).save()
message1 = RapidProMessageBase(**self.message).save()
location_messages = RapidProMessageBase.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'
district = Location(**dict(name=location_name, parent=None, type='district')).save()
message_attr = self.message.copy()
message_attr['location'] = district
message = RapidProMessageBase(**message_attr).save()
message1_attr = message_attr.copy()
location_name = 'Wakiso'
district_son = Location(**dict(name=location_name, parent=district, type='village')).save()
message1_attr["location"] = district_son
message1 = RapidProMessageBase(**message1_attr).save()
location_messages = RapidProMessageBase.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'
district = Location(
**dict(name=location_name, parent=None, type='district')).save()
message_attr = self.message.copy()
message_attr['location'] = district
message = RapidProMessageBase(**message_attr).save()
message1_attr = message_attr.copy()
location_name = 'Wakiso'
district_son = Location(**dict(
name=location_name, parent=district, type='village')).save()
message1_attr["location"] = district_son
message1 = RapidProMessageBase(**message1_attr).save()
location_messages = RapidProMessageBase.from_(district)
self.assertEqual(2, location_messages.count())
self.assertIn(message, location_messages)
self.assertIn(message1, location_messages)
