Made test_wrregioncache_table independent of wrregioncache.id.
[philipp/winterrodeln/wrpylib.git] / tests / test_wrmwdb.py
index 72929eaad131a141c0d1f10c59d4fcf37d3a036b..5ff54ef1efd6c619b4fc8b909d8f0e62d0340363 100644 (file)
@@ -1,22 +1,55 @@
 #!/usr/bin/python3.4
-# -*- coding: iso-8859-15 -*-
-import wrpylib.wrmwdb
-import sqlalchemy
 import unittest
+import sqlalchemy
+from sqlalchemy.engine import create_engine
+from sqlalchemy.orm import sessionmaker
+import wrpylib.wrmwdb
 
 
 class TestWrMwDb(unittest.TestCase):
-    def test_wrreport_table(self):
-        metadata = sqlalchemy.MetaData()
-        wrreport_table = wrpylib.wrmwdb.wrreport_table(metadata)
+    @classmethod
+    def setUpClass(cls):
+        cls.engine = create_engine('mysql://philipp@localhost:3306/philipp_winterrodeln_wiki?charset=utf8&use_unicode=1')
+        cls.Session = sessionmaker(bind=cls.engine)
 
+    def setUp(self):
+        self.metadata = sqlalchemy.MetaData()
+        self.session = self.Session()
 
-    def test_wrsledruncache_table(self):
-        metadata = sqlalchemy.MetaData()
-        wrsledruncache_table = wrpylib.wrmwdb.wrsledruncache_table(metadata)
+    def tearDown(self):
+        self.session.rollback()
 
+    def test_wrreport_table(self):
+        WrReport = wrpylib.wrmwdb.wrreport_table(self.metadata)
+        wrreport = self.session.query(WrReport).filter(WrReport.c.id==130).first()
+        self.assertEqual(wrreport.id, 130)
+        self.assertEqual(wrreport.page_title, 'Simmeringalm')
+
+    def test_wrsledruncache_table(self):
+        WrSledrunCache = wrpylib.wrmwdb.wrsledruncache_table(self.metadata)
+        wrsledruncache = self.session.query(WrSledrunCache).filter(WrSledrunCache.c.page_id==226).first()
+        self.assertEqual(wrsledruncache.page_id, 226)
+        self.assertEqual(wrsledruncache.page_title, 'Axamer_Lizum')
 
     def test_wrinncache_table(self):
-        metadata = sqlalchemy.MetaData()
-        wrinncache_table = wrpylib.wrmwdb.wrinncache_table(metadata)
+        WrInnCache = wrpylib.wrmwdb.wrinncache_table(self.metadata)
+        wrinncache = self.session.query(WrInnCache).filter(WrInnCache.c.page_id==324).first()
+        self.assertEqual(wrinncache.page_id, 324)
+        self.assertEqual(wrinncache.page_title, 'Alpengasthaus_Gleinserhof')
+
+    def test_wrreportcache_table(self):
+        WrReportCache = wrpylib.wrmwdb.wrreportcache_table(self.metadata)
+        wrreport = self.session.query(WrReportCache).filter(WrReportCache.c.page_id==226).first()
+        self.assertEqual(wrreport.page_title, 'Axamer Lizum')
+
+    def test_wrregion_table(self):
+        WrRegion = wrpylib.wrmwdb.wrregion_table(self.metadata)
+        wrregion = self.session.query(WrRegion).filter(WrRegion.c.id==18).first()
+        self.assertEqual(wrregion.id, 18)
+        self.assertEqual(wrregion.name, 'Salzburg')
 
+    def test_wrregioncache_table(self):
+        WrRegionCache = wrpylib.wrmwdb.wrregioncache_table(self.metadata)
+        wrregioncache = self.session.query(WrRegionCache).filter((WrRegionCache.c.region_id==27) & (WrRegionCache.c.page_id==54)).first()
+        self.assertEqual(wrregioncache.region_id, 27)
+        self.assertEqual(wrregioncache.page_id, 54)