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'])