#!/usr/bin/python3.4
-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')