def test_csv_name(self):
generator = ResponseFormatterGenerator('csv').get_formatter()
response = generator.run(self.query, {})
self.assertEquals(
response.get('Content-Disposition'),
'attachment; filename="{}"'.format(
constants.CSV_RESPONSE_FILENAME))
def test_csv_different_decimal_empty_rows(self):
query = Query()
field = Field.objects.get(identifier=self.single_series)
query.add_series(field)
query.add_series(field, rep_mode='percent_change_a_year_ago')
generator = ResponseFormatterGenerator('csv').get_formatter()
response = generator.run(query, {'decimal': ','})
self.assertFalse("None" in str(response.content))
def test_csv_response_header_description(self):
generator = ResponseFormatterGenerator('csv').get_formatter()
response = generator.run(self.query, {'header': 'descriptions'})
line_end = str(response.content).find('\n')
header = response.content[:line_end]
self.assertIn(self.series_desc, str(header))
def test_csv_response_header(self):
generator = ResponseFormatterGenerator('csv').get_formatter()
response = generator.run(self.query, {'header': 'titles'})
line_end = str(response.content).find('\n')
header = response.content[:line_end]
self.assertTrue(self.series_name in str(header))
def test_json_response(self):
generator = ResponseFormatterGenerator('json').get_formatter()
response = generator.run(self.query, {})
self.assertEqual(response.get('Content-Type'), 'application/json')
def test_csv_response(self):
generator = ResponseFormatterGenerator('csv').get_formatter()
response = generator.run(self.query, {})
self.assertTrue(response.get('Content-Type'), 'text/csv')
def get_formatter(_format):
generator = ResponseFormatterGenerator(_format)
return generator.get_formatter()
def setupHeader(self, query, header):
generator = ResponseFormatterGenerator('csv').get_formatter()
response = generator.run(query, {'header': header})
line_end = str(response.content).find('\n')
return response.content[:line_end]
