diff --git a/main.py b/main.py index f99dffc..5db7cb3 100644 --- a/main.py +++ b/main.py @@ -70,5 +70,5 @@ def upload_file(): return render_template('index.html', mimetypes=mimetypes) -if __name__ == '__main__': +if __name__ == '__main__': # pragma: no cover app.run() diff --git a/tests.py b/tests.py index 9f56079..456cb4d 100644 --- a/tests.py +++ b/tests.py @@ -47,6 +47,15 @@ class FlaskrTestCase(unittest.TestCase): rv.data) self.assertEqual(rv.status_code, 200) + def test_get_upload_no_file_name(self): + rv = self.app.post('/', + data=dict( + file=(io.BytesIO(b"aaa"), ''), + ), follow_redirects=True) + self.assertIn(b'No file part', rv.data) + self.assertEqual(rv.status_code, 200) + + def test_get_upload_harmless_file(self): rv = self.app.post('/', data=dict(