Guest User

Untitled

a guest
Nov 23rd, 2017
73
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.63 KB | None | 0 0
  1. def getCSV():
  2. response = HttpResponse(content_type='text/csv')
  3. response['Content-Disposition'] = 'attachment; filename=export.csv'
  4. writer = csv.writer(response, csv.excel)
  5.  
  6. # ... Write some CSV content ...
  7.  
  8. return response
  9.  
  10. c = Client()
  11. r = c.get('/my/export/api')
  12. print(r.content)
  13.  
  14. import csv
  15. import io
  16.  
  17. def test_csv_export(self):
  18.  
  19. response = self.client.get('/my/export/api')
  20. self.assertEqual(response.status_code, 200)
  21.  
  22. content = response.content.decode('utf-8')
  23. cvs_reader = csv.reader(io.StringIO(content))
  24. body = list(cvs_reader)
  25. headers = body.pop(0)
  26.  
  27. print(body)
  28. print(headers)
Add Comment
Please, Sign In to add comment