In this post I'll shortly explain how to unittest your Flask-Upload-Method. Normally you don't want to use your hard drive, so an in memory file would be ideal.
def test_fileupload(self):
user = User(email='john.doe@example.com', password='john1234', username='johnny')
user.admin = False
db.session().add(user)
db.session().commit()
with self.client as c:
rv = login(c, 'johnny', 'john1234')
self.assertTrue('index_page_7890' in str(rv.data))
rv = c.get('/upload', follow_redirects=True)
self.assertTrue("upload_site_234889" in str(rv.data))
testfile_bytes = b"fdjasdfjksjkadf"
testfile = (io.BytesIO(testfile_bytes), 'testing_134.txt')
data = {'files[]': testfile}
rv = c.post('/_upload', data=data, follow_redirects=True,
content_type='multipart/form-data')
print(rv)
ret = rv.json['files'][0]
self.assertEqual('testing_134.txt', ret['name'])
url = ret['url']
getfile = c.get(url, follow_redirects=True)
self.assertEqual(testfile_bytes, getfile.data)
As you can see I use with self.client as c:
to preserve the client context and navigate at my site. I created an user named johnny
and committed him to the databse. Then I navigate to the upload site and verify it's the correct page. I do this with an tag which only renders when testing is enabled.
With testfile = (io.BytesIO(testfile_bytes), 'testing_134.txt')
I create an tuple which contains the bytes from testfile_bytes
. This is in my case some gibberish.
I then post the data to my upload method. As I use the JQuery File Upload from Blueimp my upload method returns a json. ret = rv.json['files'][0]
Then I do some checks , e.g. if the filename is correct and I get the Url from the returned Json.
Then I download the file from the returned url and verify it contains the same bytes I uploaded.
If you press this Button it will Load Disqus-Comments. More on Disqus Privacy: Link