Kiểm tra tệp tải lên trong Flask

Question

Tôi đang sử dụng Flask-Testing để kiểm tra tích hợp Flask của mình. Tôi có một biểu mẫu có tệp tải lên cho biểu trưng mà tôi đang cố gắng viết thử nhưng tôi vẫn gặp lỗi khi nói: TypeError: 'str' does not support the buffer interface.

Tôi đang sử dụng Python 3. Câu trả lời gần nhất tôi đã tìm thấy là this nhưng nó không hoạt động đối với tôi.

Đây là những gì một trong nhiều nỗ lực của tôi trông giống như:

def test_edit_logo(self): 
    """Test can upload logo.""" 
    data = {'name': 'this is a name', 'age': 12} 
    data['file'] = (io.BytesIO(b"abcdef"), 'test.jpg') 
    self.login() 
    response = self.client.post(
     url_for('items.save'), data=data, follow_redirects=True) 
    }) 
    self.assertIn(b'Your item has been saved.', response.data) 
    advert = Advert.query.get(1) 
    self.assertIsNotNone(item.logo) 

Làm thế nào để một bài kiểm tra tải lên tập tin trong Flask?

Answer 1

Vấn đề này đã kết thúc không phải là khi thêm content_type='multipart/form-data' vào phương thức post, nó sẽ mong đợi tất cả các giá trị trong data cho một trong hai là các tệp hoặc chuỗi. Đã có số nguyên trong dict dữ liệu của tôi mà tôi nhận ra nhờ nhận xét this.

Vì vậy, giải pháp cuối cùng đã kết thúc tìm kiếm như thế này:

def test_edit_logo(self): 
    """Test can upload logo.""" 
    data = {'name': 'this is a name', 'age': 12} 
    data = {key: str(value) for key, value in data.items()} 
    data['file'] = (io.BytesIO(b"abcdef"), 'test.jpg') 
    self.login() 
    response = self.client.post(
     url_for('adverts.save'), data=data, follow_redirects=True, 
     content_type='multipart/form-data' 
    ) 
    self.assertIn(b'Your item has been saved.', response.data) 
    advert = Item.query.get(1) 
    self.assertIsNotNone(item.logo) 

Answer 2

Bạn cần hai điều:

1.) content_type='multipart/form-data' trong bạn .post()
2.) trong data= đường chuyền của bạn trong file=(BytesIO(b'my file contents'), "file_name.jpg")

Một ví dụ đầy đủ:

data = dict(
     file=(BytesIO(b'my file contents'), "work_order.123"), 
    ) 

    response = app.post(url_for('items.save'), content_type='multipart/form-data', data=data)