Testowanie przesyłania plików w kolbie

Question

Używam Flask-Testing do moich testów integracji z Flask. Mam formularz, który ma plik do przesłania dla logo, do którego próbuję napisać testy, ale ciągle pojawia się komunikat o błędzie: TypeError: 'str' does not support the buffer interface.

Używam Pythona 3. Najbliższa odpowiedź, którą znalazłem to this, ale nie działa dla mnie.

To właśnie jeden z moich wielu prób wygląda następująco:

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) 

jaki sposób można przetestować przesłać plik w kolbie?

Answer 1

Emisja zakończyła się nie jest tak, gdy dodaje się content_type='multipart/form-data' do metody, jakiej oczekuje się od wszystkich wartości w data być plikami lub ciągami. W moim dyktowaniu danych znalazły się liczby całkowite, które zrozumiałem dzięki komentarzowi this.

więc rozwiązanie koniec skończyło się patrząc tak:

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

trzeba dwie rzeczy:

1.) content_type='multipart/form-data' w swoim .post()
2) w swoim podaniu data= w file=(BytesIO(b'my file contents'), "file_name.jpg")

Pełne przykład:

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)