mirror of
https://0xacab.org/jvoisin/mat2-web.git
synced 2025-02-23 08:39:57 -05:00
26 lines
617 B
Python
26 lines
617 B
Python
|
import unittest
|
||
|
|
|
||
|
|
import main
|
||
|
|
|
||
|
|
|
||
|
|
class FlaskrTestCase(unittest.TestCase):
|
||
|
|
def setUp(self):
|
||
|
|
main.app.testing = True
|
||
|
|
self.app = main.app.test_client()
|
||
|
|
|
||
|
|
def test_get_root(self):
|
||
|
|
rv = self.app.get('/')
|
||
|
|
self.assertIn(b'mat2-web', rv.data)
|
||
|
|
|
||
|
|
def test_get_download_dangerous_file(self):
|
||
|
|
rv = self.app.get('/download/\..\filename')
|
||
|
|
self.assertEqual(rv.status_code, 302)
|
||
|
|
|
||
|
|
def test_get_download_nonexistant_file(self):
|
||
|
|
rv = self.app.get('/download/non_existant')
|
||
|
|
self.assertEqual(rv.status_code, 302)
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
unittest.main()
|
||
|
|
|