diff options
Diffstat (limited to 'test/test.py')
| -rw-r--r-- | test/test.py | 12 |
1 files changed, 12 insertions, 0 deletions
diff --git a/test/test.py b/test/test.py index 2d09662..7431881 100644 --- a/test/test.py +++ b/test/test.py | |||
| @@ -179,6 +179,18 @@ class Mat2WebTestCase(TestCase): | |||
| 179 | self.assertIn(b'.mp2', rv.data) | 179 | self.assertIn(b'.mp2', rv.data) |
| 180 | self.assertEqual(rv.status_code, 200) | 180 | self.assertEqual(rv.status_code, 200) |
| 181 | 181 | ||
| 182 | def test_get_upload_naughty_input(self): | ||
| 183 | rv = self.client.post( | ||
| 184 | '/', | ||
| 185 | data=dict( | ||
| 186 | file=(io.BytesIO(b"a"), '﷽'), | ||
| 187 | ), | ||
| 188 | follow_redirects=True | ||
| 189 | ) | ||
| 190 | self.assertEqual(rv.status_code, 200) | ||
| 191 | self.assertIn(b'Invalid Filename', rv.data) | ||
| 192 | |||
| 193 | |||
| 182 | 194 | ||
| 183 | if __name__ == '__main__': | 195 | if __name__ == '__main__': |
| 184 | unittest.main() | 196 | unittest.main() |
