Django Unit Test để thử nghiệm một download tập tin

Question

Ngay bây giờ tôi chỉ kiểm tra phản ứng của liên kết như vậy:

self.client = Client() 
response = self.client.get(url) 
self.assertEqual(response.status_code, 200) 

Có cách nào Django-ic để thử nghiệm một liên kết để xem nếu một tập tin tải về sự kiện thực sự diễn ra? Có vẻ như không thể tìm thấy nhiều tài nguyên về chủ đề này.

Answer 1

Nếu url có nghĩa là tạo một tệp thay vì phản hồi http "bình thường", thì content-type và/hoặc content-disposition của nó sẽ khác nhau.

đối tượng phản ứng về cơ bản là một cuốn từ điển, vì vậy bạn có thể để một cái gì đó giống như

self.assertEquals(
    response.get('Content-Disposition'), 
    "attachment; filename=mypic.jpg" 
) 

biết thêm: https://docs.djangoproject.com/en/dev/ref/request-response/#telling-the-browser-to-treat-the-response-as-a-file-attachment

UPD: Nếu bạn muốn đọc các nội dung thực tế của các tập tin đính kèm, bạn có thể sử dụng response.content. Ví dụ về tệp zip:

try: 
    f = io.BytesIO(response.content) 
    zipped_file = zipfile.ZipFile(f, 'r') 

    self.assertIsNone(zipped_file.testzip())   
    self.assertIn('my_file.txt', zipped_file.namelist()) 
finally: 
    zipped_file.close() 
    f.close()