Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- def getCSV():
- response = HttpResponse(content_type='text/csv')
- response['Content-Disposition'] = 'attachment; filename=export.csv'
- writer = csv.writer(response, csv.excel)
- # ... Write some CSV content ...
- return response
- c = Client()
- r = c.get('/my/export/api')
- print(r.content)
- import csv
- import io
- def test_csv_export(self):
- response = self.client.get('/my/export/api')
- self.assertEqual(response.status_code, 200)
- content = response.content.decode('utf-8')
- cvs_reader = csv.reader(io.StringIO(content))
- body = list(cvs_reader)
- headers = body.pop(0)
- print(body)
- print(headers)
Add Comment
Please, Sign In to add comment