]> ToastFreeware Gitweb - philipp/winterrodeln/wradmin.git/blobdiff - tests/test_wradmin.py
Use flask_sqlalchemy to manage the database connection.
[philipp/winterrodeln/wradmin.git] / tests / test_wradmin.py
index a4417015db79785d2d2bb0826656c0bc75a77d79..44dce22f7924ffb5ca322935cd7b8321cfcee4f9 100644 (file)
@@ -1,13 +1,20 @@
 import unittest
+
 import bs4  # beautiful soup
+
+import wradmin.app
 import wradmin.model
-import wradmin
+from wradmin.app import db
 
 
 class WradminTestBase(unittest.TestCase):
     def setUp(self):
         wradmin.app.config['TESTING'] = True
         self.app = wradmin.app.test_client()
+        self.db = db
+
+    def login(self):
+        return self.app.post('/login', data={'user_name': 'Johndoe', 'password': 'doejohn'}, follow_redirects=True)
 
 
 class TestNoDbWradmin(WradminTestBase):
@@ -19,6 +26,7 @@ class TestNoDbWradmin(WradminTestBase):
         self.assertEqual(soup.title.text, 'Hauptmenü')
 
     def test_coordtool(self):
+        self.login()
         result = self.app.get('/coordtool/index')
         self.assertEqual(result.status_code, 200)
         self.assertTrue(result.data.startswith(b'<!doctype html'))
@@ -26,6 +34,7 @@ class TestNoDbWradmin(WradminTestBase):
         self.assertEqual(soup.title.text, 'Koordinaten-Rechner')
 
     def test_coordtool_convert_ok(self):
+        self.login()
         result = self.app.post('/coordtool/convert', data={'input': '47.987, 11.123'}, follow_redirects=True)
         self.assertEqual(result.status_code, 200)
         self.assertTrue(result.data.startswith(b'<!doctype html'))
@@ -34,6 +43,7 @@ class TestNoDbWradmin(WradminTestBase):
         self.assertIn('GPX Format', str(soup))
 
     def test_coordtool_convert_error(self):
+        self.login()
         result = self.app.post('/coordtool/convert', data={'input': 'abc'}, follow_redirects=True)
         self.assertEqual(result.status_code, 200)
         self.assertTrue(result.data.startswith(b'<!doctype html'))
@@ -49,8 +59,8 @@ class TestDbWradmin(WradminTestBase):
             # fill database
             with open('tests/testdb.sql', 'r') as f:
                 sql = f.read()
-            with wradmin.model.meta.engine.begin() as con:
-                con.execute(sql)
+            with wradmin.db.engine.begin() as con:
+                con.execution_options(no_parameters=True).execute(sql)
 
                 # update dates
                 seconds_diff, = con.execute("SELECT TIMESTAMPDIFF(SECOND, '2017-03-31 07:00', now())").fetchone()
@@ -62,9 +72,6 @@ class TestDbWradmin(WradminTestBase):
                 con.execute("UPDATE wrreport SET date_invalid = date_invalid + INTERVAL %s SECOND", (seconds_diff,))
                 con.execute("UPDATE wrreport SET delete_date = delete_date + INTERVAL %s SECOND", (seconds_diff,))
 
-    def login(self):
-        return self.app.post('/login', data={'user': 'john', 'password': 'doe'})
-
     def login_and_get(self, url: str):
         result = self.app.get(url)
         self.assertEqual(result.status_code, 302)
@@ -88,11 +95,14 @@ class TestDbWradmin(WradminTestBase):
         self.assertIn('Brandstatt Alm', str(soup.table))
 
     def test_bericht_change_date_invalid_twoweeks(self):
+        self.login()
         url = '/bericht/change_date_invalid/19591'
         post_data = {'date_invalid': 'two_weeks', 'date_userdefined': '2018-01-30 18:26'}
-        result = self.app.post(url, data=post_data, follow_redirects=True)
+        result = self.app.post(url, data=post_data)
         self.assertEqual(result.status_code, 302)
-        self.login()
+        result = self.app.post(url, data=post_data, follow_redirects=True)
+        self.assertEqual(result.status_code, 200)
+        result = self.login()
         self.assertEqual(result.status_code, 200)
         result = self.app.post(url, data=post_data, follow_redirects=True)
         self.assertEqual(result.status_code, 200)
@@ -102,6 +112,7 @@ class TestDbWradmin(WradminTestBase):
         self.assertIn('Datum wurde erfolgreich geändert', str(soup))
 
     def test_bericht_change_date_invalid_userdefined(self):
+        self.login()
         post_data = {'date_invalid': 'userdefined', 'date_userdefined': '2018-01-30 15:09'}
         result = self.app.post('/bericht/change_date_invalid/19591', data=post_data, follow_redirects=True)
         self.assertEqual(result.status_code, 200)
@@ -112,7 +123,7 @@ class TestDbWradmin(WradminTestBase):
         self.assertIn('2018-01-30 15:09:00', str(soup))
 
     def test_gasthaus_list(self):
