#!/usr/bin/python3.4 import unittest import sqlalchemy from sqlalchemy.engine import create_engine from sqlalchemy.orm import sessionmaker import wrpylib.wrmwdb class TestWrMwDb(unittest.TestCase): @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 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): 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')