Git-Mirrors/mat2-web
1
0
Fork 0
mirror of https://0xacab.org/jvoisin/mat2-web.git synced 2025-02-23 16:49:59 -05:00
Code Issues Packages Projects Releases Wiki Activity

Add a test with a valid file

Browse Source
This commit is contained in:
jvoisin 2018-12-16 21:37:15 +01:00
parent a71a390145
commit c289edb916
1 changed files with 14 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
tests.py
View File

@ -47,6 +47,20 @@ class FlaskrTestCase(unittest.TestCase):
rv.data) rv.data)
self.assertEqual(rv.status_code, 200) self.assertEqual(rv.status_code, 200)
def test_get_upload_harmless_file(self):
rv = self.app.post('/',
data=dict(
file=(io.BytesIO(b"Some text"), 'test.txt'),
), follow_redirects=True)
self.assertIn(b'/download/test.cleaned.txt', rv.data)
self.assertEqual(rv.status_code, 200)
rv = self.app.get('/download/test.cleaned.txt')
self.assertEqual(rv.status_code, 200)
rv = self.app.get('/download/test.cleaned.txt')
self.assertEqual(rv.status_code, 302)
if __name__ == '__main__': if __name__ == '__main__':
unittest.main() unittest.main()