-        result = self.app.get('/gasthaus/list')
+        result = self.login_and_get('/gasthaus/list')
         self.assertEqual(result.status_code, 200)
         self.assertTrue(result.data.startswith(b'<!doctype html'))
         soup = bs4.BeautifulSoup(result.data, 'html.parser')
@@ -120,7 +131,7 @@ class TestDbWradmin(WradminTestBase):
         self.assertIn('Meißner', str(soup.table))
 
     def test_gasthaus_view(self):
-        result = self.app.get('/gasthaus/view/362')
+        result = self.login_and_get('/gasthaus/view/362')
         self.assertEqual(result.status_code, 200)
         self.assertTrue(result.data.startswith(b'<!doctype html'))
         soup = bs4.BeautifulSoup(result.data, 'html.parser')
@@ -128,6 +139,7 @@ class TestDbWradmin(WradminTestBase):
         self.assertIn('Meißner', str(soup))
 
     def test_gasthaus_update_success(self):
+        self.login()
         result = self.app.get('/gasthaus/update', follow_redirects=True)
         self.assertEqual(result.status_code, 200)
         self.assertTrue(result.data.startswith(b'<!doctype html'))
@@ -135,18 +147,18 @@ class TestDbWradmin(WradminTestBase):
         self.assertIn('Die Gasthausliste wurde erfolgreich aktualisiert.', str(soup))
 
     def test_gasthaus_update_fail(self):
-        session = wradmin.model.meta.Session
-        text = session.query(wradmin.model.MwText).get(8719)
-        text.old_text = text.old_text.replace('maria-waldrast@aon.at', 'abc@def@example.com')
-        session.commit()
+        text = self.db.session.query(wradmin.model.MwText).get(14415)
+        text.old_text = text.old_text.replace('lindauerhuette@aon.at', 'abc@def@example.com')
+        self.db.session.commit()
+        self.login()
         result = self.app.get('/gasthaus/update', follow_redirects=True)
         self.assertEqual(result.status_code, 200)
         self.assertTrue(result.data.startswith(b'<!doctype html'))
         soup = bs4.BeautifulSoup(result.data, 'html.parser')
-        self.assertIn('Fehler bei Gasthaus \'Maria Waldrast (Klostergasthaus)\'', str(soup))
+        self.assertIn('Fehler bei Gasthaus \'Lindauer Hütte (Gasthaus)\'', str(soup))
 
     def test_rodelbahn_list(self):
-        result = self.app.get('/rodelbahn/list')
+        result = self.login_and_get('/rodelbahn/list')
         self.assertEqual(result.status_code, 200)
         self.assertTrue(result.data.startswith(b'<!doctype html'))
         soup = bs4.BeautifulSoup(result.data, 'html.parser')
@@ -157,14 +169,14 @@ class TestDbWradmin(WradminTestBase):
         self.assertEqual(len(rows), 9+1)
 
     def test_rodelbahn_view(self):
-        result = self.app.get('/rodelbahn/view/926')
-        self.assertEqual(result.status_code, 200)
+        result = self.login_and_get('/rodelbahn/view/926')
         self.assertTrue(result.data.startswith(b'<!doctype html'))
         soup = bs4.BeautifulSoup(result.data, 'html.parser')
         self.assertIn('Rodelbahn', soup.title.text)
         self.assertIn('Hühnerspiel', str(soup))
 
     def test_rodelbahn_update_success(self):
+        self.login()
         result = self.app.get('/rodelbahn/update', follow_redirects=True)
         self.assertEqual(result.status_code, 200)
         self.assertTrue(result.data.startswith(b'<!doctype html'))
@@ -172,10 +184,10 @@ class TestDbWradmin(WradminTestBase):
         self.assertIn('Die Rodelbahnliste wurde erfolgreich aktualisiert.', str(soup))
 
     def test_rodelbahn_update_fail(self):
-        session = wradmin.model.meta.Session
-        text = session.query(wradmin.model.MwText).get(12106)
+        text = self.db.session.query(wradmin.model.MwText).get(15032)
         text.old_text = text.old_text.replace('Schwierigkeit        = mittel', 'Schwierigkeit        = geht so')
-        session.commit()
+        self.db.session.commit()
+        self.login()
         result = self.app.get('/rodelbahn/update', follow_redirects=True)
         self.assertEqual(result.status_code, 200)
         self.assertTrue(result.data.startswith(b'<!doctype html'))
@@ -183,6 +195,7 @@ class TestDbWradmin(WradminTestBase):
         self.assertIn('Fehler bei Rodelbahn \'Juifenalm\'', str(soup))
 
     def test_rodelbahn_update_regioncache(self):
+        self.login()
         result = self.app.get('/rodelbahn/update_regioncache', follow_redirects=True)
         self.assertEqual(result.status_code, 200)
         self.assertTrue(result.data.startswith(b'<!doctype html'))