def test_get_disasters_given_from_date_and_to_date(self):
date_time = datetime.datetime(2014, 9, 17, 16, 0, 49, 807000)
attributes = dict(name=self.disaster_type,
locations=[self.district],
description="Big Flood",
date=date_time,
status="Assessment")
disaster1 = Disaster(**attributes).save()
attr2 = attributes.copy()
attr2["date"] = datetime.datetime(2014, 8, 17, 16, 0, 49, 807000)
disaster2 = Disaster(**attr2).save()
location_disasters = Disaster.from_(
self.district, **{
'from': '2014-08-17',
'to': '2014-09-17'
})
self.assertEqual(1, location_disasters.count())
self.assertIn(disaster2, location_disasters)
self.assertNotIn(disaster1, location_disasters)
location_disasters = Disaster.from_(self.district, **{
'from': None,
'to': None
})
self.assertEqual(2, location_disasters.count())
def test_get_disasters_given_from_date_and_to_date(self):
date_time = datetime.datetime(2014, 9, 17, 16, 0, 49, 807000)
attributes = dict(name=self.disaster_type, locations=[self.district], description="Big Flood",
date=date_time, status="Assessment")
disaster1 = Disaster(**attributes).save()
attr2 = attributes.copy()
attr2["date"] = datetime.datetime(2014, 8, 17, 16, 0, 49, 807000)
disaster2 = Disaster(**attr2).save()
location_disasters = Disaster.from_(self.district, **{'from': '2014-08-17', 'to': '2014-09-17'})
self.assertEqual(1, location_disasters.count())
self.assertIn(disaster2, location_disasters)
self.assertNotIn(disaster1, location_disasters)
location_disasters = Disaster.from_(self.district, **{'from': None, 'to': None})
self.assertEqual(2, location_disasters.count())
def test_get_disaster_from_a_location(self):
attributes = dict(name=self.disaster_type, locations=[self.district], description="Big Flood",
date="2014-12-01", status="Assessment")
disaster1 = Disaster(**attributes).save()
attr2 = attributes.copy()
attr2["locations"] = [Location(**dict(name='Some other location', type='district', parent=None)).save()]
disaster2 = Disaster(**attr2).save()
location_disasters = Disaster.from_(self.district)
self.assertEqual(1, location_disasters.count())
self.assertIn(disaster1, location_disasters)
self.assertNotIn(disaster2, location_disasters)
def test_get_messages_from_children_are_also_added(self):
attributes = dict(name=self.disaster_type, locations=[self.district], description="Big Flood",
date="2014-12-01",
status="Assessment")
disaster1 = Disaster(**attributes).save()
attr2 = attributes.copy()
attr2["locations"] = [Location(**dict(name='Kampala subcounty', type='subcounty', parent=self.district)).save()]
disaster2 = Disaster(**attr2).save()
location_disasters = Disaster.from_(self.district)
self.assertEqual(2, location_disasters.count())
self.assertIn(disaster1, location_disasters)
self.assertIn(disaster2, location_disasters)
def test_get_disaster_from_a_location(self):
attributes = dict(name=self.disaster_type,
locations=[self.district],
description="Big Flood",
date="2014-12-01",
status="Assessment")
disaster1 = Disaster(**attributes).save()
attr2 = attributes.copy()
attr2["locations"] = [
Location(**dict(name='Some other location',
type='district',
parent=None)).save()
]
disaster2 = Disaster(**attr2).save()
location_disasters = Disaster.from_(self.district)
self.assertEqual(1, location_disasters.count())
self.assertIn(disaster1, location_disasters)
self.assertNotIn(disaster2, location_disasters)
def test_get_messages_from_children_are_also_added(self):
attributes = dict(name=self.disaster_type,
locations=[self.district],
description="Big Flood",
date="2014-12-01",
status="Assessment")
disaster1 = Disaster(**attributes).save()
attr2 = attributes.copy()
attr2["locations"] = [
Location(**dict(name='Kampala subcounty',
type='subcounty',
parent=self.district)).save()
]
disaster2 = Disaster(**attr2).save()
location_disasters = Disaster.from_(self.district)
self.assertEqual(2, location_disasters.count())
self.assertIn(disaster1, location_disasters)
self.assertIn(disaster2, location_disasters)
