def test_should_return_filtered_csv_when_csv_endpoint_is_called(self):
Disaster(**self.disaster).save()
disaster_attr2 = self.disaster.copy()
disaster_attr2['status'] = "Closed"
Disaster(**disaster_attr2).save()
disaster2 = self.disaster.copy()
disaster2['name'] = self.disaster_type2
sdisaster = Disaster(**disaster2).save()
response = self.client.get(self.CSV_ENDPOINT, format='csv')
expected_response = "name,description,location,status,date\r\n" \
"Fire,Big Flood,%s,Assessment,2014-12-01 00:00:00" % (self.district.name,)
expected_response = expected_response + "\r\nFire,Big Flood,%s,Closed,2014-12-01 00:00:00" % (self.district.name)
expected_response = expected_response + "\r\nFlood,Big Flood,%s,Assessment,2014-12-01 00:00:00" % (self.district.name)
self.assertEqual(200, response.status_code)
self.assertEqual(3, len(response.data))
self.assertTrue(isinstance(response.accepted_renderer, CSVRenderer))
self.assertEqual(collections.Counter(split_text(expected_response)), collections.Counter(split_text(response.content)))
response = self.client.get(self.CSV_ENDPOINT + '?status=Assessment&from=undefined', format='csv')
expected_response = "name,description,location,status,date\r\n" \
"Fire,Big Flood,%s,Assessment,2014-12-01 00:00:00" % (self.district.name,)
expected_response = expected_response + "\r\nFlood,Big Flood,%s,Assessment,2014-12-01 00:00:00" % (self.district.name)
self.assertEqual(200, response.status_code)
self.assertEqual(2, len(response.data))
self.assertTrue(isinstance(response.accepted_renderer, CSVRenderer))
self.assertEqual(collections.Counter(split_text(expected_response)), collections.Counter(split_text(response.content)))
sdisaster.date = "2014-12-02 00:00:00"
sdisaster.save()
response = self.client.get(self.CSV_ENDPOINT + '?status=Assessment&from=2014-12-02 00:00:00', format='csv')
expected_response = "name,description,location,status,date\r\n" \
"Flood,Big Flood,%s,Assessment,2014-12-02 00:00:00" % (self.district.name)
self.assertEqual(200, response.status_code)
self.assertEqual(1, len(response.data))
self.assertTrue(isinstance(response.accepted_renderer, CSVRenderer))
self.assertEqual(collections.Counter(split_text(expected_response)), collections.Counter(split_text(response.content)))
response = self.client.get(self.CSV_ENDPOINT + '?to=2014-12-01 00:00:00', format='csv')
expected_response = "name,description,location,status,date\r\n" \
"Fire,Big Flood,%s,Assessment,2014-12-01 00:00:00" % (self.district.name,)
expected_response = expected_response + "\r\nFire,Big Flood,%s,Closed,2014-12-01 00:00:00" % (self.district.name)
self.assertEqual(200, response.status_code)
self.assertEqual(2, len(response.data))
self.assertTrue(isinstance(response.accepted_renderer, CSVRenderer))
self.assertEqual(collections.Counter(split_text(expected_response)), collections.Counter(split_text(response.content)))
def test_should_return_filtered_csv_when_csv_endpoint_is_called_with_location_filter(self):
self.mobile_user['email'] = '*****@*****.**'
UserProfile(**self.mobile_user).save()
UserProfile(**self.masaka_user).save()
response = self.client.get(self.CSV_ENDPOINT + '?location=%s' % self.masaka.id, format='csv')
expected_response = "name,phone,email,district,subcounty\r\n" \
"Munamasaka,+256775019441,[email protected],%s,%s" % (self.masaka.name, self.subcounty.name)
self.assertEqual(200, response.status_code)
self.assertEqual(1, len(response.data))
self.assertTrue(isinstance(response.accepted_renderer, CSVRenderer))
result_set = set(split_text(expected_response)) & set(split_text(response.content))
self.assertEqual(collections.Counter(split_text(expected_response)), collections.Counter(split_text(response.content)))
def test_should_return_csv_when_csv_endpoint_is_called(self):
self.mobile_user['email'] = '*****@*****.**'
UserProfile(**self.mobile_user).save()
UserProfile(**self.masaka_user).save()
response = self.client.get(self.CSV_ENDPOINT, format='csv')
expected_response = "name,phone,email,district,subcounty\r\n" \
"timothy,+256775019449,[email protected],%s,%s" % (self.district.name,'')
expected_response = expected_response + "\r\nMunamasaka,+256775019441,[email protected],%s,%s" % (
self.masaka.name, self.subcounty.name)
self.assertEqual(200, response.status_code)
self.assertEqual(2, len(response.data))
self.assertTrue(isinstance(response.accepted_renderer, CSVRenderer))
self.assertEqual(collections.Counter(split_text(expected_response)),
collections.Counter(split_text(response.content)))
def __init__(self, text):
text = text or ''
self.text = split_text(text)
def test_text_messages_are_split_and_white_space_removed(self):
text_message = "NECOC* Kampala * Central Division* Fire"
self.assertEqual(["NECOC", "Kampala", "Central", "Division", "Fire"],
split_text(text_message))
def test_text_messages_are_split_and_white_space_removed(self):
text_message = "NECOC* Kampala * Central Division* Fire"
self.assertEqual(["NECOC", "Kampala", "Central", "Division", "Fire"], split_text(text_message))
