def test_get_data_ok(self):
data_point_ = DataPoint()
category_ = Category.objects.create()
data_point_.category_id = category_.pk
data_point_.name = 'Test Data Point'
data_point_.resource = 'https://data.pr.gov/resource/abcd-1234.json'
data_point_.date_field = 'date'
data_point_.data_field = 'total'
data_point_.save()
responses.add(responses.GET, 'https://data.pr.gov/resource/abcd-1234.json',
body=self.test_data, status=200, content_type='application/json')
r = DataPoint.get_data(data_point_, token={'access_token': '123abc'})
self.assertEqual(200, r.status_code)
self.assertEqual([{'date': '2014-12-01T00:00:00', 'total': '6.534'},
{'date': '2014-11-01T00:00:00', 'total': '5.855'},
{'date': '2014-10-01T00:00:00', 'total': '6.154'},
{'date': '2014-09-01T00:00:00', 'total': '6.409'},
{'date': '2014-08-01T00:00:00', 'total': '6.499'},
{'date': '2014-07-01T00:00:00', 'total': '5.95'},
{'date': '2014-06-01T00:00:00', 'total': '5.759'},
{'date': '2014-05-01T00:00:00', 'total': '6.671'},
{'date': '2014-04-01T00:00:00', 'total': '6.276'},
{'date': '2014-03-01T00:00:00', 'total': '6.142'},
{'date': '2014-02-01T00:00:00', 'total': '6.715'},
{'date': '2014-01-01T00:00:00', 'total': '6.359'},
{'date': '2013-12-01T00:00:00', 'total': '6.134'}], r.json())
def test_get_data_error(self):
data_point_ = DataPoint()
category_ = Category.objects.create()
data_point_.category_id = category_.pk
data_point_.name = 'Test Data Point'
data_point_.resource = 'https://data.pr.gov/resource/abcd-1234.json'
data_point_.date_field = 'date'
data_point_.data_field = 'total'
data_point_.save()
responses.add(responses.GET, 'https://data.pr.gov/resource/abcd-1234.json',
body=self.test_data, status=404, content_type='application/json')
r = DataPoint.get_data(data_point_, token={'access_token': '123abc'})
self.assertEqual(404, r['status_code'])
self.assertEqual('Test Data Point', r['name'])
