From 19dfe057076a09615c5cc7515f7f50fb99b7e846 Mon Sep 17 00:00:00 2001
From: jvoisin <julien.voisin@dustri.org>
Date: Sun, 16 Dec 2018 20:36:02 +0100
Subject: [PATCH] Add some tests

---
 tests.py | 25 +++++++++++++++++++++++++
 1 file changed, 25 insertions(+)
 create mode 100644 tests.py

diff --git a/tests.py b/tests.py
new file mode 100644
index 0000000..7447554
--- /dev/null
+++ b/tests.py
@@ -0,0 +1,25 @@
+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()
+