Add test_rodelbahn_list.

[philipp/winterrodeln/wradmin.git] / tests / test_wradmin.py
diff --git a/tests/test_wradmin.py b/tests/test_wradmin.py

index 15256d979bbb36990a8e270182bd01367bef7161..7562803ab59bed4cee36e0d64cca2048c29a57b1 100644 (file)
--- a/tests/test_wradmin.py
+++ b/tests/test_wradmin.py
@@ -36,7 +36,7 @@ class TestDbWradmin(WradminTestBase):
              # fill database
              with open('tests/testdb.sql', 'r') as f:
                  sql = f.read()
-            with wradmin.get_db() as con:
+            with wradmin.model.meta.engine.begin() as con:
                  con.execute(sql)
  
                  # update dates
@@ -55,3 +55,14 @@ class TestDbWradmin(WradminTestBase):
          self.assertTrue(result.data.startswith(b'<!DOCTYPE html'))
          soup = bs4.BeautifulSoup(result.data, 'html.parser')
          self.assertEqual(soup.title.text, 'Rodelbahnberichte')
+
+    def test_rodelbahn_list(self):
+        result = self.app.get('/rodelbahn/list')
+        self.assertEqual(result.status_code, 200)
+        self.assertTrue(result.data.startswith(b'<!DOCTYPE html'))
+        soup = bs4.BeautifulSoup(result.data, 'html.parser')
+        self.assertEqual(soup.title.text, 'Rodelbahnen')
+        table = soup.find('table')
+        self.assertIsNotNone(table)
+        rows = table.find_all('tr')
+        self.assertEqual(len(rows), 9+1)